Searched refs:expand_tile_dims_indices (Results 1 – 1 of 1) sorted by relevance
329 std::vector<int64> expand_tile_dims_indices(rank, -1); in PartialReplicateReshardCompatibleSharding() local342 expand_tile_dims_indices[dim] = num_expand_dims++; in PartialReplicateReshardCompatibleSharding()376 if (expand_tile_dims_indices[dim] > -1) { in PartialReplicateReshardCompatibleSharding()377 perm.emplace_back(expand_tile_dims_indices[dim] + rank); in PartialReplicateReshardCompatibleSharding()