Home
last modified time | relevance | path

Searched refs:TileToPartialReplicateHaloExchange (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h389 absl::optional<HloInstruction*> TileToPartialReplicateHaloExchange(
Dspmd_partitioner_util.cc408 absl::optional<HloInstruction*> TileToPartialReplicateHaloExchange( in TileToPartialReplicateHaloExchange() function
Dspmd_partitioner.cc946 auto halo_exchange = TileToPartialReplicateHaloExchange( in ReshardToPartialReplicateWithAllGather()