Searched refs:old_group_members (Results 1 – 1 of 1) sorted by relevance
114 absl::flat_hash_map<int64, std::set<int64>> old_group_members; in MergeSharding() local127 old_group_members[get_group_index(indices, old)].insert(device); in MergeSharding()151 if (old_group_members[old_group_id].empty() || in MergeSharding()157 int64 smallest_old = *old_group_members[old_group_id].begin(); in MergeSharding()160 if (old_group_members[old_group_id].count(smallest_new) == 0) { in MergeSharding()172 old_group_members[old_group_id].erase(*device); in MergeSharding()