Searched refs:gemm_kernel_driver (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 | 510 static dnnl_status_t gemm_kernel_driver(int ithr, dim_t m, dim_t n, dim_t k, in gemm_kernel_driver() function 1733 return gemm_kernel_driver(0, arg->m, arg->n, arg->k, arg->a, arg->b, in gemm_threading_driver() 1798 thread_arg[0].result = gemm_kernel_driver(0, arg->m, arg->n, arg->k, in gemm_threading_driver() 1850 thread_arg[ithr].result = gemm_kernel_driver(ithr, m, n, in gemm_threading_driver()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
H A D | gemm_driver.cpp | 510 static dnnl_status_t gemm_kernel_driver(int ithr, dim_t m, dim_t n, dim_t k, in gemm_kernel_driver() function 1733 return gemm_kernel_driver(0, arg->m, arg->n, arg->k, arg->a, arg->b, in gemm_threading_driver() 1798 thread_arg[0].result = gemm_kernel_driver(0, arg->m, arg->n, arg->k, in gemm_threading_driver() 1850 thread_arg[ithr].result = gemm_kernel_driver(ithr, m, n, in gemm_threading_driver()
|