Searched refs:best_splits (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | training_ops.cc | 64 std::map<int32, SplitCandidate>* best_splits) { in UpdateBestSplit() argument
89 auto best_split_it = best_splits->find(partition_id); in UpdateBestSplit()
90 if (best_split_it == best_splits->end()) { in UpdateBestSplit()
91 best_splits->insert(std::make_pair(partition_id, std::move(*split))); in UpdateBestSplit()
377 std::map<int32, SplitCandidate> best_splits; in Compute() local
381 gains_list, splits_list, &best_splits); in Compute()
385 FindBestSplitOblivious(context, gains_list, splits_list, &best_splits); in Compute()
390 if (best_splits.empty()) { in Compute()
410 for (auto& split_entry : best_splits) { in Compute()
417 SplitTreeLayer(&best_splits[0], tree_config, ensemble_resource); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 93 std::map<int32, SplitCandidate> best_splits; in Compute() local
95 &best_splits); in Compute()
101 if (best_splits.empty()) { in Compute()
114 for (auto& split_entry : best_splits) { in Compute()
|