Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h81 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter_unnested.h173 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc549 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitter
Dir_emitter_unnested.cc806 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitterUnnested
809 return IrEmitter::HandleConvolution(convolution); in HandleConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc53 Status HandleConvolution(HloInstruction* convolution) override;
416 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::__anon8a0b6e710111::ConvolutionVisitor
Dhlo_cost_analysis.h75 Status HandleConvolution(const HloInstruction* convolution) override;
Ddfs_hlo_visitor_with_default.h89 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution() function
Dhlo_verifier.h55 Status HandleConvolution(HloInstruction* convolution) override;
Ddfs_hlo_visitor.h115 virtual Status HandleConvolution(HloInstructionPtr hlo) = 0;
Ddynamic_dimension_inference.cc119 Status HandleConvolution(HloInstruction* hlo) override;
560 Status DynamicDimensionInferenceVisitor::HandleConvolution( in HandleConvolution() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc569 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution() function in xla::HloCostAnalysis
Dhlo_verifier.cc150 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::ShapeVerifier
Dalgebraic_simplifier.cc309 Status HandleConvolution(HloInstruction* convolution) override;
5517 Status AlgebraicSimplifierVisitor::HandleConvolution( in HandleConvolution() function in xla::AlgebraicSimplifierVisitor
Dhlo_evaluator_typed_visitor.h1264 Status HandleConvolution(HloInstruction* conv) override {
Dhlo_instruction.cc3127 return visitor->HandleConvolution(this); in Visit()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h140 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc870 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h431 Status HandleConvolution(HloInstruction* hlo) override;
Dconvolution_handler.cc987 Status SpmdPartitioningVisitor::HandleConvolution(HloInstruction* hlo) { in HandleConvolution() function in xla::spmd::SpmdPartitioningVisitor