Searched refs:HandleAllReduceSingleReplica (Results 1 – 2 of 2) sorted by relevance
188 Status HandleAllReduceSingleReplica(HloInstruction* crs);
1066 Status IrEmitter::HandleAllReduceSingleReplica(HloInstruction* crs) { in HandleAllReduceSingleReplica() function in xla::cpu::IrEmitter1203 return HandleAllReduceSingleReplica(crs); in HandleAllReduce()