Searched refs:start_ocb (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_brgemm_conv_utils.cpp | 1447 auto start_ocb = 4; in init_conf() local
1449 start_ocb = nstl::min(jcp.ic > 128 ? (jcp.ic > 256 ? 8 : 16) : 32, in init_conf()
1451 start_ocb = nstl::min(div_up(jcp.oc, 16), start_ocb); in init_conf()
1454 for (auto ocb = start_ocb; ocb >= finish_ocb; ocb--) { in init_conf()
1631 auto start_ocb = 4; in init_1x1_conf() local
1633 start_ocb = nstl::min(jcp.ic > 128 ? (jcp.ic > 256 ? 8 : 16) : 32, in init_1x1_conf()
1635 start_ocb = nstl::min(div_up(jcp.oc, 16), start_ocb); in init_1x1_conf()
1638 for (auto ocb = start_ocb; ocb >= finish_ocb; ocb--) { in init_1x1_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_brgemm_conv_utils.cpp | 1761 auto start_ocb = (is_amx(isa) && jcp.is_os_blocking) ? 2 : 4; in init_conf() local
1763 start_ocb = nstl::min(jcp.ic > 128 ? (jcp.ic > 256 ? 8 : 16) : 32, in init_conf()
1765 start_ocb = nstl::min(div_up(jcp.oc, 16), start_ocb); in init_conf()
1768 for (auto ocb = start_ocb; ocb >= finish_ocb; ocb--) { in init_conf()
1997 auto start_ocb = 4; in init_1x1_conf() local
1999 start_ocb = nstl::min(jcp.ic > 128 ? (jcp.ic > 256 ? 8 : 16) : 32, in init_1x1_conf()
2001 start_ocb = nstl::min(div_up(jcp.oc, 16), start_ocb); in init_1x1_conf()
2004 for (auto ocb = start_ocb; ocb >= finish_ocb; ocb--) { in init_1x1_conf()
|