Searched refs:min_mblk (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 | 1246 auto min_mblk = mayiuse(avx512_core) ? (MBLK / 2) : arg->um; in set_thread_opts_pack() local 1247 min_mblk = do_m_blocking ? min_mblk : m; in set_thread_opts_pack() 1248 min_mblk = do_m_blocking_only ? arg->um : min_mblk; in set_thread_opts_pack() 1251 std::tie(nthr_m, nthr_n) = partition_2d_minblk(m, n, MBLK, NBLK, min_mblk, in set_thread_opts_pack()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
H A D | gemm_driver.cpp | 1246 auto min_mblk = mayiuse(avx512_core) ? (MBLK / 2) : arg->um; in set_thread_opts_pack() local 1247 min_mblk = do_m_blocking ? min_mblk : m; in set_thread_opts_pack() 1248 min_mblk = do_m_blocking_only ? arg->um : min_mblk; in set_thread_opts_pack() 1251 std::tie(nthr_m, nthr_n) = partition_2d_minblk(m, n, MBLK, NBLK, min_mblk, in set_thread_opts_pack()
|