Searched refs:PostPruneTree (Results 1 – 3 of 3) sorted by relevance
144 void PostPruneTree(const int32 current_tree, const int32 logits_dimension);
135 ensemble_resource->PostPruneTree(current_tree, logits_dimension); in Compute()355 ensemble_resource->PostPruneTree(current_tree, logits_dim_); in Compute()
379 void BoostedTreesEnsembleResource::PostPruneTree(const int32 current_tree, in PostPruneTree() function in tensorflow::BoostedTreesEnsembleResource