Searched defs:HandleAllReduce (Results 1 – 7 of 7) sorted by relevance
188 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
101 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
215 Status ShapeVerifier::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::ShapeVerifier1683 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anonfe08e0500711::InstructionVerifier
707 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
692 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
1217 Status IrEmitterUnnested::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitterUnnested
1507 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter