Searched defs:HandleAllReduce (Results 1 – 7 of 7) sorted by relevance
179 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
100 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
187 Status ShapeVerifier::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::ShapeVerifier1415 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anon5f5427a50811::InstructionVerifier
576 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
655 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
1389 Status IrEmitterUnnested::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitterUnnested
1309 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter