Searched refs:HandleFusion (Results 1 – 16 of 16) sorted by relevance
141 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
94 Status HandleFusion(HloInstruction* fusion) override;
185 Status HandleFusion(HloInstruction* fusion) override;
576 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitter
1873 Status IrEmitterUnnested::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::gpu::IrEmitterUnnested
239 Status HandleFusion(HloInstruction* fusion) override;
101 Status HandleFusion(const HloInstruction* fusion) override;
158 Status HandleFusion(HloInstructionPtr fusion) override { in HandleFusion() function
84 Status HandleFusion(HloInstruction*) override;
251 virtual Status HandleFusion(HloInstructionPtr hlo) = 0;
748 Status ShapeVerifier::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::ShapeVerifier1797 Status HandleFusion(HloInstruction* fusion) override { in HandleFusion() function in xla::__anon45533f2c0811::InstructionVerifier
814 Status HloCostAnalysis::HandleFusion(const HloInstruction* fusion) { in HandleFusion() function in xla::HloCostAnalysis
1938 Status HloEvaluator::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::HloEvaluator
3233 return visitor->HandleFusion(this); in Visit()
161 Status HandleFusion(HloInstruction* fusion) override;
2063 Status IrEmitter::HandleFusion(HloInstruction* fusion) { in HandleFusion() function in xla::cpu::IrEmitter