Home
last modified time | relevance | path

Searched refs:HandleInfeed (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h85 Status HandleInfeed(HloInstruction* infeed) override;
Dir_emitter_unnested.h173 Status HandleInfeed(HloInstruction* xla_infeed) override;
Dir_emitter.cc748 Status IrEmitter::HandleInfeed(HloInstruction*) { in HandleInfeed() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc1473 Status IrEmitterUnnested::HandleInfeed(HloInstruction* infeed) { in HandleInfeed() function in xla::gpu::IrEmitterUnnested
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.h80 Status HandleInfeed(const HloInstruction* infeed) override;
Ddfs_hlo_visitor_with_default.h115 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() function
Dhlo_evaluator.h259 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed() function
Dhlo_verifier.h62 Status HandleInfeed(HloInstruction*) override;
Ddfs_hlo_visitor.h220 virtual Status HandleInfeed(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc255 Status HloCostAnalysis::HandleInfeed(const HloInstruction*) { in HandleInfeed() function in xla::HloCostAnalysis
Dhlo_verifier.cc246 Status ShapeVerifier::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::ShapeVerifier
Dhlo_instruction.cc2626 return visitor->HandleInfeed(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h158 Status HandleInfeed(HloInstruction* infeed) override;
Dir_emitter.cc325 Status IrEmitter::HandleInfeed(HloInstruction* instruction) { in HandleInfeed() function in xla::cpu::IrEmitter