| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_1x1_x86_fp32.c | 51 int in_sw_step = sw_param->in_sw_step_; in Conv1x1SWFp32() local
103 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW3x32Kernel()
379 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x32Kernel()
526 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW4x24Kernel()
804 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x24Kernel()
929 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW6x16Kernel()
1219 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x16Kernel()
1322 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW12x8Kernel()
1476 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x8Kernel()
1557 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SWOWxOCKernel()
|
| D | conv_common_fp32.c | 354 int in_sw_step = sw_param->in_sw_step_; in ConvSWFp32() local
435 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv3x32Kernel()
594 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x32Kernel()
689 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv4x24Kernel()
864 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x24Kernel()
951 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv6x16Kernel()
1129 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x16Kernel()
1209 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv12x8Kernel()
1389 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv4x8Kernel()
1486 … size_t ic_algin, size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder, in SWConv1x8Kernel()
[all …]
|
| D | conv_depthwise_fp32.c | 65 int in_sw_step = conv_param->stride_w_ * conv_param->input_channel_; in ConvDw() local
263 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_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()
1073 int in_sw_step = sw_param->in_sw_step_; in DepthwiseSWAvxFp32() local
1139 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSWWxKKernel()
1200 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSW3x32Kernel()
1345 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSW1x32Kernel()
1424 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSW4x24Kernel()
1569 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSW1x24Kernel()
1641 … size_t in_kw_step, size_t in_kh_step, size_t in_sw_step, size_t kw_remainder) { in DepthwiseSW4x16Kernel()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/ |
| D | DepthwiseFp32_Sse.c | 112 … size_t in_sw_step, size_t in_kh_step, size_t in_kw_step, size_t relu, size_t relu6) { 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()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 425 int in_sw_step = conv_param->stride_w_ * conv_param->input_channel_; in ConvDwFp16() local
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()
680 int in_sw_step, int in_kh_step, int in_kw_step) { in DeconvDepthwiseCenterFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv_depthwise_int8.c | 116 int in_sw_step = conv_param->stride_w_ * conv_param->input_channel_; in ConvDwInt8() local
569 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_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()
|