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

Functions
mkldnn_status_t MKLDNN_API	mkldnn_lrn_forward_desc_init (mkldnn_lrn_desc_t lrn_desc, mkldnn_prop_kind_t prop_kind, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t data_desc, mkldnn_dim_t local_size, float alpha, float beta, float k)
	Initializes an `lrn_desc` for forward propagation using `prop_kind` (possible values are mkldnn_forward_training and mkldnn_forward_inference), `alg_kind`, memory descriptor `data_desc`, and regularization parameters `local_size`, `alpha`, `beta`, and `k`. More...

mkldnn_status_t MKLDNN_API	mkldnn_lrn_backward_desc_init (mkldnn_lrn_desc_t lrn_desc, mkldnn_alg_kind_t alg_kind, const mkldnn_memory_desc_t diff_data_desc, const mkldnn_memory_desc_t *data_desc, mkldnn_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: mkldnn_primitive_desc_query and mkldnn_primitive_desc_query_pd; Local Response Normalization (LRN) in developer guide; LRN in C++ API

Function Documentation

◆ mkldnn_lrn_forward_desc_init()

mkldnn_status_t MKLDNN_API mkldnn_lrn_forward_desc_init	(	mkldnn_lrn_desc_t *	lrn_desc,
		mkldnn_prop_kind_t	prop_kind,
		mkldnn_alg_kind_t	alg_kind,
		const mkldnn_memory_desc_t *	data_desc,
		mkldnn_dim_t	local_size,
		float	alpha,
		float	beta,
		float	k
	)

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

Inputs:

src (mkldnn_query_src_md, 0)

Outputs:

dst (mkldnn_query_dst_md, 0)
workspace (mkldnn_query_workspace_md, 0), if the underlying implementation requires

Examples:: cpu_cnn_inference_f32.c, and cpu_cnn_training_f32.c.

◆ mkldnn_lrn_backward_desc_init()

mkldnn_status_t MKLDNN_API mkldnn_lrn_backward_desc_init	(	mkldnn_lrn_desc_t *	lrn_desc,
		mkldnn_alg_kind_t	alg_kind,
		const mkldnn_memory_desc_t *	diff_data_desc,
		const mkldnn_memory_desc_t *	data_desc,
		mkldnn_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.