Home
last modified time | relevance | path

Searched refs:ExchangeHaloAndGetValidData (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h252 absl::optional<HloInstruction*> ExchangeHaloAndGetValidData(
Dconvolution_handler.cc485 auto concat = ExchangeHaloAndGetValidData( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
712 auto concat = ExchangeHaloAndGetValidData( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
Dspmd_partitioner.cc804 auto resharded = ExchangeHaloAndGetValidData( in ReshardAsWindowedInput()
3285 auto resharded_source = ExchangeHaloAndGetValidData( in HandleSelectAndScatter()
3303 auto resharded_data = ExchangeHaloAndGetValidData( in HandleSelectAndScatter()
Dspmd_partitioner_util.cc922 absl::optional<HloInstruction*> ExchangeHaloAndGetValidData( in ExchangeHaloAndGetValidData() function