Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc129 const Eigen::Tensor<float, 1, Eigen::RowMajor>& splits_square, in RegressionSplitScore() argument
144 splits_square.slice(offsets, extents), left_count); in RegressionSplitScore()
172 const auto splits_square = in GetTwoBestRegression() local
184 const auto right_squares = tc_square.broadcast(bcast) - splits_square; in GetTwoBestRegression()
188 totals_count_accessor, splits_sum, splits_square, in GetTwoBestRegression()