Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h217 absl::optional<HloInstruction*> ExchangeHalo(
226 absl::optional<HloInstruction*> ExchangeHalo(
Dspmd_partitioner_util.cc466 auto halo_exchange_result = ExchangeHalo( in TileToPartialReplicateHaloExchange()
566 auto halo_exchange_result = ExchangeHalo( in PadFromPartialReplicateShape()
798 absl::optional<HloInstruction*> ExchangeHalo( in ExchangeHalo() function
899 absl::optional<HloInstruction*> ExchangeHalo( in ExchangeHalo() function
911 auto concat = ExchangeHalo(visiting_hlo, left_halo_size_functions[dim], in ExchangeHalo()
933 ExchangeHalo(hlo, left_halo_size_function, right_halo_size_function, dim, in ExchangeHaloAndGetValidData()
Dfft_handler.cc80 ExchangeHalo(hlo, left_halo_size_function, right_halo_size_function, in PadEachPartitionWithHaloExchange()