| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | gemm_convolution.cpp | 59 status_t st_thr = execute_forward_thr_nspc(ctx, ithr, nthr, src_base, in execute_forward_nspc() local
387 status_t st_thr in execute_forward_ncsp() local
402 status_t st_thr in execute_forward_ncsp() local
430 status_t st_thr = execute_backward_data_thr_nspc(ithr, nthr, in execute_backward_data_nspc() local
565 status_t st_thr = extended_sgemm("N", "T", &os_block, &N, &K, in execute_backward_data_ncsp() local
678 status_t st_thr = extended_sgemm("N", in execute_backward_weights_nspc() local
843 status_t st_thr = extended_sgemm("T", "N", &M, &N, in execute_backward_weights_ncsp() local
|
| H A D | gemm_x8s8s32x_convolution.cpp | 135 status_t st_thr = execute_forward_thr(ithr, nthr, src_base, wei_base, in execute_forward() local
310 status_t st_thr = execute_backward_data_thr(ithr, nthr, diff_dst_base, in execute_backward_data() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | gemm_convolution.cpp | 59 status_t st_thr = execute_forward_thr_nspc(ctx, ithr, nthr, src_base, in execute_forward_nspc() local
384 status_t st_thr in execute_forward_ncsp() local
399 status_t st_thr in execute_forward_ncsp() local
427 status_t st_thr = execute_backward_data_thr_nspc(ithr, nthr, in execute_backward_data_nspc() local
562 status_t st_thr = extended_sgemm("N", "T", &os_block, &N, &K, in execute_backward_data_ncsp() local
675 status_t st_thr = extended_sgemm("N", in execute_backward_weights_nspc() local
840 status_t st_thr = extended_sgemm("T", "N", &M, &N, in execute_backward_weights_ncsp() local
|
| H A D | gemm_x8s8s32x_convolution.cpp | 134 status_t st_thr = execute_forward_thr(ithr, nthr, src_base, wei_base, in execute_forward() local
320 status_t st_thr = execute_backward_data_thr(ithr, nthr, diff_dst_base, in execute_backward_data() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | gemm_bf16_convolution.cpp | 397 status_t st_thr = execute_forward_thr_nspc(ithr, nthr, src_base, in execute_forward_nspc() local
611 status_t st_thr = gemm_bf16bf16f32("N", "N", &m, &N, &K, &one, in execute_forward_ncsp() local
692 status_t st_thr = execute_backward_data_thr_nspc( in execute_backward_data_nspc() local
856 status_t st_thr = gemm_bf16bf16f32("N", "T", &os_block, &N, &K, in execute_backward_data_ncsp() local
1080 status_t st_thr = gemm_bf16bf16f32("N", in execute_backward_weights_nspc() local
1269 status_t st_thr = gemm_bf16bf16f32("T", "N", &M, &N, in execute_backward_weights_ncsp() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | gemm_bf16_convolution.cpp | 387 status_t st_thr = execute_forward_thr_nspc(ithr, nthr, src_base, in execute_forward_nspc() local
601 status_t st_thr = gemm_bf16bf16f32("N", "N", &m, &N, &K, &one, in execute_forward_ncsp() local
684 status_t st_thr = execute_backward_data_thr_nspc( in execute_backward_data_nspc() local
848 status_t st_thr = gemm_bf16bf16f32("N", "T", &os_block, &N, &K, in execute_backward_data_ncsp() local
1071 status_t st_thr = gemm_bf16bf16f32("N", in execute_backward_weights_nspc() local
1260 status_t st_thr = gemm_bf16bf16f32("T", "N", &M, &N, in execute_backward_weights_ncsp() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/matmul/ |
| H A D | gemm_bf16_matmul.cpp | 276 status_t st_thr = gemm_bf16bf16f32(&transB, &transA, &gemm_N, in execute_ref() local
|
| H A D | gemm_f32_matmul.cpp | 270 status_t st_thr = extended_sgemm(&transB, &transA, &gemm_N, in execute_ref() local
|
| H A D | gemm_x8s8s32x_matmul.cpp | 320 status_t st_thr = gemm_s8x8s32(&transB, &transA, "F", &gemm_N, in execute_ref() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/matmul/ |
| H A D | gemm_f32_matmul.cpp | 299 status_t st_thr = extended_sgemm(&transB, &transA, &gemm_N, in execute_ref() local
|
| H A D | gemm_bf16_matmul.cpp | 299 status_t st_thr = gemm_bf16bf16f32(&transB, &transA, &gemm_N, in execute_ref() local
|
| H A D | gemm_x8s8s32x_matmul.cpp | 336 status_t st_thr = status::runtime_error; in execute_ref() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/f32/ |
| H A D | jit_avx512_core_gemm_smalln_tn_f32_kern.cpp | 827 auto st_thr = sgemm_smalln_tn(mpart, n, k, alpha, &A[ithr * lda * mid], in jit_avx512_core_gemm_smalln_tn_f32() local
|
| H A D | jit_avx512_common_gemm_f32.cpp | 2045 dnnl_status_t st_thr = sgemm_nocopy_driver(transa, transb, myM, in jit_avx512_common_gemm_f32() local
|
| H A D | jit_avx_gemm_f32.cpp | 2492 dnnl_status_t st_thr = sgemm_nocopy_driver(transa, transb, myM, in jit_avx_gemm_f32() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/f32/ |
| H A D | jit_avx512_core_gemm_smalln_tn_f32_kern.cpp | 827 auto st_thr = sgemm_smalln_tn(mpart, n, k, alpha, &A[ithr * lda * mid], in jit_avx512_core_gemm_smalln_tn_f32() local
|
| H A D | jit_avx512_common_gemm_f32.cpp | 2045 dnnl_status_t st_thr = sgemm_nocopy_driver(transa, transb, myM, in jit_avx512_common_gemm_f32() local
|
| H A D | jit_avx_gemm_f32.cpp | 2492 dnnl_status_t st_thr = sgemm_nocopy_driver(transa, transb, myM, in jit_avx_gemm_f32() local
|