Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc287 auto* new_split = node->mutable_bucketized_split(); in AddBucketizedSplitNode() local
288 new_split->set_feature_id(candidate.feature_id); in AddBucketizedSplitNode()
289 new_split->set_threshold(candidate.threshold); in AddBucketizedSplitNode()
290 new_split->set_dimension_id(candidate.dimension_id); in AddBucketizedSplitNode()
291 new_split->set_left_id(*left_node_id); in AddBucketizedSplitNode()
292 new_split->set_right_id(*right_node_id); in AddBucketizedSplitNode()
299 new_split->set_default_direction(boosted_trees::DEFAULT_RIGHT); in AddBucketizedSplitNode()
301 new_split->set_default_direction(boosted_trees::DEFAULT_LEFT); in AddBucketizedSplitNode()
312 auto* new_split = node->mutable_categorical_split(); in AddCategoricalSplitNode() local
313 new_split->set_feature_id(candidate.feature_id); in AddCategoricalSplitNode()
[all …]