Searched refs:rnn_common_fwd_desc_init (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | rnn.cpp | 246 status_t rnn_common_fwd_desc_init(dnnl_rnn_desc_t *rnn_desc, in rnn_common_fwd_desc_init() function
460 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_vanilla_rnn_forward_desc_init()
516 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_lstm_forward_desc_init_v3()
533 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_gru_forward_desc_init()
549 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, dnnl_lbr_gru, in dnnl_lbr_gru_forward_desc_init()
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | rnn.cpp | 247 status_t rnn_common_fwd_desc_init(dnnl_rnn_desc_t *rnn_desc, in rnn_common_fwd_desc_init() function
461 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_vanilla_rnn_forward_desc_init()
517 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_lstm_forward_desc_init_v3()
534 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, in dnnl_gru_forward_desc_init()
550 status_t st = rnn_common_fwd_desc_init(rnn_desc, prop_kind, dnnl_lbr_gru, in dnnl_lbr_gru_forward_desc_init()
|