Home
last modified time | relevance | path

Searched refs:kernel_w (Results 1 – 25 of 45) sorted by relevance

12

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Ddepthwise_conv2d_tests.cc26 OpParameter *CreateParameter(int kernel_h, int kernel_w, int stride_h, int stride_w, int pad_u, int… in CreateParameter() argument
30 param->kernel_w_ = kernel_w; in CreateParameter()
49 int kernel_w = 3; in TEST_F() local
62 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F()
84 …auto *param = CreateParameter(kernel_h, kernel_w, stride_h, stride_w, pad_u, pad_d, pad_l, pad_r, … in TEST_F()
95 int kernel_w = 3; in TEST_F() local
108 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F()
131 …auto *param = CreateParameter(kernel_h, kernel_w, stride_h, stride_w, pad_u, pad_d, pad_l, pad_r, … in TEST_F()
142 int kernel_w = 2; in TEST_F() local
155 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.h47 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
58 int kernel_w, bool is_relu, bool is_relu6, SlidingWindowParam *sw_param);
62 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
68 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
73 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
78 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
83 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
88 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
93 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
98 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
[all …]
Dconv_depthwise_fp32.h29 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step,
68 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
72 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
76 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
80 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
84 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
88 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
92 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
96 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
100 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin,
[all …]
Dcommon_func_fp32.h33 size_t kernel_w; member
52 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
58 … size_t in_kh_step, size_t in_kw_step, size_t kernel_w, size_t relu, size_t relu6);
61 … size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, size_t in_sw_step,
73 size_t in_kh_step, size_t in_kw_step, size_t kernel_w);
76 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t ic4,
Dconv_common_fp32.c352 int kernel_w = conv_param->kernel_w_; in ConvSWFp32() local
418 … dst_oc + ow * out_w_step, src_w, weight, bias, kernel_h, kernel_w, act_type, ow_block, oc_block, in ConvSWFp32()
434 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv3x32Kernel() argument
512 …: "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_step), … in SWConv3x32Kernel()
593 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x32Kernel() argument
641 …: "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_step), … in SWConv1x32Kernel()
688 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x24Kernel() argument
782 …: "r"(src), "r"(weight), "r"(src_3_step), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_s… in SWConv4x24Kernel()
863 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x24Kernel() argument
907 …: "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_step), … in SWConv1x24Kernel()
[all …]
Dconv_depthwise_fp32.c242 param->kernel_w = conv_param->kernel_w_ * C4NUM * sizeof(float); in ConvDwBorder()
263 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, in ConvDwCenter() argument
279 for (int kw = 0; kw < kernel_w; kw++) { in ConvDwCenter()
287 weight_kh += kernel_w * C4NUM; in ConvDwCenter()
930 … int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, int in_kh_step, in DeconvDwCenter() argument
943 for (int kw = 0; kw < kernel_w; kw++) { in DeconvDwCenter()
951 weight_kh += kernel_w * C4NUM; in DeconvDwCenter()
1067 int kernel_w = conv_param->kernel_w_; in DepthwiseSWAvxFp32() local
1121 …dst_w + ow * oc_algin, src_w, weight, bias, kernel_h, kernel_w, act_type, ow_block, oc_block, oc_a… in DepthwiseSWAvxFp32()
1138 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSWWxKKernel() argument
[all …]
Dpack_fp32.c36 int kernel_w = conv_param->kernel_w_; in Im2ColPackUnitFp32() local
37 int kernel_plane = kernel_h * kernel_w; in Im2ColPackUnitFp32()
54 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitFp32()
59 int input_plane_offset = (j * kernel_w + kw_s) * in_channel + i * in_channel * kernel_plane; in Im2ColPackUnitFp32()
68 int input_plane_offset = (j * kernel_w + k) * in_channel + i * in_channel * kernel_plane; in Im2ColPackUnitFp32()
162 void PackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int inp… in PackNHWCToNXHWCXFp32() argument
175 int plane = kernel_w * kernel_h; in PackNHWCToNXHWCXFp32()
234 void SWPackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int i… in SWPackNHWCToNXHWCXFp32() argument
240 int index = i * C8NUM * kernel_h * kernel_w * input_channel; in SWPackNHWCToNXHWCXFp32()
243 for (int w = 0; w < kernel_w; ++w) { in SWPackNHWCToNXHWCXFp32()
[all …]
Dpack_fp32.h64 void PackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int inp…
67 void SWPackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int i…
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c33 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp32() local
41 const int stride = kernel_h * kernel_w; in RollingIm2ColPackDwUnitFp32()
52 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackDwUnitFp32()
83 const int kernel_w = conv_param->kernel_w_; in rolling_im2col_hwc() local
102 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2col_hwc()
122 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2col_hwc()
154 const int kernel_w = conv_param->kernel_w_; in rolling_im2row_hwc() local
166 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in rolling_im2row_hwc()
202 const int kernel_w = conv_param->kernel_w_; in col2im_hwc() local
221 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in col2im_hwc()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c32 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp16() local
40 const int stride = kernel_h * kernel_w; in RollingIm2ColPackDwUnitFp16()
51 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackDwUnitFp16()
82 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackUnitFp16() local
101 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackUnitFp16()
121 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingIm2ColPackUnitFp16()
149 const int kernel_w = conv_param->kernel_w_; in RollingCol2ImPackUnitFp16() local
168 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingCol2ImPackUnitFp16()
187 for (kernel_col = 0; kernel_col < kernel_w; kernel_col++) { in RollingCol2ImPackUnitFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c54 int kernel_w = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(weight); in Deconv2dInferShape() local
56 MS_CHECK_FALSE(INT_MUL_OVERFLOW(kernel_h, kernel_w), NNACL_ERR); in Deconv2dInferShape()
68 MS_CHECK_FALSE(INT_MUL_OVERFLOW(kernel_w, dilate_w), NNACL_ERR); in Deconv2dInferShape()
73 …output_w = (input_w - 1) * stride_w + ((kernel_w - 1) * dilate_w + 1) - param->pad_l_ - param->pad… in Deconv2dInferShape()
79 output_w = (input_w - 1) * stride_w + kernel_w; in Deconv2dInferShape()
95 param->pad_l_ = ((input_w - 1) * stride_w + (kernel_w - 1) * dilate_w + 1 - output_w) / 2; in Deconv2dInferShape()
111 param->kernel_w_ = kernel_w; in Deconv2dInferShape()
Dconv2d_infer.c20 int kernel_w = param->kernel_w_; in ConvInferShape() local
30 if (INT_MUL_OVERFLOW(kernel_h, dilate_h) || INT_MUL_OVERFLOW(kernel_w, dilate_w)) { in ConvInferShape()
37 int pad_w_all = ((*output_w - 1) * stride_w + (kernel_w - 1) * dilate_w + 1 - input_w); in ConvInferShape()
51 …*output_w = ceil(((float)(input_w) + param->pad_l_ + param->pad_r_ - ((float)(kernel_w)-1) * (floa… in ConvInferShape()
56 int kernel_width = (kernel_w - 1) * dilate_w + 1; in ConvInferShape()
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc59 … int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetOutputShape() argument
73 …static_cast<int64_t>(std::floor((in_w + pad_list[4] + pad_list[5] - kernel_w + stride_w - 1) / str… in GetOutputShape()
86 …out_w = static_cast<int64_t>(std::floor((in_w + pad_list[4] + pad_list[5] - kernel_w) / stride_w +… in GetOutputShape()
92 …ing(int64_t in_d, int64_t in_h, int64_t in_w, int64_t kernel_d, int64_t kernel_h, int64_t kernel_w, in GetPadsByPadding() argument
106 int64_t pad_w = std::max((tail_w > 0 ? kernel_w - tail_w : kernel_w - stride_w), (int64_t)0); in GetPadsByPadding()
139 auto kernel_w = kernel_size[2]; in InferShape() local
144 …GetPadsByPadding(in_d, in_h, in_w, kernel_d, kernel_h, kernel_w, stride_d, stride_h, stride_w, pad… in InferShape()
152 …GetOutputShape(in_shape, kernel_d, kernel_h, kernel_w, stride_d, stride_h, stride_w, new_pad_list,… in InferShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
DDepthwiseFp32_Sse.c111 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, in ConvDwFp32Center() argument
136 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
138 for (int kw = 0; kw < kernel_w; kw++, src_kw += in_kw_step, weight_kw += C4NUM) { in ConvDwFp32Center()
182 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
184 for (int kw = 0; kw < kernel_w; kw++, src_kw += in_kw_step, weight_kw += C4NUM) { in ConvDwFp32Center()
211 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
213 for (int kw = 0; kw < kernel_w; kw++, src_kw += in_kw_step, weight_kw += C4NUM) { in ConvDwFp32Center()
233 … size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, size_t in_sw_step, in DeconvDwFp32Center() argument
255 int c4 = DOWN_DIV(kernel_w, C4NUM) * C4NUM; in DeconvDwFp32Center()
256 int c2 = DOWN_DIV(kernel_w, C2NUM) * C2NUM; in DeconvDwFp32Center()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c47 int kernel_w = conv_param->kernel_w_; in SelectOutputUnit() local
62 float common_cost = (float)out_h * out_w * in_c * out_c * kernel_h * kernel_w; in SelectOutputUnit()
65 int input_unit = i + kernel_w - 1; in SelectOutputUnit()
69 float penalty = ((float)input_unit * input_unit) / ((float)kernel_h * kernel_w) * 0.12f; in SelectOutputUnit()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.h30 … size_t height, size_t width, size_t in_kh_step, size_t in_kw_step, size_t kernel_w, size_t relu,
33 … size_t height, size_t width, size_t kernel_h, size_t kernel_w, size_t out_h_step,
37 size_t in_kh_step, size_t in_kw_step, size_t kernel_w);
39 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_slidewindow_fp32.cc45 int kernel_w = filter_tensor->Width(); in Init() local
46 int kernel_plane = kernel_h * kernel_w; in Init()
193 int kernel_w = filter_tensor->Width(); in PackWeight() local
197 PackNHWCToNXHWCXFp32(kernel_h, kernel_w, output_channel, oc_block_num, input_channel, in PackWeight()
206 int kernel_w = filter_tensor->Width(); in MallocWeightBiasData() local
209 int kernel_plane = kernel_h * kernel_w; in MallocWeightBiasData()
Dadder_fp32.cc70 int kernel_w = filter_tensor->Width(); in InitWeightBias() local
75 MS_CHECK_FALSE(INT_MUL_OVERFLOW(kernel_h, kernel_w), RET_ERROR); in InitWeightBias()
76 int kernel_plane = kernel_h * kernel_w; in InitWeightBias()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc64 int kernel_w = conv_param->kernel_w_; in TEST_F() local
79 int kernel_plane = kernel_h * kernel_w; in TEST_F()
127 int kernel_w = conv_param->kernel_w_; in TEST_F() local
142 int kernel_plane = kernel_h * kernel_w; in TEST_F()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc61 auto kernel_w = weight_tensor->Width(); in PackWeight() local
65 input_channel, kernel_w * kernel_h, output_channel); in PackWeight()
73 auto kernel_w = weight_tensor->Width(); in MallocWeightBiasData() local
74 …size_t weight_pack_size = input_channel * kernel_w * kernel_h * UP_ROUND(output_channel, C8NUM) * … in MallocWeightBiasData()
195 auto kernel_w = weight_tensor->Width(); in Init() local
196 …size_t weight_pack_size = input_channel * kernel_w * kernel_h * UP_ROUND(output_channel, C8NUM) * … in Init()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc90 int32_t kernel_w = filter_tensor_->Width(); in InitWeightBias() local
153 ….CodeFunctionWithCheck("ConvInit", filter_tensor_, bias_tensor_, filter_zp_str, kernel_h, kernel_w, in InitWeightBias()
157 ….CodeFunctionWithCheck("ConvInit", filter_tensor_, bias_tensor_, filter_zp_str, kernel_h, kernel_w, in InitWeightBias()
264 int kernel_w = conv_param->kernel_w_; in CPUConv2DINT8CoderCreator() local
271 …if (kernel_h == 3 && kernel_w == 3 && stride_h == 1 && stride_w == 1 && dilation_h == 1 && dilatio… in CPUConv2DINT8CoderCreator()
273 } else if (kernel_h == 1 && kernel_w == 1) { in CPUConv2DINT8CoderCreator()
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv_init_int8_wrapper.c23 … int kernel_w, int input_channel, int output_channel, int32_t input_zp, bool filter_peroc, in ConvInit() argument
27 int kernel_plane = kernel_h * kernel_w; in ConvInit()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconvolution_fp32_coder.cc60 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
65 int kernel_plane = kernel_h * kernel_w; in InitWeightBias()
/third_party/mindspore/mindspore/core/ops/fusion/
Davg_pool_fusion.cc77 auto kernel_w = kernel_size[3]; in InferShape() local
84 out_w = static_cast<int64_t>(ceil((in_w - (kernel_w - 1)) / static_cast<float>(stride_w))); in InferShape()
Dmax_pool_fusion.cc74 auto kernel_w = kernel_size[3]; in InferShape() local
81 out_w = static_cast<int64_t>(ceil((in_w - (kernel_w - 1)) / static_cast<float>(stride_w))); in InferShape()

12