Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding.cc57 const HloInstruction& convolution, in CanFoldOperandsIntoConvolution()
120 auto& convolution = *pair.first; in FoldTransposeIntoConvolution() local
Dtranspose_folding_test.cc53 const TransposeFolding::OperandIndices& candidate_operands) { in FoldTranspose()
102 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
130 const TransposeFolding::OperandIndices& candidate_operands) { in TEST_F()
Dconvolution_group_converter.cc201 Status ConvolutionVisitor::HandleBatchGroupCount(HloInstruction* convolution) { in HandleBatchGroupCount()
319 Status ConvolutionVisitor::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dhlo_instruction.cc3177 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config() local
3187 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config() local
3440 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in convolution_dimension_numbers() local
3451 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in set_convolution_dimension_numbers() local
3461 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in feature_group_count() local
3473 if (auto convolution = DynCast<HloConvolutionInstruction>(this)) { in batch_group_count() local
Ddfs_hlo_visitor_with_default.h88 Status HandleConvolution(HloInstructionPtr convolution) override { in HandleConvolution()
Dalgebraic_simplifier.cc3394 HloInstruction* convolution) { in FoldConvInputPad()
3461 HloInstruction* convolution) { in FoldConvFilterPad()
3527 HloInstruction* convolution) { in SimplifyConvToDot()
3642 HloInstruction* convolution) { in HandleConvolution()
Dhlo_cost_analysis.cc434 Status HloCostAnalysis::HandleConvolution(const HloInstruction* convolution) { in HandleConvolution()
Dhlo_verifier.cc154 Status ShapeVerifier::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dhlo_parser_test.cc2468 auto* convolution = local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc44 const HloInstruction& convolution, in PotentiallyImplementedAsEigenConvolution()
Dir_emitter.cc951 const HloConvolutionInstruction* convolution, in EmitElementalConvolution()
1093 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.cc637 Status IrEmitter::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
Dir_emitter_unnested.cc367 Status IrEmitterUnnested::HandleConvolution(HloInstruction* convolution) { in HandleConvolution()
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py918 def convolution(inputs, function
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py760 def convolution( function