Searched refs:nodes_changes (Results 1 – 1 of 1) sorted by relevance
391 std::vector<std::pair<int32, std::vector<float>>> nodes_changes; in PostPruneTree() local392 nodes_changes.reserve(num_nodes); in PostPruneTree()395 nodes_changes.emplace_back(i, prune_logit_changes); in PostPruneTree()402 &nodes_changes); in PostPruneTree()436 CalculateParentAndLogitUpdate(i, nodes_changes, &new_id, &logit_changes); in PostPruneTree()514 const std::vector<std::pair<int32, std::vector<float>>>& nodes_changes, in CalculateParentAndLogitUpdate() argument516 const int logits_dimension = nodes_changes[start_node_id].second.size(); in CalculateParentAndLogitUpdate()521 int32 parent = nodes_changes[node_id].first; in CalculateParentAndLogitUpdate()524 changes->at(i) += nodes_changes[node_id].second[i]; in CalculateParentAndLogitUpdate()527 parent = nodes_changes[node_id].first; in CalculateParentAndLogitUpdate()