Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h65 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.h77 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc600 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc312 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitterUnnested
314 return IrEmitter::HandleConvolution(convolution); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.h43 Status HandleConvolution(HloInstruction* convolution) override;
Dhlo_cost_analysis.h69 Status HandleConvolution(const HloInstruction* convolution) override;
Ddfs_hlo_visitor_with_default.h85 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution() function
Ddfs_hlo_visitor.h105 virtual Status HandleConvolution(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc381 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution() function in xla::HloCostAnalysis
Dhlo_verifier.cc75 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc141 Status HandleConvolution(HloInstruction* convolution) override;
1851 Status AlgebraicSimplifierVisitor::HandleConvolution( in HandleConvolution() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator.cc887 Status HandleConvolution(HloInstruction* conv) override { in HandleConvolution() function in xla::HloEvaluator::TypedVisitor
Dhlo_instruction.cc2587 return visitor->HandleConvolution(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h126 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc846 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::cpu::IrEmitter