Searched refs:ensemble_resource (Results 1 – 4 of 4) sorted by relevance
57 BoostedTreesEnsembleResource* ensemble_resource; in Compute() local59 &ensemble_resource)); in Compute()60 core::ScopedUnref unref_me(ensemble_resource); in Compute()61 mutex_lock l(*ensemble_resource->get_mutex()); in Compute()63 ensemble_resource->set_stamp(ensemble_resource->stamp() + 1); in Compute()98 UpdateGlobalAttemptsAndRetrieveGrowableTree(ensemble_resource); in Compute()107 ensemble_resource->GetNumLayersGrown(current_tree) + 1; in Compute()112 int32 node_id_start = ensemble_resource->GetNumNodes(current_tree); in Compute()146 ensemble_resource->AddBucketizedSplitNode( in Compute()151 int32 node_id_end = ensemble_resource->GetNumNodes(current_tree); in Compute()[all …]
72 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute() local74 &ensemble_resource)); in Compute()75 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()76 core::ScopedUnref unref_me(ensemble_resource); in Compute()80 output_stamp_token_t->scalar<int64>()() = ensemble_resource->stamp(); in Compute()91 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute() local93 &ensemble_resource)); in Compute()94 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()95 core::ScopedUnref unref_me(ensemble_resource); in Compute()99 output_stamp_token_t->scalar<int64>()() = ensemble_resource->stamp(); in Compute()[all …]
166 DecisionTreeEnsembleResource* ensemble_resource; in Compute() local169 &ensemble_resource)); in Compute()171 core::ScopedUnref unref_me(ensemble_resource); in Compute()173 tf_shared_lock l(*ensemble_resource->get_mutex()); in Compute()174 DoCompute(context, ensemble_resource, in Compute()177 DoCompute(context, ensemble_resource, in Compute()188 DecisionTreeEnsembleResource* ensemble_resource, in DoCompute() argument252 if (ensemble_resource->decision_tree_ensemble().has_growing_metadata()) { in DoCompute()255 trees_not_to_drop.insert(ensemble_resource->num_trees() - 1); in DoCompute()257 const std::vector<float> weights = ensemble_resource->GetTreeWeights(); in DoCompute()[all …]
196 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource; in Compute() local198 &ensemble_resource)); in Compute()199 core::ScopedUnref unref_me(ensemble_resource); in Compute()200 mutex_lock l(*ensemble_resource->get_mutex()); in Compute()209 CHECK(ensemble_resource->is_stamp_valid(stamp_token)); in Compute()219 ensemble_resource->set_stamp(next_stamp_token); in Compute()229 RetrieveBias(ensemble_resource, logits_dimension); in Compute()247 ensemble_resource->LastTreeMetadata()->set_is_finalized(true); in Compute()259 boosted_trees::models::DecisionTreeEnsembleResource* ensemble_resource, in RetrieveBias() argument261 const int32 num_trees = ensemble_resource->num_trees(); in RetrieveBias()[all …]