struct dnnl::layer_normalization_backward::desc

Overview

Descriptor for a layer normalization backward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_layer_normalization_desc_t data;

    // construction

    desc(
        prop_kind aprop_kind,
        const memory::desc& diff_data_desc,
        const memory::desc& data_desc,
        const memory::desc& stat_desc,
        float epsilon,
        normalization_flags flags
        );

    desc(
        prop_kind aprop_kind,
        const memory::desc& diff_data_desc,
        const memory::desc& data_desc,
        float epsilon,
        normalization_flags flags
        );
};

Detailed Documentation

Descriptor for a layer normalization backward propagation primitive.

Construction

desc(
    prop_kind aprop_kind,
    const memory::desc& diff_data_desc,
    const memory::desc& data_desc,
    const memory::desc& stat_desc,
    float epsilon,
    normalization_flags flags
    )

Constructs a descriptor for layer normalization backward propagation primitive.

Parameters:

aprop_kind

Propagation kind. Possible values are dnnl::prop_kind::backward_data and dnnl::prop_kind::backward (diffs for all parameters are computed in this case).

diff_data_desc

Diff source and diff destination memory descriptor.

data_desc

Source memory descriptor.

stat_desc

Statistics memory descriptors.

epsilon

Layer normalization epsilon parameter.

flags

Layer normalization flags (dnnl::normalization_flags).

desc(
    prop_kind aprop_kind,
    const memory::desc& diff_data_desc,
    const memory::desc& data_desc,
    float epsilon,
    normalization_flags flags
    )

Constructs a descriptor for layer normalization backward propagation primitive.

Parameters:

aprop_kind

Propagation kind. Possible values are dnnl::prop_kind::backward_data and dnnl::prop_kind::backward (diffs for all parameters are computed in this case).

diff_data_desc

Diff source and diff destination memory descriptor.

data_desc

Source memory descriptor.

epsilon

Layer normalization epsilon parameter.

flags

Layer normalization flags (dnnl::normalization_flags).