Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dconvolution_handler.cc504 auto ar = collective_ops_creator.create_cross_partition_all_reduce( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
731 lhs.state().collective_ops_creator.create_cross_partition_all_reduce( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
H A Dspmd_partitioner.h134 create_cross_partition_all_reduce; member
H A Dspmd_partitioner_util.cc1504 result.create_cross_partition_all_reduce = in GetPerGroupCollectiveOpsCreator()
1509 return creator.create_cross_partition_all_reduce( in GetPerGroupCollectiveOpsCreator()
H A Dspmd_partitioner.cc1476 auto result = state_.collective_ops_creator.create_cross_partition_all_reduce( in Broadcast()
2368 .create_cross_partition_all_reduce( in HandleConcatenate()
4445 return collectives_creator.create_cross_partition_all_reduce( in AllReduceAlongShardingDimsInternal()
4455 result = collectives_creator.create_cross_partition_all_reduce( in AllReduceAlongShardingDimsInternal()