Home
last modified time | relevance | path

Searched refs:label_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dmodel_ops.cc346 const int32 label_dim = in Compute() local
351 param_proto_.is_regression() ? (std::max(1, label_dim)) : 1; in Compute()
Dstats_ops.cc293 const int32 label_dim = in Compute() local
298 param_proto_.is_regression() ? (std::max(1, label_dim)) : 1; in Compute()