Home
last modified time | relevance | path

Searched refs:HandleAllReduce (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding.cc43 Status HandleAllReduce(HloInstruction* crs) override;
190 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
Dhlo_cost_analysis.h80 Status HandleAllReduce(const HloInstruction* crs) override;
Ddfs_hlo_visitor_with_default.h104 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
Dhlo_verifier.h60 Status HandleAllReduce(HloInstruction* hlo) override;
Ddfs_hlo_visitor.h120 virtual Status HandleAllReduce(HloInstructionPtr hlo) = 0;
Dhlo_verifier.cc280 Status ShapeVerifier::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::ShapeVerifier
1877 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anon45533f2c0811::InstructionVerifier
Dhlo_cost_analysis.cc748 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
Dhlo_instruction.cc3133 return visitor->HandleAllReduce(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h83 Status HandleAllReduce(HloInstruction* crs) override;
Dir_emitter_unnested.h211 Status HandleAllReduce(HloInstruction* hlo) override;
Dir_emitter.cc567 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc3178 Status IrEmitterUnnested::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h142 Status HandleAllReduce(HloInstruction* crs) override;
Dir_emitter.cc1201 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h409 Status HandleAllReduce(HloInstruction* hlo) override;
Dspmd_partitioner.cc2230 Status SpmdPartitioningVisitor::HandleAllReduce(HloInstruction* hlo) { in HandleAllReduce() function in xla::spmd::SpmdPartitioningVisitor