Searched defs:HandleAllReduce (Results 1 – 7 of 7) sorted by relevance
189 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
108 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
411 Status ShapeVerifier::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::ShapeVerifier2470 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anonb51dd00a0e11::InstructionVerifier
783 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
549 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
1279 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter
2891 Status SpmdPartitioningVisitor::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::spmd::SpmdPartitioningVisitor