Searched refs:tree_logits (Results 1 – 1 of 1) sorted by relevance
242 std::vector<float> tree_logits(logits_dimension_, 0.0); in Compute() local251 tree_logits[j] += tree_weight * leaf_logits[j]; in Compute()266 output_logits(i, j) = tree_logits[j]; in Compute()348 const auto& tree_logits = resource->node_value(0, 0); in Compute() local349 DCHECK_EQ(tree_logits.size(), 1); in Compute()350 float tree_logit = resource->GetTreeWeight(0) * tree_logits[0]; in Compute()374 const auto& tree_logits = resource->node_value(tree_id, node_id); in Compute() local375 DCHECK_EQ(tree_logits.size(), 1); in Compute()376 tree_logit = resource->GetTreeWeight(tree_id) * tree_logits[0]; in Compute()