Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dconv_parameter.h67 int ic_align_; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.c319 const float *src_w = src_h + iw * sw_param->ic_align_; in SWBorder()
322 … weight + (start_kh * conv_param->kernel_w_ + start_kw) * sw_param->ic_align_ * C8NUM * oc_block; 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()
326 … (conv_param->kernel_w_ - end_kw + start_kw) * C8NUM * oc_block * sw_param->ic_align_, write_mode); in SWBorder()
353 int ic_algin = sw_param->ic_align_; in ConvSWFp32()
Dconv_depthwise_fp32.c150 sliding->ic_align_ = conv_param->input_channel_; in AppendSlidingParamConv()
152 sliding->ic_align_ = UP_DIV(conv_param->input_channel_, input_block) * input_block; in AppendSlidingParamConv()
154 …sliding->in_step_ = conv_param->input_h_ * conv_param->input_w_ * sliding->ic_align_; // for batc… in AppendSlidingParamConv()
155 sliding->in_h_step_ = conv_param->input_w_ * sliding->ic_align_; in AppendSlidingParamConv()
156 …sliding->in_sh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->stride_h_; // st… in AppendSlidingParamConv()
157 …sliding->in_sw_step_ = sliding->ic_align_ * conv_param->stride_w_; // st… in AppendSlidingParamConv()
158 …sliding->in_kh_step_ = conv_param->input_w_ * sliding->ic_align_ * conv_param->dilation_h_; // ke… in AppendSlidingParamConv()
159 …sliding->in_kw_step_ = sliding->ic_align_ * conv_param->dilation_w_; // ke… in AppendSlidingParamConv()
160 …sliding->kernel_step_ = conv_param->kernel_w_ * conv_param->kernel_h_ * sliding->ic_align_ * weigh… in AppendSlidingParamConv()
Dconv_1x1_x86_fp32.c50 int ic_align = sw_param->ic_align_; in Conv1x1SWFp32()