Home
last modified time | relevance | path

Searched defs:kernel_chw (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_im2col_base_fp32.cc49 int kernel_chw = kernel_hw * conv_param_->input_channel_; in InitTmpBuffer() local
90 size_t kernel_chw = in_channel * kernel_hw; in Prepare() local
Dconvolution_im2col_avx512_fp32.cc40 int kernel_chw = kernel_hw * conv_param_->input_channel_; in InitTmpBuffer() local
Dconvolution_im2col_avx_fp32.cc40 int kernel_chw = kernel_hw * conv_param_->input_channel_; in InitTmpBuffer() local
Dadder_fp32.cc81 int kernel_chw = in_channel * kernel_hw; in InitWeightBias() local
Dconvolution_slidewindow_fp32.cc54 int kernel_chw = input_channel * kernel_hw; in Prepare() local
Dconvolution_fp32.cc47 int kernel_chw = kernel_hw * conv_param_->input_channel_; in InitTmpBuffer() local
Ddeconvolution_fp32.cc212 int kernel_chw = input_channel * kernel_hw; in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconvolution_im2col_base.c115 int kernel_chw = conv->compute_.kernel_hw_ * conv->compute_.in_c_; in ConvIm2ColBaseInitTmpBuffer() local
219 int kernel_chw = conv_im2col->conv_.compute_.in_c_ * conv_im2col->conv_.compute_.kernel_hw_; in ConvolutionIm2colBasePrepare() local
Dconvolution_im2col_avx.c30 int kernel_chw = conv_im2col->conv_.compute_.kernel_hw_ * conv_im2col->conv_.compute_.in_c_; in ConvIm2ColAVXInitTmpBuffer() local
Dconvolution_im2col_avx512.c38 int kernel_chw = compute->kernel_hw_ * compute->in_c_; in ConvIm2ColAVX512InitTmpBuffer() local
Dconvolution_slidewindow.c218 int kernel_chw = input_channel * kernel_hw; in ConvolutionSWPrepare() local
Ddeconvolution.c261 int kernel_chw = deconv->conv_.compute_.in_c_ * deconv->conv_.compute_.kernel_hw_; in DeConvPrepare() local