Searched defs:collective_ops_creator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 54 const SPMDCollectiveOpsCreator& collective_ops_creator, in PadEachPartitionWithHaloExchange()
161 const SPMDCollectiveOpsCreator& collective_ops_creator, in ShuffleDataWithAllToAll()
229 const SPMDCollectiveOpsCreator& collective_ops_creator, in GetFinalFftUsingCollectivePermute()
|
| D | spmd_partitioner.h | 188 SPMDCollectiveOpsCreator collective_ops_creator) in SpmdPartitioner()
288 SPMDCollectiveOpsCreator collective_ops_creator; member
|
| D | spmd_partitioner_util.cc | 412 const SPMDCollectiveOpsCreator& collective_ops_creator, in TileToPartialReplicateHaloExchange()
506 const SPMDCollectiveOpsCreator& collective_ops_creator, in PadFromPartialReplicateShape()
802 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHalo()
904 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHalo()
930 const SPMDCollectiveOpsCreator& collective_ops_creator, in ExchangeHaloAndGetValidData()
|
| D | convolution_handler.cc | 317 const auto& collective_ops_creator = lhs.state().collective_ops_creator; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
|
| D | spmd_partitioner.cc | 1353 const SPMDCollectiveOpsCreator& collective_ops_creator, in SpmdPartitioningVisitor()
3643 const SPMDCollectiveOpsCreator& collective_ops_creator, in CreateVisitor()
|
| D | spmd_partitioner_test.cc | 49 auto collective_ops_creator = in PartitionComputation() local
|