Searched refs:split_sums (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 145 Tensor split_sums = in TEST() local
161 EXPECT_EQ(BestFeatureRegression(total_sums, total_squares, split_sums, in TEST()
180 Tensor split_sums = in TEST() local
194 EXPECT_EQ(BestFeatureRegression(total_sums, total_squares, split_sums, in TEST()
|
| D | tree_utils.cc | 156 const Tensor& split_sums, const Tensor& split_squares, in GetTwoBestRegression() argument
159 const int32 num_splits = static_cast<int32>(split_sums.shape().dim_size(1)); in GetTwoBestRegression()
161 static_cast<int32>(split_sums.shape().dim_size(2)); in GetTwoBestRegression()
171 split_sums.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
178 const auto splits_count_accessor = split_sums.tensor<float, 3>(); in GetTwoBestRegression()
196 const Tensor& split_sums, in BestFeatureRegression() argument
202 GetTwoBestRegression(total_sums, total_squares, split_sums, split_squares, in BestFeatureRegression()
210 const Tensor& split_sums, in BestSplitDominatesRegression() argument
|
| D | tree_utils.h | 115 const Tensor& split_sums,
122 const Tensor& split_sums,
|