| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | cell_gru_lbr.cpp | 151 rnn.ws_gates_ld, B, ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
185 rnn.ws_gates_ld, B, ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | cell_gru.cpp | 192 ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
228 ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | cell_common.cpp | 239 rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
286 rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | rnn_utils.hpp | 268 int diff_weights_iter_ld = 0, diff_weights_iter_nld = 0; member
877 set_dims(diff_weights_iter_d, rnn.diff_weights_iter_ld, in set_conf()
1134 data, rnn.diff_weights_iter_nld, rnn.diff_weights_iter_ld) in ws_diff_w_iter_aoc_t()
|
| H A D | ref_rnn.cpp | 181 rnn.n_dir, rnn.diff_weights_iter_nld * rnn.diff_weights_iter_ld);
444 rnn.diff_weights_iter_ld));
456 rnn.diff_weights_iter_ld));
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | cell_gru_lbr.cpp | 151 rnn.ws_gates_ld, B, ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
187 rnn.ws_gates_ld, B, ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | cell_gru.cpp | 192 ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
228 ldb, 1.0f, C, rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | cell_common.cpp | 248 rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
298 rnn.diff_weights_iter_ld); in rnn_cell_execution_sig()
|
| H A D | rnn_utils.hpp | 278 int diff_weights_iter_ld = 0, diff_weights_iter_nld = 0; member
915 set_dims(diff_weights_iter_d, rnn.diff_weights_iter_ld, in set_conf()
1248 data, rnn.diff_weights_iter_nld, rnn.diff_weights_iter_ld) in ws_diff_w_iter_aoc_t()
|
| H A D | ref_rnn.cpp | 183 rnn.n_dir, rnn.diff_weights_iter_nld * rnn.diff_weights_iter_ld);
448 rnn.diff_weights_iter_ld));
460 rnn.diff_weights_iter_ld));
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
| H A D | rnn_utils.hpp | 140 int diff_weights_iter_ld, diff_weights_iter_nld; member
|
| H A D | rnn_utils.cpp | 224 set_dims(diff_weights_iter_d, rnn.diff_weights_iter_ld, in set_rnn_conf()
497 rnn.diff_weights_iter_nld * rnn.diff_weights_iter_ld) in set_offsets_bwd_gemm()
|
| H A D | ref_rnn.cpp | 716 rnn_conf.diff_weights_iter_ld, in init()
724 rnn_conf.diff_weights_iter_ld, in init()
734 rnn_conf.diff_weights_iter_ld, in init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
| H A D | rnn_utils.cpp | 224 set_dims(diff_weights_iter_d, rnn.diff_weights_iter_ld, in set_rnn_conf()
482 rnn.diff_weights_iter_nld * rnn.diff_weights_iter_ld) in set_offsets_bwd_gemm()
|
| H A D | rnn_utils.hpp | 163 int diff_weights_iter_ld, diff_weights_iter_nld; member
|
| H A D | ref_rnn.cpp | 716 rnn_conf.diff_weights_iter_ld, in init()
724 rnn_conf.diff_weights_iter_ld, in init()
734 rnn_conf.diff_weights_iter_ld, in init()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 711 diff_wei_conf.LDC_iter = rnn.diff_weights_iter_ld; in configure_brgemm()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
| H A D | rnn_brgemm_utils.cpp | 784 diff_wei_conf.LDC_iter = rnn.diff_weights_iter_ld; in configure_brgemm()
|