Searched refs:m_padd (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 | 206 auto m_padd = get_m_padd(ithr, m, arg); in get_m_padd_parallel_a() local
211 m_padd *= nstl::max(nthrs, multiplier); in get_m_padd_parallel_a()
212 if (m_padd > m) m_padd = utils::rnd_up(m, arg->um); in get_m_padd_parallel_a()
215 return m_padd; in get_m_padd_parallel_a()
550 dim_t m_padd = get_m_padd(ithr, m, arg); in gemm_kernel_driver() local
561 size_t a_buf_nelems = m_padd * k_padd; in gemm_kernel_driver()
570 size_t a_row_sum_nelems = m_padd; in gemm_kernel_driver()
624 if (sizeM > m_padd) sizeM = m_padd; in gemm_kernel_driver()
1390 size_t a_buf_nelems = m_padd * k_padd; in parallel_a_copy()
1407 size_t a_row_sum_nelems = m_padd; in parallel_a_copy()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
| H A D | gemm_driver.cpp | 206 auto m_padd = get_m_padd(ithr, m, arg); in get_m_padd_parallel_a() local
211 m_padd *= nstl::max(nthrs, multiplier); in get_m_padd_parallel_a()
212 if (m_padd > m) m_padd = utils::rnd_up(m, arg->um); in get_m_padd_parallel_a()
215 return m_padd; in get_m_padd_parallel_a()
550 dim_t m_padd = get_m_padd(ithr, m, arg); in gemm_kernel_driver() local
561 size_t a_buf_nelems = m_padd * k_padd; in gemm_kernel_driver()
570 size_t a_row_sum_nelems = m_padd; in gemm_kernel_driver()
624 if (sizeM > m_padd) sizeM = m_padd; in gemm_kernel_driver()
1390 size_t a_buf_nelems = m_padd * k_padd; in parallel_a_copy()
1407 size_t a_row_sum_nelems = m_padd; in parallel_a_copy()
[all …]
|