Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc284 auto* new_split = node->mutable_bucketized_split(); in AddBucketizedSplitNode() local
285 new_split->set_feature_id(feature_id); in AddBucketizedSplitNode()
286 new_split->set_threshold(threshold); in AddBucketizedSplitNode()
287 new_split->set_left_id(*left_node_id); in AddBucketizedSplitNode()
288 new_split->set_right_id(*right_node_id); in AddBucketizedSplitNode()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dcandidate_graph_runner.cc108 auto* new_split = in GetSplit() local
111 *new_split->add_features() = id; in GetSplit()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc812 TreeNode new_split = *split_info.mutable_split_node(); in SplitTreeLayer() local
815 *new_split.mutable_node_metadata()->add_original_oblivious_leaves() = in SplitTreeLayer()
819 *tree_config->mutable_nodes(depth) = new_split; in SplitTreeLayer()