Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc370 OffsetCalculation left_halo_size_function = in TileToPartialReplicateHaloExchange() local
469 OffsetCalculation left_halo_size_function = in PadFromPartialReplicateShape() local
716 HloInstruction* hlo, const OffsetCalculation& left_halo_size_function, in ExchangeHalo()
886 const OffsetCalculation& left_halo_size_function, in ExchangeHaloAndGetValidData()
H A Dfft_handler.cc62 OffsetCalculation left_halo_size_function = in PadEachPartitionWithHaloExchange() local