Searched refs:split_squares (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 153 Tensor split_squares = in TEST() local
162 split_squares, 2), in TEST()
187 Tensor split_squares = test::AsTensor<float>( in TEST() local
195 split_squares, 2), in TEST()
|
| D | tree_utils.cc | 156 const Tensor& split_sums, const Tensor& split_squares, in GetTwoBestRegression() argument
173 split_squares.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
197 const Tensor& split_squares, int32 accumulator) { in BestFeatureRegression() argument
202 GetTwoBestRegression(total_sums, total_squares, split_sums, split_squares, in BestFeatureRegression()
211 const Tensor& split_squares, in BestSplitDominatesRegression() argument
|
| D | tree_utils.h | 116 const Tensor& split_squares, int32 accumulator);
123 const Tensor& split_squares,
|