[−][src]Struct onednn_sys::dnnl_layer_normalization_desc_t

#[repr(C)]pub struct dnnl_layer_normalization_desc_t {
    pub primitive_kind: dnnl_primitive_kind_t,
    pub prop_kind: dnnl_prop_kind_t,
    pub data_desc: dnnl_memory_desc_t,
    pub diff_data_desc: dnnl_memory_desc_t,
    pub data_scaleshift_desc: dnnl_memory_desc_t,
    pub diff_data_scaleshift_desc: dnnl_memory_desc_t,
    pub stat_desc: dnnl_memory_desc_t,
    pub layer_norm_epsilon: f32,
    pub flags: c_uint,
}

A descriptor of a Layer Normalization operation.

Fields

primitive_kind: dnnl_primitive_kind_t

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

prop_kind: dnnl_prop_kind_t

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

data_desc: dnnl_memory_desc_t

Source and destination memory descriptor.

diff_data_desc: dnnl_memory_desc_t

Source and destination gradient memory descriptor.

data_scaleshift_desc: dnnl_memory_desc_t

Scale and shift data and gradient memory descriptors.

Scaleshift memory descriptor uses 2D #dnnl_ab format[2, normalized_dim] where 1-st dimension contains gamma parameter, 2-nd dimension contains beta parameter. Normalized_dim is equal to the last logical dimension of the data tensor across which normalization is performed.

diff_data_scaleshift_desc: dnnl_memory_desc_tstat_desc: dnnl_memory_desc_t

Mean and variance data memory descriptors.

Statistics (mean and variance) memory descriptor is the k-dimensional tensor where k is equal to data_tensor_ndims - 1 and may have any plain (stride[last_dim] == 1) user-provided format.

layer_norm_epsilon: f32

Layer normalization epsilon parameter.

flags: c_uint

Trait Implementations

impl Clone for dnnl_layer_normalization_desc_t[src]

impl Copy for dnnl_layer_normalization_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.