Home
last modified time | relevance | path

Searched defs:dilation_h (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Ddepthwise_conv2d_tests.cc27 … int pad_r, int dilation_h, int dilation_w, ActType act_type, int input_channel) { in CreateParameter()
56 int dilation_h = 1; in TEST_F() local
102 int dilation_h = 1; in TEST_F() local
149 int dilation_h = 1; in TEST_F() local
193 int dilation_h = 1; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c29 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackDwUnitFp32() local
79 const int dilation_h = conv_param->dilation_h_; in rolling_im2col_hwc() local
150 const int dilation_h = conv_param->dilation_h_; in rolling_im2row_hwc() local
198 const int dilation_h = conv_param->dilation_h_; in col2im_hwc() local
246 const int dilation_h = conv_param->dilation_h_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c28 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackDwUnitFp16() local
78 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackUnitFp16() local
145 const int dilation_h = conv_param->dilation_h_; in RollingCol2ImPackUnitFp16() local
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dconvolution_depthwise_npu.cc23 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvDwParam() local
Ddeconvolution_npu.cc46 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetDeconvParam() local
Dconvolution_npu.cc42 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvParam() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc267 int dilation_h = conv_param->dilation_h_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore/core/ops/grad/
Dconv2d_backprop_input.cc53 auto dilation_h = dilation[2]; in SetPadList() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc52 auto dilation_h = ori_conv_prim->get_dilation().at(kIndexH); in GetSplitPadList() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c38 int dilation_h = conv_param->dilation_h_; in Im2ColPackUnitFp32() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c749 int dilation_h = conv_param->dilation_h_; in Im2ColPackUnitInt8Opt() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c50 int dilation_h = conv_param->dilation_h_; in Im2ColPackUnitFp16() local