Searched refs:tmp_partial_replicate_sharding (Results 1 – 1 of 1) sorted by relevance
1928 auto tmp_partial_replicate_sharding = in ReshardPartialReplicateWithAllToAll() local1933 sharding(), tmp_partial_replicate_sharding)) { in ReshardPartialReplicateWithAllToAll()1935 ReshardWithAllToAll(tmp_partial_replicate_sharding, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()1939 auto partitioned_hlo = Reshard(tmp_partial_replicate_sharding); in ReshardPartialReplicateWithAllToAll()