Searched refs:HandleCustomCall (Results 1 – 15 of 15) sorted by relevance
99 Status HandleCustomCall(HloInstruction* custom_call) override;
164 Status HandleCustomCall(HloInstruction* custom_call) override;
744 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
373 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::IrEmitterUnnested529 return IrEmitter::HandleCustomCall(custom_call); in HandleCustomCall()
94 Status HandleCustomCall(const HloInstruction* custom_call) override;
145 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
241 Status HandleCustomCall(HloInstruction* custom_call) override;
77 Status HandleCustomCall(HloInstruction*) override;
236 virtual Status HandleCustomCall(HloInstructionPtr hlo) = 0;
643 Status HloCostAnalysis::HandleCustomCall(const HloInstruction*) { in HandleCustomCall() function in xla::HloCostAnalysis
544 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
1576 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
2640 return visitor->HandleCustomCall(this); in Visit()
177 Status HandleCustomCall(HloInstruction* custom_call) override;
2229 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter