Searched refs:partial_tree_logits (Results 1 – 1 of 1) sorted by relevance
128 std::vector<float> partial_tree_logits(logits_dimension_, 0.0); in Compute() local135 &partial_tree_logits); in Compute()143 partial_tree_logits[j] -= node_logits[j]; in Compute()159 tree_weight * (partial_tree_logits[j] + leaf_logits[j]); in Compute()160 partial_tree_logits[j] = 0; in Compute()