Searched refs:nodes_changes (Results 1 – 1 of 1) sorted by relevance
392 std::vector<std::pair<int32, std::vector<float>>> nodes_changes; in PostPruneTree() local393 nodes_changes.reserve(num_nodes); in PostPruneTree()396 nodes_changes.emplace_back(i, prune_logit_changes); in PostPruneTree()403 &nodes_changes); in PostPruneTree()436 CalculateParentAndLogitUpdate(i, nodes_changes, &new_id, &logit_changes); in PostPruneTree()513 const std::vector<std::pair<int32, std::vector<float>>>& nodes_changes, in CalculateParentAndLogitUpdate() argument515 const int logits_dimension = nodes_changes[start_node_id].second.size(); in CalculateParentAndLogitUpdate()520 int32 parent = nodes_changes[node_id].first; in CalculateParentAndLogitUpdate()523 changes->at(i) += nodes_changes[node_id].second[i]; in CalculateParentAndLogitUpdate()526 parent = nodes_changes[node_id].first; in CalculateParentAndLogitUpdate()