Searched refs:GetTreeWeight (Results 1 – 3 of 3) sorted by relevance
156 const float tree_weight = resource->GetTreeWeight(tree_id); in Compute()247 const float tree_weight = resource->GetTreeWeight(tree_id); in Compute()350 float tree_logit = resource->GetTreeWeight(0) * tree_logits[0]; in Compute()376 tree_logit = resource->GetTreeWeight(tree_id) * tree_logits[0]; in Compute()
129 float GetTreeWeight(const int32 tree_id) const;
209 float BoostedTreesEnsembleResource::GetTreeWeight(const int32 tree_id) const { in GetTreeWeight() function in tensorflow::BoostedTreesEnsembleResource