Home
last modified time | relevance | path

Searched defs:collective_ops_creator (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc52 const SPMDCollectiveOpsCreator& collective_ops_creator, in PadEachPartitionWithHaloExchange()
159 const SPMDCollectiveOpsCreator& collective_ops_creator, in ShuffleDataWithAllToAll()
227 const SPMDCollectiveOpsCreator& collective_ops_creator, in GetFinalFftUsingCollectivePermute()
Dstateful_rng_spmd_partitioner.h33 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in StatefulRngSpmdPartitioningVisitor()
Dstateful_rng_spmd_partitioner.cc51 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
Dspmd_partitioner.h207 SPMDCollectiveOpsCreator collective_ops_creator) in SpmdPartitioner()
328 SPMDCollectiveOpsCreator collective_ops_creator; member
Dspmd_partitioner_util.cc327 const SPMDCollectiveOpsCreator& collective_ops_creator, in TileToPartialReplicateHaloExchange()
421 const SPMDCollectiveOpsCreator& collective_ops_creator, in PadFromPartialReplicateShape()
719 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHalo()
866 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHalo()
892 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHaloAndGetValidData()
Dconvolution_handler.cc315 const auto& collective_ops_creator = lhs.state().collective_ops_creator; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
Dspmd_partitioner.cc1990 const SPMDCollectiveOpsCreator& collective_ops_creator, in SpmdPartitioningVisitor()
4472 const SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
Dspmd_partitioner_test.cc60 auto collective_ops_creator = in PartitionComputation() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_spmd_partitioner.h32 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in GpuSpmdPartitioningVisitor()
Dgpu_spmd_partitioner.cc51 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()