[−][src]Struct onednn_sys::dnnl_rnn_desc_t

#[repr(C)]pub struct dnnl_rnn_desc_t {
    pub primitive_kind: dnnl_primitive_kind_t,
    pub prop_kind: dnnl_prop_kind_t,
    pub cell_kind: dnnl_alg_kind_t,
    pub direction: dnnl_rnn_direction_t,
    pub src_layer_desc: dnnl_memory_desc_t,
    pub src_iter_desc: dnnl_memory_desc_t,
    pub src_iter_c_desc: dnnl_memory_desc_t,
    pub weights_layer_desc: dnnl_memory_desc_t,
    pub weights_iter_desc: dnnl_memory_desc_t,
    pub bias_desc: dnnl_memory_desc_t,
    pub dst_layer_desc: dnnl_memory_desc_t,
    pub dst_iter_desc: dnnl_memory_desc_t,
    pub dst_iter_c_desc: dnnl_memory_desc_t,
    pub weights_peephole_desc: dnnl_memory_desc_t,
    pub weights_projection_desc: dnnl_memory_desc_t,
    pub diff_src_layer_desc: dnnl_memory_desc_t,
    pub diff_src_iter_desc: dnnl_memory_desc_t,
    pub diff_src_iter_c_desc: dnnl_memory_desc_t,
    pub diff_weights_layer_desc: dnnl_memory_desc_t,
    pub diff_weights_iter_desc: dnnl_memory_desc_t,
    pub diff_bias_desc: dnnl_memory_desc_t,
    pub diff_dst_layer_desc: dnnl_memory_desc_t,
    pub diff_dst_iter_desc: dnnl_memory_desc_t,
    pub diff_dst_iter_c_desc: dnnl_memory_desc_t,
    pub diff_weights_peephole_desc: dnnl_memory_desc_t,
    pub diff_weights_projection_desc: dnnl_memory_desc_t,
    pub flags: c_uint,
    pub activation_kind: dnnl_alg_kind_t,
    pub alpha: f32,
    pub beta: f32,
}

A descriptor for an RNN operation.

Fields

primitive_kind: dnnl_primitive_kind_t

The kind of primitive. Used for self-identifying the primitive descriptor. Must be #dnnl_rnn.

prop_kind: dnnl_prop_kind_t

The kind of propagation. Possible values: #dnnl_forward_training, #dnnl_forward_inference, and #dnnl_backward.

cell_kind: dnnl_alg_kind_t

RNN cell kind. Must be one of #dnnl_vanilla_rnn, #dnnl_vanilla_lstm, #dnnl_vanilla_gru, or #dnnl_lbr_gru.

direction: dnnl_rnn_direction_t

The direction of RNN primitive execution.

src_layer_desc: dnnl_memory_desc_t

Source layer memory descriptor.

src_iter_desc: dnnl_memory_desc_t

Source iteration memory descriptor for hidden state.

src_iter_c_desc: dnnl_memory_desc_t

Source iteration memory descriptor for cell state.

weights_layer_desc: dnnl_memory_desc_t

Weights layer memory descriptor.

weights_iter_desc: dnnl_memory_desc_t

Weights iteration memory descriptor.

bias_desc: dnnl_memory_desc_t

Bias memory descriptor.

dst_layer_desc: dnnl_memory_desc_t

Destination layer memory descriptor.

dst_iter_desc: dnnl_memory_desc_t

Destination iter memory descriptor for hidden state.

dst_iter_c_desc: dnnl_memory_desc_t

Destination iter memory descriptor for cell state.

weights_peephole_desc: dnnl_memory_desc_t

Weights peephole memory descriptor. This memory descriptor is equal to zero memory descriptor in case of non-peephole LSTMs and other non-LSTM RNNs.

weights_projection_desc: dnnl_memory_desc_t

Weights projection memory descriptor. This memory descriptor is equal to zero memory descriptor in case of non-projection LSTMs and other non-LSTM RNNs.

diff_src_layer_desc: dnnl_memory_desc_t

Source gradient layer memory descriptor.

diff_src_iter_desc: dnnl_memory_desc_t

Source gradient iter memory descriptor for hidden state.

diff_src_iter_c_desc: dnnl_memory_desc_t

Source gradient iter memory descriptor for cell state.

diff_weights_layer_desc: dnnl_memory_desc_t

Weights gradient layer memory descriptor.

diff_weights_iter_desc: dnnl_memory_desc_t

Weights gradient iter memory descriptor.

diff_bias_desc: dnnl_memory_desc_t

Bias gradient memory descriptor.

diff_dst_layer_desc: dnnl_memory_desc_t

Destination gradient layer memory descriptor.

diff_dst_iter_desc: dnnl_memory_desc_t

Destination gradient iteration memory descriptor for hidden state.

diff_dst_iter_c_desc: dnnl_memory_desc_t

Destination gradient iteration memory descriptor for cell state.

diff_weights_peephole_desc: dnnl_memory_desc_t

Weights gradient peephole memory descriptor. This memory descriptor is equal to zero memory descriptor in case of non-peephole LSTMs and other non-LSTM RNNs.

diff_weights_projection_desc: dnnl_memory_desc_t

Weights gradient projection memory descriptor. This memory descriptor is equal to zero memory descriptor in case of non-projection LSTMs and other non-LSTM RNNs.

flags: c_uint

RNN cell flags

activation_kind: dnnl_alg_kind_t

Activation function used for vanilla_rnn cell kind. Must be either #dnnl_eltwise_relu or #dnnl_eltwise_tanh.

alpha: f32beta: f32

Trait Implementations

impl Clone for dnnl_rnn_desc_t[src]

impl Copy for dnnl_rnn_desc_t[src]

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, [src]

impl<T> ToOwned for T where
    T: Clone, [src]

type Owned = T

The resulting type after obtaining ownership.

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.