Searched refs:total_squares (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 140 Tensor total_squares = test::AsTensor<float>( in TEST() local
161 EXPECT_EQ(BestFeatureRegression(total_sums, total_squares, split_sums, in TEST()
175 Tensor total_squares = test::AsTensor<float>( in TEST() local
194 EXPECT_EQ(BestFeatureRegression(total_sums, total_squares, split_sums, in TEST()
|
| D | tree_utils.cc | 155 void GetTwoBestRegression(const Tensor& total_sums, const Tensor& total_squares, in GetTwoBestRegression() argument
169 total_squares.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
195 const Tensor& total_squares, in BestFeatureRegression() argument
202 GetTwoBestRegression(total_sums, total_squares, split_sums, split_squares, in BestFeatureRegression()
209 const Tensor& total_squares, in BestSplitDominatesRegression() argument
|
| D | tree_utils.h | 114 const Tensor& total_squares,
121 const Tensor& total_squares,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 805 const auto& total_squares = in ExtractFromProto() local
811 total_sum_squares_[i] = total_squares.value(i).float_value(); in ExtractFromProto()
837 auto* total_squares = slot->mutable_post_init_leaf_stats() in PackToProto() local
843 total_squares->add_value()->set_float_value(total_sum_squares_[i]); in PackToProto()
|