| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn_aux.cpp | 313 const dnnl_memory_desc_t *diff_weights_layer_d, in init_rnn_bwd_desc() argument
330 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
340 diff_src_iter_c_d, diff_weights_layer_d, in init_rnn_bwd_desc()
349 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
357 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
|
| H A D | rnn_aux.hpp | 69 const dnnl_memory_desc_t *diff_weights_layer_d,
|
| H A D | rnn.cpp | 573 diff_src_iter_d, diff_src_iter_c_d, diff_weights_layer_d, in init_pd() local
694 DNN_SAFE(dnnl_memory_desc_init_by_tag(&diff_weights_layer_d, 5, in init_pd()
736 &diff_weights_layer_d, &diff_weights_iter_d, in init_pd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn_aux.cpp | 313 const dnnl_memory_desc_t *diff_weights_layer_d, in init_rnn_bwd_desc() argument
330 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
340 diff_src_iter_c_d, diff_weights_layer_d, in init_rnn_bwd_desc()
349 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
357 diff_src_layer_d, diff_src_iter_d, diff_weights_layer_d, in init_rnn_bwd_desc()
|
| H A D | rnn_aux.hpp | 69 const dnnl_memory_desc_t *diff_weights_layer_d,
|
| H A D | rnn.cpp | 573 diff_src_iter_d, diff_src_iter_c_d, diff_weights_layer_d, in init_pd() local
694 DNN_SAFE(dnnl_memory_desc_init_by_tag(&diff_weights_layer_d, 5, in init_pd()
736 &diff_weights_layer_d, &diff_weights_iter_d, in init_pd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 845 const memory_desc_wrapper &diff_weights_layer_d, in set_conf() argument
875 set_dims(diff_weights_layer_d, rnn.diff_weights_layer_ld, in set_conf()
884 assert(IMPLICATION(diff_weights_layer_d.ndims() != 0, in set_conf()
885 (diff_weights_layer_d.data_type() in set_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | rnn_utils.hpp | 882 const memory_desc_wrapper &diff_weights_layer_d, in set_conf() argument
913 set_dims(diff_weights_layer_d, rnn.diff_weights_layer_ld, in set_conf()
922 assert(IMPLICATION(diff_weights_layer_d.ndims() != 0, in set_conf()
923 (diff_weights_layer_d.data_type() in set_conf()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
| H A D | rnn_utils.hpp | 188 const memory_desc_wrapper &diff_weights_layer_d,
|
| H A D | rnn_utils.cpp | 201 const memory_desc_wrapper &diff_weights_layer_d, in set_rnn_conf() argument
222 set_dims(diff_weights_layer_d, rnn.diff_weights_layer_ld, in set_rnn_conf()
|
| H A D | ref_rnn.cpp | 77 const memory_desc_wrapper &diff_weights_layer_d, in init_conf() argument
159 conf.diff_weights_layer_ndims = diff_weights_layer_d.ndims(); in init_conf()
171 gpu::set_offsets(diff_weights_layer_d, off.diff_weights_layer_off); in init_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
| H A D | rnn_utils.cpp | 201 const memory_desc_wrapper &diff_weights_layer_d, in set_rnn_conf() argument
222 set_dims(diff_weights_layer_d, rnn.diff_weights_layer_ld, in set_rnn_conf()
|
| H A D | rnn_utils.hpp | 210 const memory_desc_wrapper &diff_weights_layer_d,
|
| H A D | ref_rnn.cpp | 77 const memory_desc_wrapper &diff_weights_layer_d, in init_conf() argument
159 conf.diff_weights_layer_ndims = diff_weights_layer_d.ndims(); in init_conf()
171 gpu::set_offsets(diff_weights_layer_d, off.diff_weights_layer_off); in init_conf()
|