Home
last modified time | relevance | path

Searched refs:b_buf_nelems (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/
H A Dgemm_driver.cpp562 size_t b_buf_nelems = k_padd * n_padd; in gemm_kernel_driver() local
567 b_buf_nelems = utils::rnd_up(k_padd, arg->uk) in gemm_kernel_driver()
574 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in gemm_kernel_driver()
577 + b_buf_nelems * sizeof(*b) + PAGE_4K; in gemm_kernel_driver()
608 a_row_sum = (c_type *)align(bufferB + b_buf_nelems, PAGE_4K); in gemm_kernel_driver()
617 bufferC = (c_type *)align(bufferB + b_buf_nelems, PAGE_4K); in gemm_kernel_driver()
779 size_t b_buf_nelems = k * n_padd; in kernel_driver_parallel_acopiedbcopy() local
791 b_buf_nelems in kernel_driver_parallel_acopiedbcopy()
794 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in kernel_driver_parallel_acopiedbcopy()
796 size_t mem_size = b_buf_nelems * sizeof(*b) + PAGE_4K; in kernel_driver_parallel_acopiedbcopy()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/
H A Dgemm_driver.cpp562 size_t b_buf_nelems = k_padd * n_padd; in gemm_kernel_driver() local
567 b_buf_nelems = utils::rnd_up(k_padd, arg->uk) in gemm_kernel_driver()
574 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in gemm_kernel_driver()
577 + b_buf_nelems * sizeof(*b) + PAGE_4K; in gemm_kernel_driver()
608 a_row_sum = (c_type *)align(bufferB + b_buf_nelems, PAGE_4K); in gemm_kernel_driver()
617 bufferC = (c_type *)align(bufferB + b_buf_nelems, PAGE_4K); in gemm_kernel_driver()
779 size_t b_buf_nelems = k * n_padd; in kernel_driver_parallel_acopiedbcopy() local
791 b_buf_nelems in kernel_driver_parallel_acopiedbcopy()
794 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in kernel_driver_parallel_acopiedbcopy()
796 size_t mem_size = b_buf_nelems * sizeof(*b) + PAGE_4K; in kernel_driver_parallel_acopiedbcopy()
[all …]