Searched refs:off_a_packed (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/gemm/ |
| H A D | gen9_gemm.cpp | 347 int64_t off_a_packed in execute_standard() local
369 off_a_packed, false, !transa); in execute_standard()
403 size_n, size_k, *temp_buf, off_a_packed, in execute_standard()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/gemm/ |
| H A D | gen9_gemm.cpp | 347 int64_t off_a_packed in execute_standard() local
369 off_a_packed, false, !transa); in execute_standard()
403 size_n, size_k, *temp_buf, off_a_packed, in execute_standard()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/gemm/ |
| H A D | xe_hp_systolic_gemm.cpp | 904 auto off_a_packed = Bm * lda_packed + Bk * pd()->unroll_m(); in execute() local
905 if (packed_a) off_a_packed += off_a0; in execute()
924 off_a_packed, lda_packed, b_packed, off_b_packed, in execute()
|