| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | conv_1x1_avx_fp32.c | 20 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW3x32AVXKernel()
296 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x32AVXKernel()
443 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW4x24AVXKernel()
721 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x24AVXKernel()
846 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW6x16AVXKernel()
1136 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x16AVXKernel()
1239 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW12x8AVXKernel()
1393 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SW1x8AVXKernel()
1522 int oc_block = 0; in Conv1x1SWAVXFp32() local
1552 … size_t ow_block, size_t oc_block, size_t oc_align, size_t ic_align, size_t in_sw_step, in Conv1x1SWOWxOCAVXKernel()
|
| D | conv_sw_avx_fp32.c | 22 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv3x32AVXKernel()
181 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x32AVXKernel()
276 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x24AVXKernel()
451 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x24AVXKernel()
538 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv6x16AVXKernel()
716 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x16AVXKernel()
796 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv12x8AVXKernel()
976 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x8AVXKernel()
1073 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x8AVXKernel()
1165 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConvWxKAVXKernel()
|
| D | conv_depthwise_fp32.c | 1106 … const DepthwiseSWKernel kernel, int act_type, int ow_bock, int oc_block) { in DepthwiseBorderAvxFp32()
1173 int oc_block = 0; in DepthwiseSWAvxFp32() local
1216 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSWWxKKernel()
1277 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW3x32Kernel()
1422 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x32Kernel()
1501 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x24Kernel()
1646 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x24Kernel()
1718 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x16Kernel()
1832 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x16Kernel()
1897 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW8x8Kernel()
[all …]
|
| D | pack_fp32.c | 1114 int oc_block = UP_DIV(channel, C4NUM); in PackNHWCToNHWC4Fp32() local
1134 int oc_block = UP_DIV(channel, oc_tile); in PackNHWCToNHWCXFp32() local
1158 int oc_block = 0; in PackNHWCToNXHWCXFp32H1W1() local
1210 int oc_block = 0; in PackNHWCToNXHWCXFp32() local
1253 int oc_block = 0; in SWPackNHWCToNXHWCXFp32() local
1518 for (int i = 0, oc_block = 0; i < oc_block_num; i += oc_block) { in PackNHWCToCXHWNXFp32() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | adder_fp32.cc | 78 const int oc_block = C4NUM; in InitWeightBias() local
|
| D | convolution_winograd_base_fp32.cc | 36 … const float *matrix_gt, int oc_block) { in WinogradFilterTransform()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_fp32_coder.cc | 67 int oc_block = C8NUM; in InitWeightBias() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/ |
| D | convolution_winograd_fp16.cc | 24 … const float *matrix_gt, int oc_block) { in WinogradFilterTransformFp16()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | minimal_filtering_generator.c | 257 … int oc_block, int input_unit, int kernel_unit, int channel, int batch, bool pack) { in WinogradWeightTransform()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 264 … const float *matrix_gt, int oc_block, int input_unit, int kernel_unit, in WinogradWeightTransformFp16()
|