Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc132 int32 accumulator, int32 num_regression_dims, int i) { in RegressionSplitScore() argument
133 Eigen::array<Eigen::Index, 1> offsets = {i * num_regression_dims + 1}; in RegressionSplitScore()
134 Eigen::array<Eigen::Index, 1> extents = {num_regression_dims - 1}; in RegressionSplitScore()
160 const int32 num_regression_dims = in GetTwoBestRegression() local
190 num_regression_dims, std::placeholders::_1), in GetTwoBestRegression()