A primitive to perform local response normalization (LRN) across or within channels. More...

Functions
dnnl_status_t DNNL_API	dnnl_lrn_forward_desc_init (dnnl_lrn_desc_t lrn_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t data_desc, dnnl_dim_t local_size, float alpha, float beta, float k)
	Initializes an `lrn_desc` for forward propagation using `prop_kind` (possible values are dnnl_forward_training and dnnl_forward_inference), `alg_kind`, memory descriptor `data_desc`, and regularization parameters `local_size`, `alpha`, `beta`, and `k`. More...

dnnl_status_t DNNL_API	dnnl_lrn_backward_desc_init (dnnl_lrn_desc_t lrn_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t diff_data_desc, const dnnl_memory_desc_t *data_desc, dnnl_dim_t local_size, float alpha, float beta, float k)
	Initializes an `lrn_desc` for backward propagation using `alg_kind`, memory descriptors `data_desc` and `diff_data_desc`, and regularization parameters `local_size`, `alpha`, `beta`, and `k`. More...

Detailed Description

A primitive to perform local response normalization (LRN) across or within channels.

See also: dnnl_primitive_desc_query and dnnl_primitive_desc_query_pd; Local Response Normalization (LRN) in developer guide; LRN in C++ API

Function Documentation

◆ dnnl_lrn_forward_desc_init()

dnnl_status_t DNNL_API dnnl_lrn_forward_desc_init	(	dnnl_lrn_desc_t *	lrn_desc,
		dnnl_prop_kind_t	prop_kind,
		dnnl_alg_kind_t	alg_kind,
		const dnnl_memory_desc_t *	data_desc,
		dnnl_dim_t	local_size,
		float	alpha,
		float	beta,
		float	k
	)

Initializes an lrn_desc for forward propagation using prop_kind (possible values are dnnl_forward_training and dnnl_forward_inference), alg_kind, memory descriptor data_desc, and regularization parameters local_size, alpha, beta, and k.

Inputs:

src (dnnl_query_src_md, 0)

Outputs:

dst (dnnl_query_dst_md, 0)
workspace (dnnl_query_workspace_md, 0), if the underlying implementation requires

Examples:: cnn_inference_f32.c, and cpu_cnn_training_f32.c.

◆ dnnl_lrn_backward_desc_init()

dnnl_status_t DNNL_API dnnl_lrn_backward_desc_init	(	dnnl_lrn_desc_t *	lrn_desc,
		dnnl_alg_kind_t	alg_kind,
		const dnnl_memory_desc_t *	diff_data_desc,
		const dnnl_memory_desc_t *	data_desc,
		dnnl_dim_t	local_size,
		float	alpha,
		float	beta,
		float	k
	)

Initializes an lrn_desc for backward propagation using alg_kind, memory descriptors data_desc and diff_data_desc, and regularization parameters local_size, alpha, beta, and k.