Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc88 int32 current_tree = in Compute() local
98 ensemble_resource->GetNumLayersGrown(current_tree) + 1; in Compute()
100 << current_tree << " of ensemble of " << current_tree + 1 in Compute()
103 int32 node_id_start = ensemble_resource->GetNumNodes(current_tree); in Compute()
119 ensemble_resource->AddBucketizedSplitNode(current_tree, split_entry, in Compute()
124 int32 node_id_end = ensemble_resource->GetNumNodes(current_tree); in Compute()
127 ensemble_resource->SetNumLayersGrown(current_tree, new_num_layers); in Compute()
129 if (ensemble_resource->GetNumLayersGrown(current_tree) >= max_depth) { in Compute()
133 ensemble_resource->SetIsFinalized(current_tree, true); in Compute()
135 ensemble_resource->PostPruneTree(current_tree, logits_dimension); in Compute()
[all …]
Dresources.cc379 void BoostedTreesEnsembleResource::PostPruneTree(const int32 current_tree, in PostPruneTree() argument
382 auto* tree = tree_ensemble_->mutable_trees(current_tree); in PostPruneTree()
401 RecursivelyDoPostPrunePreparation(current_tree, 0, &nodes_to_delete, in PostPruneTree()
422 auto* post_prune_meta = tree_ensemble_->mutable_tree_metadata(current_tree) in PostPruneTree()
Dresources.h144 void PostPruneTree(const int32 current_tree, const int32 logits_dimension);