Searched refs:leaf_logits (Results 1 – 1 of 1) sorted by relevance
153 const auto& leaf_logits = resource->node_value(tree_id, node_id); in Compute() local154 DCHECK_EQ(leaf_logits.size(), logits_dimension_); in Compute()159 tree_weight * (partial_tree_logits[j] + leaf_logits[j]); in Compute()248 const auto& leaf_logits = resource->node_value(tree_id, node_id); in Compute() local249 DCHECK_EQ(leaf_logits.size(), logits_dimension_); in Compute()251 tree_logits[j] += tree_weight * leaf_logits[j]; in Compute()