Home
last modified time | relevance | path

Searched refs:b_col_sum_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.cpp571 size_t b_col_sum_nelems = n_padd; in gemm_kernel_driver() local
574 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in gemm_kernel_driver()
581 + b_col_sum_nelems * sizeof(*c) + PAGE_4K; in gemm_kernel_driver()
615 bufferC = (c_type *)align(b_col_sum + b_col_sum_nelems, PAGE_4K); in gemm_kernel_driver()
780 size_t b_col_sum_nelems = n_padd; in kernel_driver_parallel_acopiedbcopy() local
794 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in kernel_driver_parallel_acopiedbcopy()
798 if (is_int8) { mem_size += b_col_sum_nelems * sizeof(*c) + PAGE_4K; } in kernel_driver_parallel_acopiedbcopy()
828 bufferC = (c_type *)align(b_col_sum + b_col_sum_nelems, PAGE_4K); in kernel_driver_parallel_acopiedbcopy()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/
H A Dgemm_driver.cpp571 size_t b_col_sum_nelems = n_padd; in gemm_kernel_driver() local
574 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in gemm_kernel_driver()
581 + b_col_sum_nelems * sizeof(*c) + PAGE_4K; in gemm_kernel_driver()
615 bufferC = (c_type *)align(b_col_sum + b_col_sum_nelems, PAGE_4K); in gemm_kernel_driver()
780 size_t b_col_sum_nelems = n_padd; in kernel_driver_parallel_acopiedbcopy() local
794 if (b_packed) b_buf_nelems = b_col_sum_nelems = 0; in kernel_driver_parallel_acopiedbcopy()
798 if (is_int8) { mem_size += b_col_sum_nelems * sizeof(*c) + PAGE_4K; } in kernel_driver_parallel_acopiedbcopy()
828 bufferC = (c_type *)align(b_col_sum + b_col_sum_nelems, PAGE_4K); in kernel_driver_parallel_acopiedbcopy()