Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc192 const auto& best_split_it = best_split_per_node->find(node_id); in FindBestSplitsPerNode() local
206 if (TF_PREDICT_FALSE(best_split_it != best_split_per_node->end() && in FindBestSplitsPerNode()
207 GainsAreEqual(gain, best_split_it->second.gain))) { in FindBestSplitsPerNode()
216 } else if (best_split_it == best_split_per_node->end() || in FindBestSplitsPerNode()
217 GainIsLarger(gain, best_split_it->second.gain)) { in FindBestSplitsPerNode()
417 auto best_split_it = best_split_per_node->find(node_id); in FindBestSplitsPerNode() local
431 if (TF_PREDICT_FALSE(best_split_it != best_split_per_node->end() && in FindBestSplitsPerNode()
432 GainsAreEqual(gain, best_split_it->second.gain))) { in FindBestSplitsPerNode()
441 } else if (best_split_it == best_split_per_node->end() || in FindBestSplitsPerNode()
442 GainIsLarger(gain, best_split_it->second.gain)) { in FindBestSplitsPerNode()