Searched refs:sz_want_borrow (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_reorder.cpp | 1492 size_t sz_want_borrow = utils::div_up(tr::ker_prb_size_min, sz_ker_cur); in prb_thread_kernel_balance() local
1493 for (; prb.nodes[kdims].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1495 if (sz_want_borrow != prb.nodes[kdims].n) in prb_thread_kernel_balance()
1496 prb_node_split(prb, kdims, sz_want_borrow); in prb_thread_kernel_balance()
1508 size_t sz_want_borrow = utils::div_up(sz_drv_min, sz_drv_cur); in prb_thread_kernel_balance() local
1509 for (; prb.nodes[kdims - 1].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1511 if (sz_want_borrow != prb.nodes[kdims - 1].n) in prb_thread_kernel_balance()
1513 prb, kdims - 1, prb.nodes[kdims - 1].n / sz_want_borrow); in prb_thread_kernel_balance()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_uni_reorder.cpp | 1454 size_t sz_want_borrow = utils::div_up(tr::ker_prb_size_min, sz_ker_cur); in prb_thread_kernel_balance() local
1455 for (; prb.nodes[kdims].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1457 if (sz_want_borrow != prb.nodes[kdims].n) in prb_thread_kernel_balance()
1458 prb_node_split(prb, kdims, sz_want_borrow); in prb_thread_kernel_balance()
1469 size_t sz_want_borrow = utils::div_up(sz_drv_min, sz_drv_cur); in prb_thread_kernel_balance() local
1470 for (; prb.nodes[kdims - 1].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1472 if (sz_want_borrow != prb.nodes[kdims - 1].n) in prb_thread_kernel_balance()
1474 prb, kdims - 1, prb.nodes[kdims - 1].n / sz_want_borrow); in prb_thread_kernel_balance()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_uni_reorder.cpp | 1454 size_t sz_want_borrow = utils::div_up(tr::ker_prb_size_min, sz_ker_cur); in prb_thread_kernel_balance() local
1455 for (; prb.nodes[kdims].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1457 if (sz_want_borrow != prb.nodes[kdims].n) in prb_thread_kernel_balance()
1458 prb_node_split(prb, kdims, sz_want_borrow); in prb_thread_kernel_balance()
1469 size_t sz_want_borrow = utils::div_up(sz_drv_min, sz_drv_cur); in prb_thread_kernel_balance() local
1470 for (; prb.nodes[kdims - 1].n % sz_want_borrow; ++sz_want_borrow) in prb_thread_kernel_balance()
1472 if (sz_want_borrow != prb.nodes[kdims - 1].n) in prb_thread_kernel_balance()
1474 prb, kdims - 1, prb.nodes[kdims - 1].n / sz_want_borrow); in prb_thread_kernel_balance()
|