Searched refs:output_node_ids (Results 1 – 3 of 3) sorted by relevance
91 std::vector<int32> output_node_ids; in Compute() local146 output_node_ids.push_back(node_id); in Compute()153 const int num_nodes = output_node_ids.size(); in Compute()187 output_node_ids_vec(i) = output_node_ids[i]; in Compute()
101 auto output_node_ids = output_node_ids_t->vec<int32>(); in Compute() local108 output_node_ids.setZero(); in Compute()116 &output_node_ids, latest_tree](int32 start, int32 end) { in Compute()161 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()