Searched refs:split_counts (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils.cc | 81 const Tensor& split_counts, int32 accumulator, in GetTwoBestClassification() argument
85 const int32 num_splits = static_cast<int32>(split_counts.shape().dim_size(1)); in GetTwoBestClassification()
87 static_cast<int32>(split_counts.shape().dim_size(2)) - 1; in GetTwoBestClassification()
99 split_counts.Slice(accumulator, accumulator + 1).unaligned_flat<float>(); in GetTwoBestClassification()
114 const Tensor& split_counts, int32 accumulator) { in BestFeatureClassification() argument
119 GetTwoBestClassification(total_counts, split_counts, accumulator, &best_score, in BestFeatureClassification()
235 int MakeBootstrapWeights(const Tensor& total_counts, const Tensor& split_counts, in MakeBootstrapWeights() argument
239 static_cast<int32>(split_counts.shape().dim_size(2)) - 1; in MakeBootstrapWeights()
242 auto lc = split_counts.tensor<float, 3>(); in MakeBootstrapWeights()
262 const Tensor& split_counts, in BestSplitDominatesClassificationBootstrap() argument
[all …]
|
| D | tree_utils.h | 110 const Tensor& split_counts, int32 accumulator);
131 const Tensor& total_counts, const Tensor& split_counts, int32 accumulator,
137 const Tensor& split_counts,
144 const Tensor& split_counts,
|