Home
last modified time | relevance | path

Searched defs:halo_exchange_result (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc79 auto halo_exchange_result = in PadEachPartitionWithHaloExchange() local
Dspmd_partitioner_util.cc466 auto halo_exchange_result = ExchangeHalo( in TileToPartialReplicateHaloExchange() local
566 auto halo_exchange_result = ExchangeHalo( in PadFromPartialReplicateShape() local
932 auto halo_exchange_result = in ExchangeHaloAndGetValidData() local