Home
last modified time | relevance | path

Searched defs:HandleAllReduce (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbfloat16_conversion_folding.cc189 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
H A Ddfs_hlo_visitor_with_default.h108 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
H A Dhlo_verifier.cc411 Status ShapeVerifier::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::ShapeVerifier
2470 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anon411d6f1b0e11::InstructionVerifier
H A Dhlo_cost_analysis.cc783 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter.cc549 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc1279 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2891 Status SpmdPartitioningVisitor::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::spmd::SpmdPartitioningVisitor