Home
last modified time | relevance | path

Searched defs:src_kh (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/intrinsics/sse/
DDepthwiseFp32_Sse.c29 const float *src_kh = src; in ConvDwFp32Border() local
130 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
178 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
209 const float *src_kh = src_w, *weight_kh = weight; in ConvDwFp32Center() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dconv_depthwise_avx_fp32.c60 const float *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwAVX() local
Dconv_depthwise_fp32.c62 const float *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDw() local
131 const float *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwAVX512() local
259 const float *src_kh = src; in ConvDwBorderPixel() local
349 const float *src_kh = src_w; in ConvDwCenter() local
1220 const float *src_kh[12]; in DepthwiseSWWxKKernel() local
Dconv_sw_avx_fp32.c1169 const float *src_kh[12]; in SWConvWxKAVXKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/int8/
Dconv_depthwise_int8.c113 const int8_t *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwInt8() local
166 const int8_t *src_kh = buffer; in ConvDw3x3Int8Window() local
341 const int8_t *src_kh = src; in ConvDw3x3Int8BorderPixel() local
508 const int8_t *src_kh = src; in ConvDwInt8BorderPixel() local
581 const int8_t *src_kh = src_w; in ConvDwInt8Center() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dconv_depthwise_fp16.c424 const float16_t *src_kh = src + ih * conv_param->input_w_ * conv_param->input_channel_; in ConvDwFp16() local
461 const float16_t *src_kh = src; in DepthwiseBorderPixelFp16() local
533 const float16_t *src_kh = src_w; in DepthwiseCenterFp16() local