A descriptor for an RNN operation. More...
#include <mkldnn_types.h>
Collaboration diagram for mkldnn_rnn_desc_t:
Public Attributes | |
| mkldnn_primitive_kind_t | primitive_kind |
| The kind of primitive. More... | |
| mkldnn_prop_kind_t | prop_kind |
| The kind of propagation. More... | |
| mkldnn_alg_kind_t | cell_kind |
| RNN cell kind. More... | |
| mkldnn_rnn_direction_t | direction |
| The direction of RNN primitive execution. | |
| mkldnn_memory_desc_t | src_layer_desc |
| Source layer memory descriptor. | |
| mkldnn_memory_desc_t | src_iter_desc |
| Source iteration memory descriptor for hidden state. | |
| mkldnn_memory_desc_t | src_iter_c_desc |
| Source iteration memory descriptor for cell state. | |
| mkldnn_memory_desc_t | weights_layer_desc |
| Weights layer memory descriptor. | |
| mkldnn_memory_desc_t | weights_iter_desc |
| Weights iteration memory descriptor. | |
| mkldnn_memory_desc_t | bias_desc |
| Bias memory descriptor. | |
| mkldnn_memory_desc_t | dst_layer_desc |
| Destination layer memory descriptor. | |
| mkldnn_memory_desc_t | dst_iter_desc |
| Destination iter memory descriptor for hidden state. | |
| mkldnn_memory_desc_t | dst_iter_c_desc |
| Destination iter memory descriptor for cell state. | |
| mkldnn_memory_desc_t | placeholder_desc |
| Placeholders. | |
| mkldnn_memory_desc_t | diff_src_layer_desc |
| Source gradient layer memory descriptor. | |
| mkldnn_memory_desc_t | diff_src_iter_desc |
| Source gradient iter memory descriptor for hidden state. | |
| mkldnn_memory_desc_t | diff_src_iter_c_desc |
| Source gradient iter memory descriptor for cell state. | |
| mkldnn_memory_desc_t | diff_weights_layer_desc |
| Weights gradient layer memory descriptor. | |
| mkldnn_memory_desc_t | diff_weights_iter_desc |
| Weights gradient iter memory descriptor. | |
| mkldnn_memory_desc_t | diff_bias_desc |
| Bias gradient memory descriptor. | |
| mkldnn_memory_desc_t | diff_dst_layer_desc |
| Destination gradient layer memory descriptor. | |
| mkldnn_memory_desc_t | diff_dst_iter_desc |
| Destination gradient iteration memory descriptor for hidden state. | |
| mkldnn_memory_desc_t | diff_dst_iter_c_desc |
| Destination gradient iteration memory descriptor for cell state. | |
| mkldnn_memory_desc_t | diff_placeholder_desc |
| Placeholders. | |
| unsigned int | flags |
| RNN cell flags. | |
| mkldnn_alg_kind_t | activation_kind |
| Activation function used for vanilla_rnn cell kind. More... | |
Detailed Description
A descriptor for an RNN operation.
Member Data Documentation
◆ primitive_kind
| mkldnn_primitive_kind_t mkldnn_rnn_desc_t::primitive_kind |
The kind of primitive.
Used for self-identifying the primitive descriptor. Must be mkldnn_rnn.
◆ prop_kind
| mkldnn_prop_kind_t mkldnn_rnn_desc_t::prop_kind |
The kind of propagation.
Possible values: mkldnn_forward_training, mkldnn_forward_inference, and mkldnn_backward.
◆ cell_kind
| mkldnn_alg_kind_t mkldnn_rnn_desc_t::cell_kind |
RNN cell kind.
Must be one of mkldnn_vanilla_rnn, mkldnn_vanilla_lstm, mkldnn_vanilla_gru, or mkldnn_lbr_gru.
◆ activation_kind
| mkldnn_alg_kind_t mkldnn_rnn_desc_t::activation_kind |
Activation function used for vanilla_rnn cell kind.
Must be either mkldnn_eltwise_relu or mkldnn_eltwise_tanh.
The documentation for this struct was generated from the following file:
- include/mkldnn_types.h