Searched refs:replicate_factors (Results 1 – 1 of 1) sorted by relevance
934 std::vector<int64> replicate_factors; in ReshardToPartialReplicateWithAllGather() local940 replicate_factors.emplace_back(replicate_factor); in ReshardToPartialReplicateWithAllGather()957 GroupShardingOnDims(temp_sharding, replicate_dims, replicate_factors); in ReshardToPartialReplicateWithAllGather()