Searched refs:diff_weights_projection_d (Results 1 – 8 of 8) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn_aux.hpp | 72 const dnnl_memory_desc_t *diff_weights_projection_d,
|
| H A D | rnn_aux.cpp | 316 const dnnl_memory_desc_t *diff_weights_projection_d, in init_rnn_bwd_desc() argument
342 diff_weights_projection_d, diff_bias_d, diff_dst_layer_d, in init_rnn_bwd_desc()
|
| H A D | rnn.cpp | 575 diff_weights_projection_d {}, diff_bias_d, diff_dst_layer_d, in init_pd() local
709 DNN_SAFE(dnnl_memory_desc_init_by_tag(&diff_weights_projection_d, 4, in init_pd()
737 &diff_weights_peephole_d, &diff_weights_projection_d, in init_pd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn_aux.hpp | 72 const dnnl_memory_desc_t *diff_weights_projection_d,
|
| H A D | rnn_aux.cpp | 316 const dnnl_memory_desc_t *diff_weights_projection_d, in init_rnn_bwd_desc() argument
342 diff_weights_projection_d, diff_bias_d, diff_dst_layer_d, in init_rnn_bwd_desc()
|
| H A D | rnn.cpp | 575 diff_weights_projection_d {}, diff_bias_d, diff_dst_layer_d, in init_pd() local
709 DNN_SAFE(dnnl_memory_desc_init_by_tag(&diff_weights_projection_d, 4, in init_pd()
737 &diff_weights_peephole_d, &diff_weights_projection_d, in init_pd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 847 const memory_desc_wrapper &diff_weights_projection_d) { in set_conf() argument
879 set_dims(diff_weights_projection_d, rnn.diff_weights_projection_ld, in set_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 884 const memory_desc_wrapper &diff_weights_projection_d) { in set_conf() argument
917 set_dims(diff_weights_projection_d, rnn.diff_weights_projection_ld, in set_conf()
|