Descriptor for a layer normalization forward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::layer_normalization_forward::desc:

[legend]

Public Member Functions
	desc (prop_kind aprop_kind, const memory::desc &data_desc, const memory::desc &stat_desc, float epsilon, normalization_flags flags)
	Constructs a descriptor for layer normalization forward propagation primitive. More...

	desc (prop_kind aprop_kind, const memory::desc &data_desc, float epsilon, normalization_flags flags)
	Constructs a descriptor for layer normalization forward propagation primitive. More...

Detailed Description

Descriptor for a layer normalization forward propagation primitive.

Constructor & Destructor Documentation

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

inline

Constructs a descriptor for layer normalization forward propagation primitive.

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 descriptor.
stat_desc	Statistics memory descriptors.
epsilon	Layer normalization epsilon parameter.
flags	Layer normalization flags (dnnl::normalization_flags).

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

inline

Constructs a descriptor for layer normalization forward propagation primitive.

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 descriptor.
epsilon	Layer normalization epsilon parameter.
flags	Layer normalization flags (dnnl::normalization_flags).

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