Searched refs:replicate_factor (Results 1 – 2 of 2) sorted by relevance
453 int64 replicate_factor = src_sharding.tile_assignment().dim(dim) / in TileToPartialReplicateHaloExchange() local457 src_per_shard_size - dst_per_shard_size, 0, replicate_factor)); in TileToPartialReplicateHaloExchange()
936 int64 replicate_factor = temp_sharding.tile_assignment().dim(dim) / in ReshardToPartialReplicateWithAllGather() local938 if (replicate_factor > 1) { in ReshardToPartialReplicateWithAllGather()940 replicate_factors.emplace_back(replicate_factor); in ReshardToPartialReplicateWithAllGather()