Searched refs:leaf_logits (Results 1 – 1 of 1) sorted by relevance
141 const auto& leaf_logits = resource->node_value(tree_id, node_id); in Compute() local142 DCHECK_EQ(leaf_logits.size(), 1); in Compute()143 partial_tree_logit += leaf_logits[0]; in Compute()237 const auto& leaf_logits = resource->node_value(tree_id, node_id); in Compute() local238 DCHECK_EQ(leaf_logits.size(), logits_dimension_); in Compute()240 tree_logits[j] += tree_weight * leaf_logits[j]; in Compute()