Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc64 OffsetCalculation left_halo_size_function = in PadEachPartitionWithHaloExchange() local
Dspmd_partitioner_util.cc455 OffsetCalculation left_halo_size_function = in TileToPartialReplicateHaloExchange() local
554 OffsetCalculation left_halo_size_function = in PadFromPartialReplicateShape() local
799 HloInstruction* hlo, const OffsetCalculation& left_halo_size_function, in ExchangeHalo()
924 const OffsetCalculation& left_halo_size_function, in ExchangeHaloAndGetValidData()