Home
last modified time | relevance | path

Searched refs:ReshardWithAllToAll (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h353 PartitionedHlo ReshardWithAllToAll(
Dspmd_partitioner.cc415 return ReshardWithAllToAll(target, *src_tgt_dims); in ReshardNoCache()
1086 PartitionedHlo PartitionedHlo::ReshardWithAllToAll( in ReshardWithAllToAll() function in xla::spmd::PartitionedHlo
1224 .ReshardWithAllToAll(target, remaining_source_target_dims); in ReshardWithAllToAll()
1298 ReshardWithAllToAll(tmp_partial_replicate_sharding, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()
1306 return partitioned_hlo.ReshardWithAllToAll(target, *src_tgt_dims); in ReshardPartialReplicateWithAllToAll()