Home
last modified time | relevance | path

Searched refs:kernel_h (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
29 param->kernel_h_ = kernel_h; in CreateParameter()
48 int kernel_h = 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()
94 int kernel_h = 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()
141 int kernel_h = 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.h46 …WConvKernel)(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
57 …dst, const float *src, const float *weight, const float *bias, int height, int width, int kernel_h,
61 …onvWxKKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
67 …nv3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
72 …nv1x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
77 …nv4x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
82 …nv1x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
87 …nv6x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
92 …nv1x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
97 …nv12x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
[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,
67 …iseSWKernel)(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
71 …SW3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
75 …SW1x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
79 …SW4x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
83 …SW1x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
87 …SW4x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
91 …SW1x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
95 …eSW8x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
99 …eSW1x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h,
[all …]
Dconv_common_fp32.c351 int kernel_h = conv_param->kernel_h_; 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()
433 …nv3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, 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()
592 …nv1x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, 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()
687 …nv4x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, 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()
862 …nv1x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, 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 …]
Dcommon_func_fp32.h52 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
60 …er(float *dst, const float *src, const float *weight, size_t height, size_t width, size_t kernel_h,
76 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t ic4,
Dconv_depthwise_fp32.c263 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, in ConvDwCenter() argument
276 for (int kh = 0; kh < kernel_h; kh++) { in ConvDwCenter()
929 …onvDwCenter(float *dst, const float *src, const float *weight, int height, int width, int kernel_h, in DeconvDwCenter() argument
940 for (int kh = 0; kh < kernel_h; kh++) { in DeconvDwCenter()
1066 int kernel_h = conv_param->kernel_h_; 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()
1137 …eSWWxKKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSWWxKKernel() argument
1159 for (int kh = 0; kh < kernel_h; kh++) { in DepthwiseSWWxKKernel()
1198 …SW3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW3x32Kernel() argument
1282 : "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(in_kw_step), // 5 in DepthwiseSW3x32Kernel()
[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…
Dpack_fp32.c35 int kernel_h = conv_param->kernel_h_; in Im2ColPackUnitFp32() local
37 int kernel_plane = kernel_h * kernel_w; in Im2ColPackUnitFp32()
52 int kh_e = MSMIN(kernel_h, UP_DIV(conv_param->input_h_ - input_h, dilation_h)); 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()
242 for (int h = 0; h < kernel_h; ++h) { in SWPackNHWCToNXHWCXFp32()
248 int oc_index = oc * kernel_w * kernel_h * input_channel + ic_index; in SWPackNHWCToNXHWCXFp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c32 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackDwUnitFp32() local
41 const int stride = kernel_h * kernel_w; in RollingIm2ColPackDwUnitFp32()
50 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackDwUnitFp32()
82 const int kernel_h = conv_param->kernel_h_; in rolling_im2col_hwc() local
100 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2col_hwc()
120 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2col_hwc()
153 const int kernel_h = conv_param->kernel_h_; in rolling_im2row_hwc() local
165 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in rolling_im2row_hwc()
201 const int kernel_h = conv_param->kernel_h_; in col2im_hwc() local
219 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in col2im_hwc()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c31 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackDwUnitFp16() local
40 const int stride = kernel_h * kernel_w; in RollingIm2ColPackDwUnitFp16()
49 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackDwUnitFp16()
81 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackUnitFp16() local
99 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackUnitFp16()
119 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingIm2ColPackUnitFp16()
148 const int kernel_h = conv_param->kernel_h_; in RollingCol2ImPackUnitFp16() local
166 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingCol2ImPackUnitFp16()
185 for (kernel_row = 0; kernel_row < kernel_h; kernel_row++) { in RollingCol2ImPackUnitFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c55 int kernel_h = param->kernel_h_ != -1 ? param->kernel_h_ : GetHeight(weight); in Deconv2dInferShape() local
56 MS_CHECK_FALSE(INT_MUL_OVERFLOW(kernel_h, kernel_w), NNACL_ERR); in Deconv2dInferShape()
67 MS_CHECK_FALSE(INT_MUL_OVERFLOW(kernel_h, dilate_h), NNACL_ERR); in Deconv2dInferShape()
72 …output_h = (input_h - 1) * stride_h + ((kernel_h - 1) * dilate_h + 1) - param->pad_u_ - param->pad… in Deconv2dInferShape()
78 output_h = (input_h - 1) * stride_h + kernel_h; in Deconv2dInferShape()
94 param->pad_u_ = ((input_h - 1) * stride_h + (kernel_h - 1) * dilate_h + 1 - output_h) / 2; in Deconv2dInferShape()
110 param->kernel_h_ = kernel_h; in Deconv2dInferShape()
Dconv2d_infer.c21 int kernel_h = param->kernel_h_; in ConvInferShape() local
30 if (INT_MUL_OVERFLOW(kernel_h, dilate_h) || INT_MUL_OVERFLOW(kernel_w, dilate_w)) { in ConvInferShape()
36 int pad_h_all = ((*output_h - 1) * stride_h + (kernel_h - 1) * dilate_h + 1 - input_h); in ConvInferShape()
53 …*output_h = ceil(((float)(input_h) + param->pad_u_ + param->pad_d_ - ((float)(kernel_h)-1) * (floa… in ConvInferShape()
57 int kernel_height = (kernel_h - 1) * dilate_h + 1; in ConvInferShape()
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc58 …r<int64_t> GetOutputShape(const std::vector<int64_t> &in_shape, int64_t kernel_d, int64_t kernel_h, in GetOutputShape() argument
71 …static_cast<int64_t>(std::floor((in_h + pad_list[2] + pad_list[3] - kernel_h + stride_h - 1) / str… in GetOutputShape()
85 …out_h = static_cast<int64_t>(std::floor((in_h + pad_list[2] + pad_list[3] - kernel_h) / stride_h +… in GetOutputShape()
92 void GetPadsByPadding(int64_t in_d, int64_t in_h, int64_t in_w, int64_t kernel_d, int64_t kernel_h,… in GetPadsByPadding() argument
105 int64_t pad_h = std::max((tail_h > 0 ? kernel_h - tail_h : kernel_h - stride_h), (int64_t)0); in GetPadsByPadding()
138 auto kernel_h = kernel_size[1]; 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/lite/src/runtime/kernel/arm/fp32/
Dconvolution_slidewindow_fp32.cc44 int kernel_h = filter_tensor->Height(); in Init() local
46 int kernel_plane = kernel_h * kernel_w; in Init()
192 int kernel_h = filter_tensor->Height(); in PackWeight() local
197 PackNHWCToNXHWCXFp32(kernel_h, kernel_w, output_channel, oc_block_num, input_channel, in PackWeight()
205 int kernel_h = filter_tensor->Height(); in MallocWeightBiasData() local
209 int kernel_plane = kernel_h * kernel_w; in MallocWeightBiasData()
Dadder_fp32.cc69 int kernel_h = filter_tensor->Height(); 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/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c46 int kernel_h = conv_param->kernel_h_; in SelectOutputUnit() local
62 float common_cost = (float)out_h * out_w * in_c * out_c * kernel_h * kernel_w; in SelectOutputUnit()
69 float penalty = ((float)input_unit * input_unit) / ((float)kernel_h * kernel_w) * 0.12f; in SelectOutputUnit()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc63 int kernel_h = conv_param->kernel_h_; in TEST_F() local
79 int kernel_plane = kernel_h * kernel_w; in TEST_F()
126 int kernel_h = conv_param->kernel_h_; 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.cc60 auto kernel_h = weight_tensor->Height(); in PackWeight() local
65 input_channel, kernel_w * kernel_h, output_channel); in PackWeight()
72 auto kernel_h = weight_tensor->Height(); in MallocWeightBiasData() local
74 …size_t weight_pack_size = input_channel * kernel_w * kernel_h * UP_ROUND(output_channel, C8NUM) * … in MallocWeightBiasData()
194 auto kernel_h = weight_tensor->Height(); 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.cc89 int32_t kernel_h = filter_tensor_->Height(); in InitWeightBias() local
153 …code.CodeFunctionWithCheck("ConvInit", filter_tensor_, bias_tensor_, filter_zp_str, kernel_h, kern… in InitWeightBias()
157 …code.CodeFunctionWithCheck("ConvInit", filter_tensor_, bias_tensor_, filter_zp_str, kernel_h, kern… in InitWeightBias()
263 int kernel_h = conv_param->kernel_h_; 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/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()
182 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
211 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
232 …er(float *dst, const float *src, const float *weight, size_t height, size_t width, size_t kernel_h, in DeconvDwFp32Center() argument
251 for (int kh = 0; kh < kernel_h; kh++) { in DeconvDwFp32Center()
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv_init_int8_wrapper.c22 …Init(int8_t *origin_weight, const int32_t *ori_bias, const int32_t *filter_quant_zps, int kernel_h, in ConvInit() argument
27 int kernel_plane = kernel_h * kernel_w; in ConvInit()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.h33 … size_t height, size_t width, size_t kernel_h, size_t kernel_w, size_t out_h_step,
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/micro/coder/opcoders/nnacl/fp32/
Dconvolution_fp32_coder.cc59 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
65 int kernel_plane = kernel_h * kernel_w; in InitWeightBias()
/third_party/mindspore/mindspore/core/ops/fusion/
Davg_pool_fusion.cc76 auto kernel_h = kernel_size[2]; in InferShape() local
83 out_h = static_cast<int64_t>(ceil((in_h - (kernel_h - 1)) / static_cast<float>(stride_h))); in InferShape()
Dmax_pool_fusion.cc73 auto kernel_h = kernel_size[2]; in InferShape() local
80 out_h = static_cast<int64_t>(ceil((in_h - (kernel_h - 1)) / static_cast<float>(stride_h))); in InferShape()

12