Searched refs:weights_projection_m (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | ref_rnn_bwd.cpp | 328 dnn_mem_t &weights_projection_m, dnn_mem_t &bias_m, in compute_ref_bwd() argument
342 (float *)weights_projection_m, (float *)bias_m, in compute_ref_bwd()
349 (float *)weights_projection_m, (float *)bias_m, in compute_ref_bwd()
|
| H A D | rnn.hpp | 469 dnn_mem_t &weights_peephole_m, dnn_mem_t &weights_projection_m,
478 dnn_mem_t &weights_projection_m, dnn_mem_t &bias_m,
|
| H A D | ref_rnn_fwd.cpp | 368 dnn_mem_t &weights_peephole_m, dnn_mem_t &weights_projection_m, in compute_ref_fwd() argument
379 (float *)weights_projection_m, (float *)bias_m, in compute_ref_fwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | ref_rnn_bwd.cpp | 328 dnn_mem_t &weights_projection_m, dnn_mem_t &bias_m, in compute_ref_bwd() argument
342 (float *)weights_projection_m, (float *)bias_m, in compute_ref_bwd()
349 (float *)weights_projection_m, (float *)bias_m, in compute_ref_bwd()
|
| H A D | rnn.hpp | 465 dnn_mem_t &weights_peephole_m, dnn_mem_t &weights_projection_m,
474 dnn_mem_t &weights_projection_m, dnn_mem_t &bias_m,
|
| H A D | ref_rnn_fwd.cpp | 368 dnn_mem_t &weights_peephole_m, dnn_mem_t &weights_projection_m, in compute_ref_fwd() argument
379 (float *)weights_projection_m, (float *)bias_m, in compute_ref_fwd()
|