Searched refs:group_nodes (Results 1 – 1 of 1) sorted by relevance
904 std::vector<std::vector<NodeAndBool>>* group_nodes) { in GetGroupNodes() argument905 group_nodes->reserve(groups.group_devices.size()); in GetGroupNodes()910 (*group_nodes)[group_id].emplace_back(src, false); in GetGroupNodes()919 (*group_nodes)[group_id].emplace_back( in GetGroupNodes()925 for (const std::vector<NodeAndBool>& nodes : *group_nodes) { in GetGroupNodes()970 std::vector<std::vector<NodeAndBool>> group_nodes( in ApplyIOColocationGroups() local972 TF_RETURN_IF_ERROR(GetGroupNodes(groups, node, &group_nodes)); in ApplyIOColocationGroups()975 for (const std::vector<NodeAndBool>& nodes : group_nodes) { in ApplyIOColocationGroups()992 if (group_nodes[group_id].empty()) { in ApplyIOColocationGroups()995 const Node* group_node = group_nodes[group_id][0].first; in ApplyIOColocationGroups()