Searched refs:total_sums (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 136 Tensor total_sums = in TEST() local
161 EXPECT_EQ(BestFeatureRegression(total_sums, total_squares, split_sums, in TEST()
171 Tensor total_sums = 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
167 total_sums.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestRegression()
179 const auto totals_count_accessor = total_sums.tensor<float, 2>(); in GetTwoBestRegression()
194 int32 BestFeatureRegression(const Tensor& total_sums, in BestFeatureRegression() argument
202 GetTwoBestRegression(total_sums, total_squares, split_sums, split_squares, in BestFeatureRegression()
208 bool BestSplitDominatesRegression(const Tensor& total_sums, in BestSplitDominatesRegression() argument
|
| D | tree_utils.h | 113 int32 BestFeatureRegression(const Tensor& total_sums,
120 bool BestSplitDominatesRegression(const Tensor& total_sums,
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 803 const auto& total_sums = in ExtractFromProto() local
810 total_sum_[i] = total_sums.value(i).float_value(); in ExtractFromProto()
834 auto* total_sums = slot->mutable_post_init_leaf_stats() in PackToProto() local
842 total_sums->add_value()->set_float_value(total_sum_[i]); in PackToProto()
|