Searched refs:force_k_decomp (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/ |
| H A D | gemm_driver.cpp | 1647 gemm_threading_t force_k_decomp; in gemm_threading_driver() local
1664 set_thread_opts_pack(nthr_goal, force_k_decomp, arg); in gemm_threading_driver()
1667 if (force_k_decomp.nthrs_k > 1) force_threading = &force_k_decomp; in gemm_threading_driver()
1672 nthr_goal, force_k_decomp, arg, true, true, false); in gemm_threading_driver()
1675 if (force_k_decomp.nthrs_k > 1 && force_k_decomp.nthrs_m > 1) in gemm_threading_driver()
1676 force_threading = &force_k_decomp; in gemm_threading_driver()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
| H A D | gemm_driver.cpp | 1647 gemm_threading_t force_k_decomp; in gemm_threading_driver() local
1664 set_thread_opts_pack(nthr_goal, force_k_decomp, arg); in gemm_threading_driver()
1667 if (force_k_decomp.nthrs_k > 1) force_threading = &force_k_decomp; in gemm_threading_driver()
1672 nthr_goal, force_k_decomp, arg, true, true, false); in gemm_threading_driver()
1675 if (force_k_decomp.nthrs_k > 1 && force_k_decomp.nthrs_m > 1) in gemm_threading_driver()
1676 force_threading = &force_k_decomp; in gemm_threading_driver()
|