Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc97 int32 current_tree = in Compute() local
107 ensemble_resource->GetNumLayersGrown(current_tree) + 1; in Compute()
109 << current_tree << " of ensemble of " << current_tree + 1 in Compute()
112 int32 node_id_start = ensemble_resource->GetNumNodes(current_tree); in Compute()
147 current_tree, node_id, feature_id, threshold, gain, left_contrib, in Compute()
151 int32 node_id_end = ensemble_resource->GetNumNodes(current_tree); in Compute()
154 ensemble_resource->SetNumLayersGrown(current_tree, new_num_layers); in Compute()
156 if (ensemble_resource->GetNumLayersGrown(current_tree) >= max_depth) { in Compute()
160 ensemble_resource->SetIsFinalized(current_tree, true); in Compute()
162 ensemble_resource->PostPruneTree(current_tree); in Compute()
[all …]
Dresources.cc305 void BoostedTreesEnsembleResource::PostPruneTree(const int32 current_tree) { in PostPruneTree() argument
307 auto* tree = tree_ensemble_->mutable_trees(current_tree); in PostPruneTree()
325 RecursivelyDoPostPrunePreparation(current_tree, 0, &nodes_to_delete, in PostPruneTree()
346 auto* post_prune_meta = tree_ensemble_->mutable_tree_metadata(current_tree) in PostPruneTree()
Dresources.h135 void PostPruneTree(const int32 current_tree);
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dtraining_ops.cc579 const int32 current_tree = num_trees - 1; in UpdateTreeWeightsIfDropout() local
580 trees_not_to_drop.insert(current_tree); in UpdateTreeWeightsIfDropout()
599 dropped_trees, dropped_trees_weights, current_tree, in UpdateTreeWeightsIfDropout()