Searched refs:num_expand_dims (Results 1 – 1 of 1) sorted by relevance
332 int num_expand_dims = 0; in PartialReplicateReshardCompatibleSharding() local342 expand_tile_dims_indices[dim] = num_expand_dims++; in PartialReplicateReshardCompatibleSharding()