Searched refs:tmp_partial_replicate_sharding (Results 1 – 1 of 1) sorted by relevance
1291 auto tmp_partial_replicate_sharding = in ReshardPartialReplicateWithAllToAll() local1296 sharding(), tmp_partial_replicate_sharding)) { in ReshardPartialReplicateWithAllToAll()1298 ReshardWithAllToAll(tmp_partial_replicate_sharding, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()1302 auto partitioned_hlo = Reshard(tmp_partial_replicate_sharding); in ReshardPartialReplicateWithAllToAll()