Home
last modified time | relevance | path

Searched defs:convolution (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Ddepthwise_convolution_converter.cc101 HloInstruction* convolution) { in HandleBackwardFilterBatchGroupConvolution()
192 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dtranspose_folding.cc57 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()
120 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
Dtranspose_folding_test.cc52 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()
101 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
129 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
Dconvolution_group_converter.cc200 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()
411 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dhlo_instruction.cc3507 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local
3517 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local
3776 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local
3787 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local
3797 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local
3804 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_feature_group_count() local
3812 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local
3819 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_batch_group_count() local
Ddfs_hlo_visitor_with_default.h89 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
Dalgebraic_simplifier.cc4099 HloInstruction* convolution) { in FoldConvInputPad()
4166 HloInstruction* convolution) { in FoldConvFilterPad()
4232 HloInstruction* convolution) { in SimplifyConvToDot()
4350 HloInstruction* convolution) { in HandleConvolution()
Dhlo_cost_analysis.cc532 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
Dhlo_verifier.cc146 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dhlo_parser_test.cc2639 auto* convolution = local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc44 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
Dir_emitter.cc1011 const HloConvolutionInstruction* convolution, in EmitElementalConvolution()
1156 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
/external/eigen/bench/tensors/
Dtensor_benchmarks.h417 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc674 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dir_emitter_unnested.cc362 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py764 def convolution( function