Searched refs:num_replication (Results 1 – 1 of 1) sorted by relevance
354 int64 num_replication = reshape_dimensions.back(); in PartialReplicateReshardCompatibleSharding() local355 if (num_replication / num_target_replication != Product(expand_tile_sizes) || in PartialReplicateReshardCompatibleSharding()356 num_replication % num_target_replication != 0) { in PartialReplicateReshardCompatibleSharding()