Searched refs:weight_kh (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
| D | DepthwiseFp32_Sse.c | 30 const float *weight_kh = weight; in ConvDwFp32Border() local
35 const float *weight_kw = weight_kh; in ConvDwFp32Border()
93 weight_kh += kernel_w_step; in ConvDwFp32Border()
130 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
136 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
137 const float *src_kw = src_kh, *weight_kw = weight_kh; in ConvDwFp32Center()
178 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
182 for (int kh = 0; kh < kernel_h; kh++, src_kh += in_kh_step, weight_kh += kernel_w * C4NUM) { in ConvDwFp32Center()
183 const float *src_kw = src_kh, *weight_kw = weight_kh; in ConvDwFp32Center()
209 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 114 … const int16_t *weight_kh = weight_data + kh * conv_param->kernel_w_ * conv_param->output_channel_; in ConvDwInt8() local
130 …ConvDwInt8Row(acc_w, src_kw, weight_kh, num_pixels, conv_param->output_channel_, in_sw_step, intpu… in ConvDwInt8()
131 weight_kh += conv_param->output_channel_; in ConvDwInt8()
167 const int16_t *weight_kh = weight; in ConvDw3x3Int8Window() local
170 const int16_t *weight_kw = weight_kh; in ConvDw3x3Int8Window()
179 weight_kh += 3 * channel; in ConvDw3x3Int8Window()
342 const int16_t *weight_kh = weight; in ConvDw3x3Int8BorderPixel() local
345 const int16_t *weight_kw = weight_kh; in ConvDw3x3Int8BorderPixel()
354 weight_kh += 3 * channel; in ConvDw3x3Int8BorderPixel()
509 const int16_t *weight_kh = weight; in ConvDwInt8BorderPixel() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 423 …const float16_t *weight_kh = weight_data + kh * conv_param->kernel_w_ * conv_param->output_channel… in ConvDwFp16() local
438 … ConvDwFp16Row(dst_w, src_kw, weight_kh, num_pixels, conv_param->output_channel_, in_sw_step); in ConvDwFp16()
439 weight_kh += conv_param->output_channel_; in ConvDwFp16()
460 const float16_t *weight_kh = weight; in DepthwiseBorderPixelFp16() local
463 const float16_t *weight_kw = weight_kh; in DepthwiseBorderPixelFp16()
475 weight_kh += kernel_w_step; in DepthwiseBorderPixelFp16()
530 const float16_t *weight_kh = weight; in DepthwiseCenterFp16() local
536 const float16_t *weight_kw = weight_kh; in DepthwiseCenterFp16()
553 weight_kh += kernel_w * C8NUM; in DepthwiseCenterFp16()
625 const float16_t *weight_kh = weight; in DeconvDepthwiseBorderPixelFp16() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 63 … const float *weight_kh = weight_data + kh * conv_param->kernel_w_ * conv_param->output_channel_; in ConvDw() local
79 … ConvDwFp32Row(dst_w, src_kw, weight_kh, num_pixels, conv_param->output_channel_, in_sw_step); in ConvDw()
80 weight_kh += conv_param->output_channel_; in ConvDw()
182 const float *weight_kh = weight; in ConvDwBorderPixel() local
188 const float *weight_kw = weight_kh; in ConvDwBorderPixel()
197 weight_kh += kernel_w_step; in ConvDwBorderPixel()
272 const float *weight_kh = weight; in ConvDwCenter() local
278 const float *weight_kw = weight_kh; in ConvDwCenter()
287 weight_kh += kernel_w * C4NUM; in ConvDwCenter()
869 const float *weight_kh = weight; in DeconvDwBorderPixel() local
[all …]
|