Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc200 std::map<int32, SplitCandidate>* best_split_per_node) { in FindBestSplitsPerNode() argument
212 auto best_split_it = best_split_per_node->find(node_id); in FindBestSplitsPerNode()
218 if (TF_PREDICT_FALSE(best_split_it != best_split_per_node->end() && in FindBestSplitsPerNode()
220 const auto best_candidate = (*best_split_per_node)[node_id]; in FindBestSplitsPerNode()
226 (*best_split_per_node)[node_id] = candidate; in FindBestSplitsPerNode()
228 } else if (best_split_it == best_split_per_node->end() || in FindBestSplitsPerNode()
230 (*best_split_per_node)[node_id] = candidate; in FindBestSplitsPerNode()