Searched refs:kproj_tail (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 341 rnn.kproj_tail = 0; in configure_brgemm()
356 rnn.kproj_tail = rnn.Kproj % rnn.kproj_block; in configure_brgemm()
358 if ((rnn.kproj_tail % padding) || (rnn.kproj_block % padding)) { in configure_brgemm()
360 rnn.kproj_tail = 0; in configure_brgemm()
507 if (rnn.kproj_tail) in init_kernels()
510 rnn.kproj_tail, rnn.LDAproj, rnn.LDBproj, in init_kernels()
512 if (rnn.kproj_tail && rnn.nproj_tail) in init_kernels()
515 brgemm_np_tail, rnn.kproj_tail, rnn.LDAproj, in init_kernels()
545 if (rnn.kproj_tail) in init_kernels()
548 if (rnn.kproj_tail && rnn.nproj_tail) in init_kernels()
[all …]
|
| H A D | brgemm_cell_common_fwd.cpp | 335 if (rnn_.kproj_tail) { in kernel()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 375 rnn.kproj_tail = 0; in configure_brgemm()
390 rnn.kproj_tail = rnn.Kproj % rnn.kproj_block; in configure_brgemm()
392 if ((rnn.kproj_tail % padding) || (rnn.kproj_block % padding)) { in configure_brgemm()
394 rnn.kproj_tail = 0; in configure_brgemm()
567 if (rnn.kproj_tail) in init_kernels()
570 rnn.kproj_tail, rnn.LDAproj, rnn.LDBproj, in init_kernels()
572 if (rnn.kproj_tail && rnn.nproj_tail) in init_kernels()
575 brgemm_np_tail, rnn.kproj_tail, rnn.LDAproj, in init_kernels()
608 if (rnn.kproj_tail) in init_kernels()
611 if (rnn.kproj_tail && rnn.nproj_tail) in init_kernels()
[all …]
|
| H A D | brgemm_cell_common_fwd.cpp | 472 if (rnn_.kproj_tail) { in kernel()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 493 dim_t kproj_block, KBproj_blocks, kproj_tail; member
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 503 dim_t kproj_block, KBproj_blocks, kproj_tail; member
|