Searched refs:in_sh_step_ (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 75 int in_sh_step_; // stride H member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 156 …sliding->in_sh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->stride_h_; // st… in AppendSlidingParamConv()
171 …sliding->in_sh_step_ = conv_param->input_w_ * sliding->block_channel_ * conv_param->stride_h_; … in AppendSlidingParamConvDw()
336 … sliding->block_channel_ * sizeof(float), sliding->in_sh_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()
1007 … sliding->block_channel_ * sizeof(float), sliding->in_sh_step_ * sizeof(float), in DeconvDwSWFp32()
1013 … sliding->in_sh_step_, sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_); in DeconvDwSWFp32()
1076 int in_sh_step = sw_param->in_sh_step_; in DepthwiseSWAvxFp32()
|
| D | conv_common_fp32.c | 357 int in_sh_step = sw_param->in_sh_step_; in ConvSWFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 603 … sliding->block_channel_ * sizeof(float16_t), sliding->in_sh_step_ * sizeof(float16_t), in ConvDwC8Fp16()
609 … sliding->out_h_step_, sliding->block_channel_, sliding->in_sh_step_, sliding->in_sw_step_, in ConvDwC8Fp16()
764 … sliding->block_channel_ * sizeof(float16_t), sliding->in_sh_step_ * sizeof(float16_t), in DeconvDwC8Fp16()
770 … sliding->out_h_step_, sliding->block_channel_, sliding->in_sh_step_, in DeconvDwC8Fp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_depthwise_int8.cc | 91 sliding_->in_sh_step_ = conv_param_->input_w_ * C4NUM * conv_param_->stride_h_; // stride H in InitSlideParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 659 … sliding->in_sh_step_, sliding->in_sw_step_, sliding->in_kh_step_, sliding->in_kw_step_, in_zp, in ConvDwInt8SW()
804 … sliding->block_channel_ * sizeof(int16_t), sliding->in_sh_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()
|