Searched refs:num_target_replication (Results 1 – 1 of 1) sorted by relevance
348 int64 num_target_replication = 1; in PartialReplicateReshardCompatibleSharding() local350 num_target_replication = in PartialReplicateReshardCompatibleSharding()355 if (num_replication / num_target_replication != Product(expand_tile_sizes) || in PartialReplicateReshardCompatibleSharding()356 num_replication % num_target_replication != 0) { in PartialReplicateReshardCompatibleSharding()365 reshape_dimensions.push_back(num_target_replication); in PartialReplicateReshardCompatibleSharding()