Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.21.0
Performance library for Deep Learning
|
A descriptor for an RNN operation. More...
#include <mkldnn_types.h>
A descriptor for an RNN operation.
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.
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.
mkldnn_rnn_cell_desc_t mkldnn_rnn_desc_t::cell_desc |
The RNN cell desc.
mkldnn_rnn_direction_t mkldnn_rnn_desc_t::direction |
The direction of RNN primitive execution.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::src_layer_desc |
Source layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::src_iter_desc |
Source iteration memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::weights_layer_desc |
Weights layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::weights_iter_desc |
Weights iteration memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::bias_desc |
Bias memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::dst_layer_desc |
Destination layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::dst_iter_desc |
Destination iter memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_src_layer_desc |
Source gradient layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_src_iter_desc |
Source gradient iter memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_weights_layer_desc |
Weights gradient layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_weights_iter_desc |
Weights gradient iter memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_bias_desc |
Bias gradient memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_dst_layer_desc |
Destination gradient layer memory descriptor.
mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_dst_iter_desc |
Destination gradient iteration memory descriptor.