Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger.cc140 Status HandleFusion(HloInstruction* fusion);
161 TF_RETURN_IF_ERROR(HandleFusion(instruction)); in Run()
179 Status FusionInstructionMerger::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::FusionInstructionMerger
Dir_emitter.h97 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter_unnested.h167 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter.cc722 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc575 Status IrEmitterUnnested::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitterUnnested
686 return IrEmitter::HandleFusion(fusion); in HandleFusion()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h92 Status HandleFusion(const HloInstruction* fusion) override;
Ddfs_hlo_visitor_with_default.h139 Status HandleFusion(HloInstructionPtr fusion) override { in HandleFusion() function
Dhlo_evaluator.h217 Status HandleFusion(HloInstruction* fusion) override;
Dhlo_verifier.h75 Status HandleFusion(HloInstruction*) override;
Ddfs_hlo_visitor.h234 virtual Status HandleFusion(HloInstructionPtr hlo) = 0;
Dhlo_verifier.cc513 Status ShapeVerifier::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::ShapeVerifier
1335 Status HandleFusion(HloInstruction* fusion) override { in HandleFusion() function in xla::__anond588ae4e0811::InstructionVerifier
Dhlo_cost_analysis.cc614 Status HloCostAnalysis::HandleFusion(const HloInstruction* fusion) { in HandleFusion() function in xla::HloCostAnalysis
Dhlo_evaluator.cc1226 Status HloEvaluator::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::HloEvaluator
Dhlo_instruction.cc2634 return visitor->HandleFusion(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h175 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter.cc2149 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::cpu::IrEmitter