| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_transpose_utils.hpp | 37 int nthr_oc_b; /* number of threads process given src image */ member
68 int nthr_oc_b; /* number of threads process given src image */ member
|
| H A D | jit_avx512_common_1x1_conv_kernel.cpp | 1462 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1492 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_brgemm_primitive_conf.hpp | 76 int nthr, nthr_mb, nthr_oc_b, nthr_ic_b; member
|
| H A D | jit_brgemm_inner_product_utils.cpp | 663 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in thread_balance() local
|
| H A D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 1844 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1877 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_avx512_common_conv_kernel.cpp | 5998 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
6155 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
6192 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
6206 auto calc_comp_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
6221 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 4470 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
4583 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
4646 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 5099 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
5220 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
5293 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_primitive_conf.hpp | 123 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b, nthr_oh; member
601 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_transpose_utils.hpp | 37 int nthr_oc_b; /* number of threads process given src image */ member
68 int nthr_oc_b; /* number of threads process given src image */ member
|
| H A D | jit_avx512_common_1x1_conv_kernel.cpp | 1440 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1470 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_brgemm_primitive_conf.hpp | 77 int nthr, nthr_mb, nthr_oc_b, nthr_ic_b; member
|
| H A D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 1825 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1858 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_brgemm_inner_product_utils.cpp | 682 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in thread_balance() local
|
| H A D | jit_avx512_common_conv_kernel.cpp | 5999 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
6156 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
6193 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
6207 auto calc_comp_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
6222 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 4461 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
4574 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
4637 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 5125 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
5246 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
5319 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_primitive_conf.hpp | 125 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b, nthr_oh; member
604 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_sve_512_1x1_conv_kernel.cpp | 1174 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1199 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_sve_512_conv_kernel.cpp | 4174 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
4292 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
4328 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_primitive_conf.hpp | 119 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
353 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_sve_512_1x1_conv_kernel.cpp | 1174 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
1199 int nthr_mb = 1, nthr_oc_b = 1, nthr_ic_b = 1; in balance() local
|
| H A D | jit_sve_512_conv_kernel.cpp | 4174 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; in init_conf() local
4292 auto calc_mem_cost = [=](int nthr_mb, int nthr_oc_b, int nthr_ic_b) { in balance()
4328 for (int nthr_oc_b = 1; nthr_oc_b <= nthr_oc_b_max; ++nthr_oc_b) { in balance() local
|
| H A D | jit_primitive_conf.hpp | 119 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
353 int nthr, nthr_mb, nthr_g, nthr_oc_b, nthr_ic_b; member
|