| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | common_func_fp32.h | 52 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
61 … size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, size_t in_sw_step,
77 … size_t in_sh_step, size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, size_t relu,
|
| D | conv_depthwise_fp32.h | 29 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step,
|
| D | conv_depthwise_fp32.c | 263 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, in ConvDwCenter() argument
299 src_h += in_sh_step; in ConvDwCenter()
930 … int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, int in_kh_step, in DeconvDwCenter() argument
956 dst_h += in_sh_step; in DeconvDwCenter()
1076 int in_sh_step = sw_param->in_sh_step_; in DepthwiseSWAvxFp32() local
1094 const float *src_h = input_data + in_start + (oh - out_top) * in_sh_step; in DepthwiseSWAvxFp32()
|
| D | conv_common_fp32.c | 357 int in_sh_step = sw_param->in_sh_step_; in ConvSWFp32() local
410 const float *src_w = src_h + (oh - top) * in_sh_step; in ConvSWFp32()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
| D | DepthwiseFp32_Sse.c | 111 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, in ConvDwFp32Center() argument
115 in_sh_step /= sizeof(float); in ConvDwFp32Center()
228 src_h += in_sh_step; in ConvDwFp32Center()
233 … size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, size_t in_sw_step, in DeconvDwFp32Center() argument
237 in_sh_step /= sizeof(float); in DeconvDwFp32Center()
322 dst_h += in_sh_step; in DeconvDwFp32Center()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.h | 34 … size_t block_channel, size_t in_sh_step, size_t in_sw_step, size_t in_kh_step, size_t in_kw_step,
39 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
|
| D | conv_depthwise_fp16.c | 522 … int in_sh_step, int in_sw_step, int in_kh_step, int in_kw_step, bool is_relu, bool is_relu6) { in DepthwiseCenterFp16() argument
565 src_h += in_sh_step; in DepthwiseCenterFp16()
679 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, in DeconvDepthwiseCenterFp16() argument
713 dst_h += in_sh_step; in DeconvDepthwiseCenterFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | common_func_int8.h | 45 … size_t in_sh_step, size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, const int8_t *in_zp,
49 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step,
|
| D | conv_depthwise_int8.c | 569 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, in ConvDwInt8Center() argument
614 src_h += in_sh_step; in ConvDwInt8Center()
719 … int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, int in_kh_step, in DeconvDwInt8Center() argument
745 dst_h += in_sh_step; in DeconvDwInt8Center()
|