Searched refs:ow_block_num (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_1x1_x86_fp32.c | 57 const int ow_block_num[4] = {12, 6, 4, 3}; in Conv1x1SWFp32() local
83 int hw_block = ow_block_num[oc_block - 1]; in Conv1x1SWFp32()
89 …kernel[oc_block - 1][hw_block / ow_block_num[oc_block - 1]](dst_w, src_w, weight, bias, act_type, … in Conv1x1SWFp32()
|
| D | conv_common_fp32.c | 381 const int ow_block_num[4] = {12, 6, 4, 3}; in ConvSWFp32() local
411 int ow_block = ow_block_num[oc_block - 1]; // 12 6 4 3 in ConvSWFp32()
414 if (ow_block < ow_block_num[oc_block - 1]) { // ow is not enough and process one ow in ConvSWFp32()
417 kernel[oc_block - 1][ow_block / ow_block_num[oc_block - 1]]( in ConvSWFp32()
|
| D | conv_depthwise_fp32.c | 1086 const int ow_block_num[4] = {8, 4, 4, 3}; in DepthwiseSWAvxFp32() local
1115 int ow_block = ow_block_num[oc_block - 1]; // 8 4 4 3 in DepthwiseSWAvxFp32()
1120 kernel[oc_block - 1][ow_block / ow_block_num[oc_block - 1]]( in DepthwiseSWAvxFp32()
|