Collaboration diagram for Dnnl_api_batch_normalization:

Classes
struct	dnnl_batch_normalization_desc_t
	A descriptor of a Batch Normalization operation. More...

Functions
dnnl_status_t DNNL_API	dnnl_batch_normalization_forward_desc_init (dnnl_batch_normalization_desc_t bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t data_desc, float epsilon, unsigned flags)

dnnl_status_t DNNL_API	dnnl_batch_normalization_backward_desc_init (dnnl_batch_normalization_desc_t bnrm_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, float epsilon, unsigned flags)

Detailed Description

Function Documentation

dnnl_status_t DNNL_API dnnl_batch_normalization_backward_desc_init	(	dnnl_batch_normalization_desc_t *	bnrm_desc,
		dnnl_prop_kind_t	prop_kind,
		const dnnl_memory_desc_t *	diff_data_desc,
		const dnnl_memory_desc_t *	data_desc,
		float	epsilon,
		unsigned	flags
	)

Initializes a descriptor for a batch normalization backward propagation primitive.

Note: In-place operation is supported: the diff_dst can refer to the same memory as the diff_src.

Parameters

bnrm_desc	Output descriptor for batch normalization primitive.
prop_kind	Propagation kind. Possible values are dnnl_backward_data and dnnl_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	Batch normalization epsilon parameter.
flags	Batch normalization flags (dnnl_normalization_flags_t).

Returns: dnnl_success on success and a status describing the error otherwise.

dnnl_status_t DNNL_API dnnl_batch_normalization_forward_desc_init	(	dnnl_batch_normalization_desc_t *	bnrm_desc,
		dnnl_prop_kind_t	prop_kind,
		const dnnl_memory_desc_t *	data_desc,
		float	epsilon,
		unsigned	flags
	)

Initializes a descriptor for a batch normalization forward propagation primitive.

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

Parameters

bnrm_desc	Output descriptor for batch normalization primitive.
prop_kind	Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference.
data_desc	Source and destination memory descriptor.
epsilon	Batch normalization epsilon parameter.
flags	Batch normalization flags (dnnl_normalization_flags_t).

Returns: dnnl_success on success and a status describing the error otherwise.