Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc51 Status HandleConvolution(HloInstruction* convolution) override;
319 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::__anon9c1c05530111::ConvolutionVisitor
Ddynamic_dimension_inference.cc62 Status HandleConvolution(HloInstruction* hlo) override;
261 Status DynamicDimensionInferenceVisitor::HandleConvolution( in HandleConvolution() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.h72 Status HandleConvolution(const HloInstruction* convolution) override;
Ddfs_hlo_visitor_with_default.h88 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution() function
Dhlo_verifier.h53 Status HandleConvolution(HloInstruction* convolution) override;
Ddfs_hlo_visitor.h112 virtual Status HandleConvolution(HloInstructionPtr hlo) = 0;
Dhlo_cost_analysis.cc434 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution() function in xla::HloCostAnalysis
Dhlo_verifier.cc154 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc202 Status HandleConvolution(HloInstruction* convolution) override;
3641 Status AlgebraicSimplifierVisitor::HandleConvolution( in HandleConvolution() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h1009 Status HandleConvolution(HloInstruction* conv) override { in HandleConvolution() function
Dhlo_instruction.cc2546 return visitor->HandleConvolution(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h82 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter_unnested.h163 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc637 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc367 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitterUnnested
370 return IrEmitter::HandleConvolution(convolution); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h155 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc1093 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::cpu::IrEmitter