Home
last modified time | relevance | path

Searched refs:dilation_w (Results 1 – 16 of 16) 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() argument
41 param->dilation_w_ = dilation_w; in CreateParameter()
57 int dilation_w = 1; in TEST_F() local
85 dilation_w, act_type, input_shape.back()); in TEST_F()
103 int dilation_w = 1; in TEST_F() local
132 dilation_w, act_type, input_shape.back()); in TEST_F()
150 int dilation_w = 1; in TEST_F() local
177 dilation_w, act_type, input_shape.back()); in TEST_F()
194 int dilation_w = 1; in TEST_F() local
246 dilation_w, act_type, input_shape.back()); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c30 const int dilation_w = conv_param->dilation_w_; in RollingIm2ColPackDwUnitFp32() local
53 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackDwUnitFp32()
80 const int dilation_w = conv_param->dilation_w_; in rolling_im2col_hwc() local
103 int input_col = -pad_left + kernel_col * dilation_w + input_w; in rolling_im2col_hwc()
123 int input_col = -pad_left + kernel_col * dilation_w + input_w; in rolling_im2col_hwc()
151 const int dilation_w = conv_param->dilation_w_; in rolling_im2row_hwc() local
174 int input_col = -pad_left + kernel_col * dilation_w; in rolling_im2row_hwc()
199 const int dilation_w = conv_param->dilation_w_; in col2im_hwc() local
222 int input_col = -pad_left + kernel_col * dilation_w + col_stride_offset; in col2im_hwc()
247 const int dilation_w = conv_param->dilation_w_; in rolling_col2im_hwc() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c29 const int dilation_w = conv_param->dilation_w_; in RollingIm2ColPackDwUnitFp16() local
52 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackDwUnitFp16()
79 const int dilation_w = conv_param->dilation_w_; in RollingIm2ColPackUnitFp16() local
102 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackUnitFp16()
122 int input_col = -pad_left + kernel_col * dilation_w + input_w; in RollingIm2ColPackUnitFp16()
146 const int dilation_w = conv_param->dilation_w_; in RollingCol2ImPackUnitFp16() local
169 int input_col = -pad_left + kernel_col * dilation_w + col_stride_offset; in RollingCol2ImPackUnitFp16()
188 int input_col = -pad_left + kernel_col * dilation_w + col_stride_offset; in RollingCol2ImPackUnitFp16()
/third_party/mindspore/mindspore/ops/operations/
D_thor_ops.py499 dilation_w = self.dilation[3]
502 w_out = math.ceil((x_shape[3] - dilation_w * (kernel_size_w - 1)) / stride_w)
510 … pad_needed_w = max(0, (w_out - 1) * stride_w + dilation_w * (kernel_size_w - 1) + 1 - x_shape[3])
516 …1 + (x_shape[3] + 2 * self.pad - kernel_size_w - (kernel_size_w - 1) * (dilation_w - 1)) / stride_w
579 dilation_w = self.dilations
582 w_out = math.ceil((x_shape[3] - dilation_w * (kernel_size_w - 1)) / stride_w)
590 … pad_needed_w = max(0, (w_out - 1) * stride_w + dilation_w * (kernel_size_w - 1) + 1 - x_shape[3])
Dnn_ops.py1570 _, _, dilation_h, dilation_w = self.dilation
1575 w_out = math.ceil((x_shape[3] - dilation_w * (kernel_size_w - 1)) / stride_w)
1585 … pad_needed_w = max(0, (w_out - 1) * stride_w + dilation_w * (kernel_size_w - 1) + 1 - x_shape[3])
1593 … 1 + (x_shape[3] + pad_left + pad_right - kernel_size_w - (kernel_size_w - 1) * (dilation_w - 1)) \
8375 dilation_w = self.dilation[4]
8380 w_out = math.ceil((x_shape[4] - dilation_w * (kernel_size_w - 1)) / stride_w)
8396 … pad_needed_w = max(0, (w_out - 1) * stride_w + dilation_w * (kernel_size_w - 1) + 1 - x_shape[4])
8407 * (dilation_w - 1)) / stride_w
8415 filter_w = (self.kernel_size[2] - 1) * dilation_w + 1
8567 dilation_w = self.dilation[4]
[all …]
D_grad_ops.py402 dilation_w = self.dilation[4]
415 …pad_needed_w = max(0, (dout_shape[4] - 1) * stride_w + dilation_w * (kernel_w - 1) + 1 - x_shape[4…
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dconvolution_depthwise_npu.cc24 auto dilation_w = static_cast<int>(*(conv_prim->dilation()->begin() + 1)); in SetConvDwParam() local
26 conv_dw_->set_attr_dilations(ge::AttrValue::LIST_INT({dilation_h, dilation_w})); in SetConvDwParam()
Ddeconvolution_npu.cc47 auto dilation_w = static_cast<int>(*(conv_prim->dilation()->begin() + 1)); in SetDeconvParam() local
49 deconv_->set_attr_dilations(ge::AttrValue::LIST_INT({dilation_h, dilation_w})); in SetDeconvParam()
Dconvolution_npu.cc43 auto dilation_w = static_cast<int>(*(conv_prim->dilation()->begin() + 1)); in SetConvParam() local
45 conv_->set_attr_dilations(ge::AttrValue::LIST_INT({dilation_h, dilation_w})); in SetConvParam()
/third_party/mindspore/tests/vm_impl/
Dvm_me.py258 dilation_w = dilation[1]
274 …out_w = 1 + int((x_w + pad_left + pad_right - filter_w - (filter_w - 1) * (dilation_w - 1)) / stri…
365 dilation_w = dilation
368 dilation_w = dilation[1]
371 dilation_w = dilation[3]
389 …out_w = (width + pad_left + pad_right - filter_w - (filter_w - 1) * (dilation_w - 1)) // stride_w …
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c51 int dilation_w = conv_param->dilation_w_; in Im2ColPackUnitFp16() local
64 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitFp16()
65 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitFp16()
66 if (dilation_h == 1 && dilation_w == 1) { in Im2ColPackUnitFp16()
78 int input_x_stride = input_y_stride + n * dilation_w * in_channel; in Im2ColPackUnitFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpack_fp32.c39 int dilation_w = conv_param->dilation_w_; in Im2ColPackUnitFp32() local
41 if (dilation_h == 0 || dilation_w == 0 || out_w == 0) { in Im2ColPackUnitFp32()
53 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitFp32()
54 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitFp32()
55 if (dilation_w == 1 && dilation_h == 1) { in Im2ColPackUnitFp32()
67 int input_x_stride = input_y_stride + k * dilation_w * in_channel; in Im2ColPackUnitFp32()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc268 int dilation_w = conv_param->dilation_w_; in CPUConv2DINT8CoderCreator() local
271 …_h == 3 && kernel_w == 3 && stride_h == 1 && stride_w == 1 && dilation_h == 1 && dilation_w == 1) { in CPUConv2DINT8CoderCreator()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpack_int8.c750 int dilation_w = conv_param->dilation_w_; in Im2ColPackUnitInt8Opt() local
758 NNACL_CHECK_ZERO_RETURN(dilation_w); in Im2ColPackUnitInt8Opt()
766 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitInt8Opt()
767 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitInt8Opt()
768 if (dilation_w == 1 && dilation_h == 1) { in Im2ColPackUnitInt8Opt()
779 int input_x_stride = input_y_stride + k * dilation_w * in_channel; in Im2ColPackUnitInt8Opt()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc54 auto dilation_w = ori_conv_prim->get_dilation().at(kIndexW); in GetSplitPadList() local
56 INT_MUL_OVERFLOW_THRESHOLD((kernel_w - 1), dilation_w, INT64_MAX)) { in GetSplitPadList()
65 …(output_w - 1) * ori_conv_prim->get_stride().at(kIndexW) + (kernel_w - 1) * dilation_w + 1 - input… in GetSplitPadList()
/third_party/mindspore/mindspore/core/ops/grad/
Dconv2d_backprop_input.cc54 auto dilation_w = dilation[3]; in SetPadList() local
70 …(dout_shape_norm[kInputIndex3] - 1) * stride_w + dilation_w * (kernel_w - 1) + 1 - x_size_v[kInput… in SetPadList()