Home
last modified time | relevance | path

Searched defs:weight_kernel (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dmatmul_avx512_mask_fp32.c47 const float *weight_kernel = weight; in GemmRowxColMaskKernelFp32() local
Dmatmul_avx512_fp32.c45 const float *weight_kernel = weight; in GemmRowxColKernelFp32() local
Dmatmul_avx_fp32.c928 const float *weight_kernel = weight; in MatVecMulRowxColKernel() local
Dconv_1x1_avx_fp32.c1581 const float *weight_kernel = weight; in Conv1x1SWOWxOCAVXKernel() local
Dconv_depthwise_fp32.c306 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in ConvDwBorder() local
990 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwBorder() local
1119 …const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM * oc_b… in DepthwiseBorderAvxFp32() local
1236 const float *weight_kernel = weight; in DepthwiseSWWxKKernel() local
Dconv_sw_avx_fp32.c1185 const float *weight_kernel = weight; in SWConvWxKAVXKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dconv_depthwise_fp16.c508 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DepthwiseBorderFp16() local
667 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DeconvDepthwiseBorderFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dconv_depthwise_int8.c556 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in ConvDwInt8Border() local
707 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwInt8Border() local