Home
last modified time | relevance | path

Searched refs:weight_kernel (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_1x1_x86_fp32.c1586 const float *weight_kernel = weight; in Conv1x1SWOWxOCKernel() local
1589 weight_data[j] = _mm256_loadu_ps(weight_kernel + j * C8NUM); in Conv1x1SWOWxOCKernel()
1596 weight_kernel += C8NUM * oc_block; in Conv1x1SWOWxOCKernel()
Dconv_depthwise_fp32.c228 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in ConvDwBorder() local
236 param->weight = weight_kernel; in ConvDwBorder()
248 …ConvDwFp32Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in ConvDwBorder()
252 …ConvDwBorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_k… in ConvDwBorder()
912 const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwBorder() local
915 … DeconvDwFp32Border(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDwBorder()
919 … DeconvDwBorderPixel(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDwBorder()
1041 …const float *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM * oc_b… in DepthwiseBorderAvxFp32() local
1042 …kernel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, act_type… in DepthwiseBorderAvxFp32()
1158 const float *weight_kernel = weight; in DepthwiseSWWxKKernel() local
[all …]
Dconv_common_fp32.c321 const float *weight_kernel = in SWBorder() local
323 …kernel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, act_type… in SWBorder()
1572 const float *weight_kernel = weight; in SWConvWxKKernel() local
1580 weight_data[j] = _mm256_loadu_ps(weight_kernel + j * C8NUM); in SWConvWxKKernel()
1587 weight_kernel += C8NUM * oc_block; in SWConvWxKKernel()
1593 weight_kernel += kw_remainder; in SWConvWxKKernel()
Dmatmul_fp32.c1495 const float *weight_kernel = weight; in MatVecMulRowxColKernel() local
1498 weight_data[j] = _mm256_loadu_ps(weight_kernel + j * C8NUM); in MatVecMulRowxColKernel()
1506 weight_kernel += C8NUM * col_block; in MatVecMulRowxColKernel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dconv_depthwise_fp16.c504 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DepthwiseBorderFp16() local
506 …ConvDwFp16Border(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - start_kw, in DepthwiseBorderFp16()
510 …DepthwiseBorderPixelFp16(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - … in DepthwiseBorderFp16()
661 … const float16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in DeconvDepthwiseBorderFp16() local
664 … DeconvDwFp16Border(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_kw, in DeconvDepthwiseBorderFp16()
668 …DeconvDepthwiseBorderPixelFp16(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - … in DeconvDepthwiseBorderFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv_depthwise_int8.c555 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C8NUM; in ConvDwInt8Border() local
557 …ConvDwInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, bias, end_kh - start_kh, end_kw - sta… in ConvDwInt8Border()
706 const int16_t *weight_kernel = weight + (start_kh * conv_param->kernel_w_ + start_kw) * C4NUM; in DeconvDwInt8Border() local
709 …DeconvDwInt8BorderPixel(dst_kernel, src_kernel, weight_kernel, end_kh - start_kh, end_kw - start_k… in DeconvDwInt8Border()