Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h77 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc531 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::gpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc52 Status HandleConvolution(HloInstruction* convolution) override;
421 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::__anon8e9cd5000111::ConvolutionVisitor
Dhlo_cost_analysis.h113 Status HandleConvolution(const HloInstruction* convolution) override;
Ddfs_hlo_visitor_with_default.h84 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution() function
Dalgebraic_simplifier.h283 Status HandleConvolution(HloInstruction* convolution) override;
Ddfs_hlo_visitor.h112 virtual Status HandleConvolution(HloInstructionPtr hlo) = 0;
Dhlo_verifier.h151 Status HandleConvolution(HloInstruction* convolution) override;
Ddynamic_dimension_inference.cc133 Status HandleConvolution(HloInstruction* hlo) override;
590 Status DynamicDimensionInferenceVisitor::HandleConvolution( in HandleConvolution() function in xla::DynamicDimensionInferenceVisitor
Dhlo_cost_analysis.cc711 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution() function in xla::HloCostAnalysis
Dhlo_verifier.cc210 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::ShapeVerifier
Dhlo_evaluator_typed_visitor.h1084 Status HandleConvolution(HloInstruction* conv) override {
Dhlo_instruction.cc3544 return visitor->HandleConvolution(this); in Visit()
Dalgebraic_simplifier.cc6664 Status AlgebraicSimplifierVisitor::HandleConvolution( in HandleConvolution() function in xla::AlgebraicSimplifierVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h148 Status HandleConvolution(HloInstruction* convolution) override;
Dir_emitter.cc905 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution() function in xla::cpu::IrEmitter
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h494 Status HandleConvolution(HloInstruction* hlo) override;
Dconvolution_handler.cc989 Status SpmdPartitioningVisitor::HandleConvolution(HloInstruction* hlo) { in HandleConvolution() function in xla::spmd::SpmdPartitioningVisitor