struct dnnl_lrn_desc_t¶
Overview¶
A descriptor of a Local Response Normalization (LRN) operation. More…
#include <dnnl_types.h> struct dnnl_lrn_desc_t { // fields dnnl_primitive_kind_t primitive_kind; dnnl_prop_kind_t prop_kind; dnnl_alg_kind_t alg_kind; dnnl_memory_desc_t data_desc; dnnl_memory_desc_t diff_data_desc; dnnl_dim_t local_size; float lrn_alpha; float lrn_beta; float lrn_k; };
Detailed Documentation¶
A descriptor of a Local Response Normalization (LRN) operation.
Fields¶
dnnl_primitive_kind_t primitive_kind
The kind of primitive.
Used for self-identifying the primitive descriptor. Must be dnnl_lrn.
dnnl_prop_kind_t prop_kind
The kind of propagation.
Possible values: dnnl_forward_training, dnnl_forward_inference, dnnl_backward, and dnnl_backward_data.
dnnl_alg_kind_t alg_kind
LRN algorithm.
Possible values: dnnl_lrn_within_channel and dnnl_lrn_across_channels.
dnnl_memory_desc_t data_desc
Source and destination memory descriptor.
dnnl_memory_desc_t diff_data_desc
Source and destination gradient memory descriptor.
dnnl_dim_t local_size
The number of channels to sum over (for cross-channel LRN) or the side length of the square region to sum over (for within-channel LRN).
float lrn_alpha
LRN alpha parameter.
float lrn_beta
LRN beta parameter.
float lrn_k
LRN k parameter.