Searched refs:GetTreeWeight (Results 1 – 4 of 4) sorted by relevance
147 resource->GetTreeWeight(tree_id) * partial_tree_logit; in Compute()236 const float tree_weight = resource->GetTreeWeight(tree_id); in Compute()343 float tree_logit = resource->GetTreeWeight(0) * tree_logits[0]; in Compute()368 tree_logit = resource->GetTreeWeight(tree_id) * tree_logits[0]; in Compute()
120 float GetTreeWeight(const int32 tree_id) const;
204 float BoostedTreesEnsembleResource::GetTreeWeight(const int32 tree_id) const { in GetTreeWeight() function in tensorflow::BoostedTreesEnsembleResource
110 float GetTreeWeight(const int32 index) const { in GetTreeWeight() function