mkldnn_rnn_desc_t Struct Reference

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
mkldnn_prop_kind_t	prop_kind
mkldnn_alg_kind_t	cell_kind
mkldnn_rnn_direction_t	direction
	The direction of RNN primitive execution. More...
mkldnn_memory_desc_t	src_layer_desc
	Source layer memory descriptor. More...
mkldnn_memory_desc_t	src_iter_desc
	Source iteration memory descriptor for hidden state. More...
mkldnn_memory_desc_t	src_iter_c_desc
	Source iteration memory descriptor for cell state. More...
mkldnn_memory_desc_t	weights_layer_desc
	Weights layer memory descriptor. More...
mkldnn_memory_desc_t	weights_iter_desc
	Weights iteration memory descriptor. More...
mkldnn_memory_desc_t	bias_desc
	Bias memory descriptor. More...
mkldnn_memory_desc_t	dst_layer_desc
	Destination layer memory descriptor. More...
mkldnn_memory_desc_t	dst_iter_desc
	Destination iter memory descriptor for hidden state. More...
mkldnn_memory_desc_t	dst_iter_c_desc
	Destination iter memory descriptor for cell state. More...
mkldnn_memory_desc_t	placeholder_desc
	Placeholders. More...
mkldnn_memory_desc_t	placeholder2_desc
mkldnn_memory_desc_t	diff_src_layer_desc
	Source gradient layer memory descriptor. More...
mkldnn_memory_desc_t	diff_src_iter_desc
	Source gradient iter memory descriptor for hidden state. More...
mkldnn_memory_desc_t	diff_src_iter_c_desc
	Source gradient iter memory descriptor for cell state. More...
mkldnn_memory_desc_t	diff_weights_layer_desc
	Weights gradient layer memory descriptor. More...
mkldnn_memory_desc_t	diff_weights_iter_desc
	Weights gradient iter memory descriptor. More...
mkldnn_memory_desc_t	diff_bias_desc
	Bias gradient memory descriptor. More...
mkldnn_memory_desc_t	diff_dst_layer_desc
	Destination gradient layer memory descriptor. More...
mkldnn_memory_desc_t	diff_dst_iter_desc
	Destination gradient iteration memory descriptor for hidden state. More...
mkldnn_memory_desc_t	diff_dst_iter_c_desc
	Destination gradient iteration memory descriptor for cell state. More...
mkldnn_memory_desc_t	diff_placeholder_desc
	Placeholders. More...
mkldnn_memory_desc_t	diff_placeholder2_desc
unsigned int	flags
	RNN cell flags. More...
mkldnn_alg_kind_t	activation_kind
float	alpha
float	beta

Detailed Description

A descriptor for an RNN operation.

Member Data Documentation

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.

float mkldnn_rnn_desc_t::alpha

float mkldnn_rnn_desc_t::beta

mkldnn_memory_desc_t mkldnn_rnn_desc_t::bias_desc

Bias memory descriptor.

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.

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_iter_c_desc

Destination gradient iteration memory descriptor for cell state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_dst_iter_desc

Destination gradient iteration memory descriptor for hidden state.

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_placeholder2_desc

mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_placeholder_desc

Placeholders.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_src_iter_c_desc

Source gradient iter memory descriptor for cell state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_src_iter_desc

Source gradient iter memory descriptor for hidden state.

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_weights_iter_desc

Weights gradient iter memory descriptor.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::diff_weights_layer_desc

Weights gradient layer memory descriptor.

mkldnn_rnn_direction_t mkldnn_rnn_desc_t::direction

The direction of RNN primitive execution.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::dst_iter_c_desc

Destination iter memory descriptor for cell state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::dst_iter_desc

Destination iter memory descriptor for hidden state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::dst_layer_desc

Destination layer memory descriptor.

unsigned int mkldnn_rnn_desc_t::flags

RNN cell flags.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::placeholder2_desc

mkldnn_memory_desc_t mkldnn_rnn_desc_t::placeholder_desc

Placeholders.

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_memory_desc_t mkldnn_rnn_desc_t::src_iter_c_desc

Source iteration memory descriptor for cell state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::src_iter_desc

Source iteration memory descriptor for hidden state.

mkldnn_memory_desc_t mkldnn_rnn_desc_t::src_layer_desc

Source 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::weights_layer_desc

Weights layer memory descriptor.

---

The documentation for this struct was generated from the following file:

• /work/mxnet/include/mkldnn/mkldnn_types.h