Home
last modified time | relevance | path

Searched refs:HandleCustomCall (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h99 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter_unnested.h164 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter.cc744 Status IrEmitter::HandleCustomCall(HloInstruction*) { in HandleCustomCall() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc373 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::gpu::IrEmitterUnnested
529 return IrEmitter::HandleCustomCall(custom_call); in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h94 Status HandleCustomCall(const HloInstruction* custom_call) override;
Ddfs_hlo_visitor_with_default.h145 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall() function
Dhlo_evaluator.h241 Status HandleCustomCall(HloInstruction* custom_call) override;
Dhlo_verifier.h77 Status HandleCustomCall(HloInstruction*) override;
Ddfs_hlo_visitor.h236 virtual Status HandleCustomCall(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc643 Status HloCostAnalysis::HandleCustomCall(const HloInstruction*) { in HandleCustomCall() function in xla::HloCostAnalysis
Dhlo_verifier.cc544 Status ShapeVerifier::HandleCustomCall(HloInstruction* instruction) { in HandleCustomCall() function in xla::ShapeVerifier
Dhlo_evaluator.cc1576 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::HloEvaluator
Dhlo_instruction.cc2640 return visitor->HandleCustomCall(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h177 Status HandleCustomCall(HloInstruction* custom_call) override;
Dir_emitter.cc2229 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall() function in xla::cpu::IrEmitter