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.cc141 Status HandleFusion(HloInstruction* fusion);
163 TF_RETURN_IF_ERROR(HandleFusion(instruction)); in Run()
182 Status FusionInstructionMerger::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::FusionInstructionMerger
Dir_emitter.h94 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter_unnested.h185 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter.cc576 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1873 Status IrEmitterUnnested::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h239 Status HandleFusion(HloInstruction* fusion) override;
Dhlo_cost_analysis.h101 Status HandleFusion(const HloInstruction* fusion) override;
Ddfs_hlo_visitor_with_default.h158 Status HandleFusion(HloInstructionPtr fusion) override { in HandleFusion() function
Dhlo_verifier.h84 Status HandleFusion(HloInstruction*) override;
Ddfs_hlo_visitor.h251 virtual Status HandleFusion(HloInstructionPtr hlo) = 0;
Dhlo_verifier.cc748 Status ShapeVerifier::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::ShapeVerifier
1797 Status HandleFusion(HloInstruction* fusion) override { in HandleFusion() function in xla::__anon45533f2c0811::InstructionVerifier
Dhlo_cost_analysis.cc814 Status HloCostAnalysis::HandleFusion(const HloInstruction* fusion) { in HandleFusion() function in xla::HloCostAnalysis
Dhlo_evaluator.cc1938 Status HloEvaluator::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::HloEvaluator
Dhlo_instruction.cc3233 return visitor->HandleFusion(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h161 Status HandleFusion(HloInstruction* fusion) override;
Dir_emitter.cc2063 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::cpu::IrEmitter