Searched refs:output_node_ids (Results 1 – 3 of 3) sorted by relevance
114 std::vector<int32> output_node_ids; in Compute() local171 output_node_ids.push_back(node_id); in Compute()178 const int num_nodes = output_node_ids.size(); in Compute()212 output_node_ids_vec(i) = output_node_ids[i]; in Compute()287 std::vector<int32> output_node_ids; in Compute() local338 output_node_ids.push_back(node_id); in Compute()349 const int num_nodes = output_node_ids.size(); in Compute()401 output_node_ids_vec(i) = output_node_ids[i]; in Compute()642 std::vector<int32> output_node_ids; in Compute() local714 output_node_ids.push_back(node_id); in Compute()[all …]
108 auto output_node_ids = output_node_ids_t->vec<int32>(); in Compute() local115 output_node_ids.setZero(); in Compute()123 &output_node_ids, latest_tree, in Compute()174 output_node_ids(i) = node_id; in Compute()
249 std::vector<int> output_node_ids; in SortParams() local256 output_node_ids.emplace_back(node_id); in SortParams()274 for (int output_node_id : output_node_ids) { in SortParams()