Home
last modified time | relevance | path

Searched refs:ic_algin (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.h48 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step,
63 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
69 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
74 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
79 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
84 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
89 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
94 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
99 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
104 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder,
[all …]
Dconv_common_fp32.c353 int ic_algin = sw_param->ic_align_; in ConvSWFp32() local
379 int center_step = in_h_start * in_h_step + in_w_start * ic_algin; in ConvSWFp32()
419 out_block_step, ic_algin, in_kw_step, in_kh_step, in_sw_step, 0, write_mode); in ConvSWFp32()
435 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv3x32Kernel() argument
512 …: "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_step), … in SWConv3x32Kernel()
594 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x32Kernel() argument
641 …: "r"(src), "r"(weight), "r"(bias), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_step), … in SWConv1x32Kernel()
689 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv4x24Kernel() argument
782 …: "r"(src), "r"(weight), "r"(src_3_step), "r"(kernel_h), "r"(kernel_w), "r"(ic_algin), "r"(in_kw_s… in SWConv4x24Kernel()
864 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x24Kernel() argument
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_depthwise_slidewindow_x86_fp32.cc37 int ic_algin = UP_DIV(conv_param_->input_channel_, oc_tile_); in InitPackedInputOutput() local
39 … conv_param_->input_batch_ * conv_param_->input_h_ * conv_param_->input_w_ * oc_tile_ * ic_algin; in InitPackedInputOutput()