Searched refs:size_to_dim (Results 1 – 1 of 1) sorted by relevance
1318 std::map<int64, std::vector<int64>>& size_to_dim) { in GetReshardAllToAllSourceTargetDims() argument1319 size_to_dim[size].erase( in GetReshardAllToAllSourceTargetDims()1320 std::remove_if(size_to_dim[size].begin(), size_to_dim[size].end(), in GetReshardAllToAllSourceTargetDims()1322 size_to_dim[size].end()); in GetReshardAllToAllSourceTargetDims()1323 if (size_to_dim[size].empty()) { in GetReshardAllToAllSourceTargetDims()1324 size_to_dim.erase(size); in GetReshardAllToAllSourceTargetDims()