Home
last modified time | relevance | path

Searched refs:best_split_it (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc212 auto best_split_it = best_split_per_node->find(node_id); in FindBestSplitsPerNode() local
218 if (TF_PREDICT_FALSE(best_split_it != best_split_per_node->end() && in FindBestSplitsPerNode()
219 GainsAreEqual(gain, best_split_it->second.gain))) { in FindBestSplitsPerNode()
228 } else if (best_split_it == best_split_per_node->end() || in FindBestSplitsPerNode()
229 GainIsLarger(gain, best_split_it->second.gain)) { in FindBestSplitsPerNode()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc89 auto best_split_it = best_splits->find(partition_id); in UpdateBestSplit() local
90 if (best_split_it == best_splits->end()) { in UpdateBestSplit()
96 SplitCandidate& best_split = best_split_it->second; in UpdateBestSplit()
529 auto best_split_it = best_splits->find(0); in FindBestSplitOblivious() local
530 if (best_split_it == best_splits->end()) { in FindBestSplitOblivious()
536 SplitCandidate& best_split = best_split_it->second; in FindBestSplitOblivious()