Searched refs:tree_ensemble_resource (Results 1 – 1 of 1) sorted by relevance
81 BoostedTreesEnsembleResource* tree_ensemble_resource; in Compute() local83 &tree_ensemble_resource)); in Compute()84 tf_shared_lock l(*tree_ensemble_resource->get_mutex()); in Compute()85 core::ScopedUnref unref_me(tree_ensemble_resource); in Compute()88 const int num_trees = tree_ensemble_resource->num_trees(); in Compute()91 tree_ensemble_resource->IsTreeFinalized(num_trees - 1)) in Compute()95 tree_ensemble_resource->GetNumLayersAttempted(); in Compute()117 output_stamp_token_t->scalar<int64>()() = tree_ensemble_resource->stamp(); in Compute()124 tree_ensemble_resource->GetLastLayerNodesRange(&range_start, &range_end); in Compute()144 BoostedTreesEnsembleResource* tree_ensemble_resource; in Compute() local[all …]