Home
last modified time | relevance | path

Searched refs:AllReduceAlongShardingDimsInternal (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h236 HloInstruction* AllReduceAlongShardingDimsInternal(
Dspmd_partitioner.cc3603 return AllReduceAlongShardingDimsInternal( in AllReduceAlongShardingDims()
3608 HloInstruction* SpmdPartitioner::AllReduceAlongShardingDimsInternal( in AllReduceAlongShardingDimsInternal() function in xla::spmd::SpmdPartitioner