Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_conversion_folding.cc38 Status HandleAllReduce(HloInstruction* crs) override;
179 Status BFloat16ConversionFoldingVisitor::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::BFloat16ConversionFoldingVisitor
Dhlo_cost_analysis.h76 Status HandleAllReduce(const HloInstruction* crs) override;
Ddfs_hlo_visitor_with_default.h100 Status HandleAllReduce(HloInstructionPtr crs) override { in HandleAllReduce() function
Dhlo_verifier.h57 Status HandleAllReduce(HloInstruction* crs) override;
Ddfs_hlo_visitor.h116 virtual Status HandleAllReduce(HloInstructionPtr hlo) = 0;
Dhlo_verifier.cc187 Status ShapeVerifier::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::ShapeVerifier
1415 Status HandleAllReduce(HloInstruction* crs) override { in HandleAllReduce() function in xla::__anond588ae4e0811::InstructionVerifier
Dhlo_cost_analysis.cc576 Status HloCostAnalysis::HandleAllReduce(const HloInstruction* crs) { in HandleAllReduce() function in xla::HloCostAnalysis
Dhlo_instruction.cc2550 return visitor->HandleAllReduce(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h84 Status HandleAllReduce(HloInstruction* crs) override;
Dir_emitter_unnested.h181 Status HandleAllReduce(HloInstruction* crs) override;
Dir_emitter.cc655 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1389 Status IrEmitterUnnested::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h157 Status HandleAllReduce(HloInstruction* crs) override;
Dir_emitter.cc1309 Status IrEmitter::HandleAllReduce(HloInstruction* crs) { in HandleAllReduce() function in xla::cpu::IrEmitter