Home
last modified time | relevance | path

Searched refs:start_kw (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_depthwise_fp32.c223 int start_kw = MSMAX(0, UP_DIV(-iw, conv_param->dilation_w_)); 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()
239 param->width = end_kw - start_kw; in ConvDwBorder()
248 …ConvDwFp32Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in ConvDwBorder()
252 …onvDwBorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in ConvDwBorder()
908 int start_kw = MSMAX(0, UP_DIV(-ow, conv_param->dilation_w_)); in DeconvDwBorder() local
912 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwBorder()
913 float *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDwBorder()
915 … DeconvDwFp32Border(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDwBorder()
[all …]
Dconv_common_fp32.c317 int start_kw = MSMAX(0, UP_DIV(-iw, conv_param->dilation_w_)); 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()
326 … (conv_param->kernel_w_ - end_kw + start_kw) * C8NUM * oc_block * sw_param->ic_align_, write_mode); in SWBorder()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.c499 int start_kw = MSMAX(0, UP_DIV(-iw, conv_param->dilation_w_)); 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 …eBorderPixelFp16(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in DepthwiseBorderFp16()
657 int start_kw = MSMAX(0, UP_DIV(-ow, conv_param->dilation_w_)); 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()
664 … DeconvDwFp16Border(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDepthwiseBorderFp16()
668 …pthwiseBorderPixelFp16(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDepthwiseBorderFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.c550 int start_kw = MSMAX(0, UP_DIV(-iw, conv_param->dilation_w_)); 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 …wInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in ConvDwInt8Border()
702 int start_kw = MSMAX(0, UP_DIV(-ow, conv_param->dilation_w_)); 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 …econvDwInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDwInt8Border()