Searched refs:get_group_index (Results 1 – 1 of 1) sorted by relevance
116 auto get_group_index = [&](absl::Span<const int64> tile_indices, in MergeSharding() local127 old_group_members[get_group_index(indices, old)].insert(device); in MergeSharding()131 new_group_members[get_group_index(indices, *to_merge)].insert(device); in MergeSharding()149 int64 old_group_id = get_group_index(old_index, old); in MergeSharding()150 int64 new_group_id = get_group_index(new_index, *to_merge); in MergeSharding()