|
mxnet
|
#include <mkldnn_types.h>

Public Attributes | |
| mkldnn_alg_kind_t | cell_kind |
| mkldnn_alg_kind_t | activation_kind |
| unsigned int | flags |
| float | alpha |
| float | clipping |
| mkldnn_alg_kind_t mkldnn_rnn_cell_desc_t::activation_kind |
Activation function used. Must be either mkldnn_eltwise_relu or mkldnn_eltwise_tanh.
| float mkldnn_rnn_cell_desc_t::alpha |
alpha is a negative slope parameter (used only if (flags & mkldnn_rnn_cell_with_relu) != 0)
| mkldnn_alg_kind_t mkldnn_rnn_cell_desc_t::cell_kind |
RNN cell kind. Must be one of mkldnn_vanilla_rnn, mkldnn_vanilla_lstm, mkldnn_vanilla_gru, or mkldnn_gru_linear_before_reset.
| float mkldnn_rnn_cell_desc_t::clipping |
clipping parameter (used only if (flags & mkldnn_rnn_cell_with_clipping) != 0)
| unsigned int mkldnn_rnn_cell_desc_t::flags |
RNN cell flags
1.8.11