Searched refs:start_kh (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 413 int start_kh = MSMAX(0, UP_DIV(-ih_origin, conv_param->dilation_h_)); in ConvDwFp16() local
419 for (int kh = start_kh; kh < end_kh; kh++) { in ConvDwFp16()
492 int start_kh = MSMAX(0, UP_DIV(-ih, conv_param->dilation_h_)); in DepthwiseBorderFp16() local
503 …const float16_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_… in DepthwiseBorderFp16()
504 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DepthwiseBorderFp16()
506 …ConvDwFp16Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in DepthwiseBorderFp16()
510 …DepthwiseBorderPixelFp16(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - … in DepthwiseBorderFp16()
650 int start_kh = MSMAX(0, UP_DIV(-oh, conv_param->dilation_h_)); in DeconvDepthwiseBorderFp16() local
661 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DeconvDepthwiseBorderFp16()
662 … float16_t *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDepthwiseBorderFp16()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 52 int start_kh = MSMAX(0, UP_DIV(-ih_origin, conv_param->dilation_h_)); in ConvDw() local
59 for (int kh = start_kh; kh < end_kh; kh++) { in ConvDw()
216 int start_kh = MSMAX(0, UP_DIV(-ih, conv_param->dilation_h_)); in ConvDwBorder() local
227 …const float *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step… in ConvDwBorder()
228 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in ConvDwBorder()
238 param->height = end_kh - start_kh; in ConvDwBorder()
248 …ConvDwFp32Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in ConvDwBorder()
252 …ConvDwBorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_k… in ConvDwBorder()
901 int start_kh = MSMAX(0, UP_DIV(-oh, conv_param->dilation_h_)); in DeconvDwBorder() local
912 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwBorder()
[all …]
|
| D | conv_common_fp32.c | 311 int start_kh = MSMAX(0, UP_DIV(-ih, conv_param->dilation_h_)); in SWBorder() local
320 …const float *src_kernel = src_w + start_kh * sw_param->in_kh_step_ + start_kw * sw_param->in_kw_st… in SWBorder()
322 … weight + (start_kh * conv_param->kernel_w_ + start_kw) * sw_param->ic_align_ * C8NUM * oc_block; in SWBorder()
323 …kernel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, act_type… in SWBorder()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 103 int start_kh = MSMAX(0, UP_DIV(-ih_origin, conv_param->dilation_h_)); in ConvDwInt8() local
110 for (int kh = start_kh; kh < end_kh; kh++) { in ConvDwInt8()
543 int start_kh = MSMAX(0, UP_DIV(-ih, conv_param->dilation_h_)); in ConvDwInt8Border() local
554 …const int8_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_ste… in ConvDwInt8Border()
555 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in ConvDwInt8Border()
557 …ConvDwInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - sta… in ConvDwInt8Border()
695 int start_kh = MSMAX(0, UP_DIV(-oh, conv_param->dilation_h_)); in DeconvDwInt8Border() local
706 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwInt8Border()
707 … int32_t *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDwInt8Border()
709 …DeconvDwInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_k… in DeconvDwInt8Border()
|