Searched refs:in_kh_step_ (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 158 …sliding->in_kh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->dilation_h_; // ke… in AppendSlidingParamConv()
173 …sliding->in_kh_step_ = conv_param->input_w_ * sliding->block_channel_ * conv_param->dilation_h_; … in AppendSlidingParamConvDw()
227 …const float *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step… in ConvDwBorder()
240 param->in_kh_step = sliding->in_kh_step_ * sizeof(float); in ConvDwBorder()
249 sliding->in_kh_step_ * sizeof(float), sliding->in_kw_step_ * sizeof(float), in ConvDwBorder()
253 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_ * C4NUM, relu, relu6); in ConvDwBorder()
337 sliding->in_sw_step_ * sizeof(float), sliding->in_kh_step_ * sizeof(float), in ConvDwSWFp32()
342 … sliding->in_sh_step_, sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_, relu, in ConvDwSWFp32()
913 float *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDwBorder()
916 sliding->in_kh_step_ * sizeof(float), sliding->in_kw_step_ * sizeof(float), in DeconvDwBorder()
[all …]
|
| D | conv_common_fp32.c | 320 …const float *src_kernel = src_w + start_kh * sw_param->in_kh_step_ + start_kw * sw_param->in_kw_st… in SWBorder()
324 …lock, sw_param->out_block_step_, sw_param->ic_align_, sw_param->in_kw_step_, sw_param->in_kh_step_, in SWBorder()
356 int in_kh_step = sw_param->in_kh_step_; in ConvSWFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 503 …const float16_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_… in DepthwiseBorderFp16()
507 … sliding->in_kh_step_ * sizeof(float16_t), sliding->in_kw_step_ * sizeof(float16_t), in DepthwiseBorderFp16()
511 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_ * C8NUM, relu, relu6); in DepthwiseBorderFp16()
604 … sliding->in_sw_step_ * sizeof(float16_t), sliding->in_kh_step_ * sizeof(float16_t), in ConvDwC8Fp16()
610 sliding->in_kh_step_, sliding->in_kw_step_, relu, relu6); in ConvDwC8Fp16()
662 … float16_t *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDepthwiseBorderFp16()
665 … sliding->in_kh_step_ * sizeof(float16_t), sliding->in_kw_step_ * sizeof(float16_t), in DeconvDepthwiseBorderFp16()
669 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_ * C8NUM); in DeconvDepthwiseBorderFp16()
765 … sliding->in_sw_step_ * sizeof(float16_t), sliding->in_kh_step_ * sizeof(float16_t), in DeconvDwC8Fp16()
771 sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_); in DeconvDwC8Fp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 77 int in_kh_step_; // kernel H member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 426 int in_kh_step = sliding->in_kh_step_; in ConvDw3x3Int8Pad()
554 …const int8_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_ste… in ConvDwInt8Border()
558 … sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_, in_zp, out_zp, in ConvDwInt8Border()
659 … sliding->in_sh_step_, sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_, in_zp, in ConvDwInt8SW()
707 … int32_t *dst_kernel = dst_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in DeconvDwInt8Border()
710 sliding->in_kh_step_, sliding->in_kw_step_, conv_param->kernel_w_); in DeconvDwInt8Border()
805 … sliding->in_sw_step_ * sizeof(int32_t), sliding->in_kh_step_ * sizeof(int32_t), in DeconvDwInt8()
810 … sliding->in_sh_step_, sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_); in DeconvDwInt8()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_depthwise_int8.cc | 93 sliding_->in_kh_step_ = conv_param_->input_w_ * C4NUM * conv_param_->dilation_h_; // kernel H in InitSlideParam()
|