Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dtree_utils.cc126 const Eigen::Tensor<float, 3, Eigen::RowMajor>& splits_count_accessor, in RegressionSplitScore() argument
135 float left_count = splits_count_accessor(accumulator, i, 0); in RegressionSplitScore()
178 const auto splits_count_accessor = split_sums.tensor<float, 3>(); in GetTwoBestRegression() local
187 std::bind(RegressionSplitScore, splits_count_accessor, in GetTwoBestRegression()