.rst .pdf

Contents

struct dnnl::batch_normalization_forward::desc

Overview

Descriptor for a batch normalization forward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_batch_normalization_desc_t data;

    // construction

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

Detailed Documentation

Descriptor for a batch normalization forward propagation primitive.

Construction

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

Constructs a batch normalization descriptor for forward propagation.

Note

In-place operation is supported: the dst can refer to the same memory as the src.

Parameters:

aprop_kind	Propagation kind. Possible values are dnnl::prop_kind::forward_training and dnnl::prop_kind::forward_inference.
data_desc	Source and destination memory descriptors.
epsilon	Batch normalization epsilon parameter.
flags	Batch normalization flags (dnnl::normalization_flags).

struct dnnl::batch_normalization_forward struct dnnl::batch_normalization_forward::primitive_desc