Searched refs:in_kw_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 | 159 …sliding->in_kw_step_ = sliding->ic_align_ * conv_param->dilation_w_; // ke… in AppendSlidingParamConv()
174 …sliding->in_kw_step_ = sliding->block_channel_ * conv_param->dilation_w_; … in AppendSlidingParamConvDw()
227 …onst float *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; in ConvDwBorder()
241 param->in_kw_step = sliding->in_kw_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()
338 sliding->in_kw_step_ * sizeof(float), relu, relu6); 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 …st float *src_kernel = src_w + start_kh * sw_param->in_kh_step_ + start_kw * sw_param->in_kw_step_; in SWBorder()
324 …oc_block, sw_param->out_block_step_, sw_param->ic_align_, sw_param->in_kw_step_, sw_param->in_kh_s… in SWBorder()
355 int in_kw_step = sw_param->in_kw_step_; in ConvSWFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 503 … float16_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; 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()
605 sliding->in_kw_step_ * sizeof(float16_t), relu, relu6); 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()
766 sliding->in_kw_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 | 78 int in_kw_step_; // kernel W member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 427 int in_kw_step = sliding->in_kw_step_; in ConvDw3x3Int8Pad()
554 …nst int8_t *src_kernel = src_w + start_kh * sliding->in_kh_step_ + start_kw * sliding->in_kw_step_; 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()
806 sliding->in_kw_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 | 94 sliding_->in_kw_step_ = C4NUM * conv_param_->dilation_w_; // kernel W in InitSlideParam()
|