Searched defs:collective_ops_creator (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 52 const SPMDCollectiveOpsCreator& collective_ops_creator, in PadEachPartitionWithHaloExchange() 159 const SPMDCollectiveOpsCreator& collective_ops_creator, in ShuffleDataWithAllToAll() 227 const SPMDCollectiveOpsCreator& collective_ops_creator, in GetFinalFftUsingCollectivePermute()
|
| D | stateful_rng_spmd_partitioner.h | 33 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in StatefulRngSpmdPartitioningVisitor()
|
| D | stateful_rng_spmd_partitioner.cc | 51 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
|
| D | spmd_partitioner.h | 207 SPMDCollectiveOpsCreator collective_ops_creator) in SpmdPartitioner() 328 SPMDCollectiveOpsCreator collective_ops_creator; member
|
| D | spmd_partitioner_util.cc | 327 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()
|
| D | convolution_handler.cc | 315 const auto& collective_ops_creator = lhs.state().collective_ops_creator; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
|
| D | spmd_partitioner.cc | 1990 const SPMDCollectiveOpsCreator& collective_ops_creator, in SpmdPartitioningVisitor() 4472 const SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
|
| D | spmd_partitioner_test.cc | 60 auto collective_ops_creator = in PartitionComputation() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_spmd_partitioner.h | 32 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in GpuSpmdPartitioningVisitor()
|
| D | gpu_spmd_partitioner.cc | 51 const spmd::SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
|