Searched refs:AllReduceAlongShardingDimsInternal (Results 1 – 2 of 2) sorted by relevance
236 HloInstruction* AllReduceAlongShardingDimsInternal(
3603 return AllReduceAlongShardingDimsInternal( in AllReduceAlongShardingDims()3608 HloInstruction* SpmdPartitioner::AllReduceAlongShardingDimsInternal( in AllReduceAlongShardingDimsInternal() function in xla::spmd::SpmdPartitioner