Searched refs:new_group_members (Results 1 – 1 of 1) sorted by relevance
115 absl::flat_hash_map<int64, std::set<int64>> new_group_members; in MergeSharding() local131 new_group_members[get_group_index(indices, *to_merge)].insert(device); in MergeSharding()152 new_group_members[new_group_id].empty()) { in MergeSharding()158 int64 smallest_new = *new_group_members[new_group_id].begin(); in MergeSharding()166 if (new_group_members[new_group_id].count(smallest_old) == 0) { in MergeSharding()173 new_group_members[new_group_id].erase(*device); in MergeSharding()