oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
dnnl::lrn_backward::desc Struct Reference

Descriptor for an LRN backward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::lrn_backward::desc:

Public Member Functions

 desc (algorithm aalgorithm, const memory::desc &data_desc, const memory::desc &diff_data_desc, memory::dim local_size, float alpha, float beta, float k=1.f)
 Constructs a descriptor for an LRN backward propagation primitive. More...
 

Detailed Description

Descriptor for an LRN backward propagation primitive.

Examples:
cnn_training_bf16.cpp, and cnn_training_f32.cpp.

Constructor & Destructor Documentation

◆ desc()

dnnl::lrn_backward::desc::desc ( algorithm  aalgorithm,
const memory::desc data_desc,
const memory::desc diff_data_desc,
memory::dim  local_size,
float  alpha,
float  beta,
float  k = 1.f 
)
inline

Constructs a descriptor for an LRN backward propagation primitive.

Parameters
aalgorithmLRN algorithm kind: either dnnl::algorithm::lrn_across_channels, or dnnl::algorithm::lrn_within_channel.
diff_data_descDiff source and diff destination memory descriptor.
data_descSource memory descriptor.
local_sizeRegularization local size.
alphaThe alpha regularization parameter.
betaThe beta regularization parameter.
kThe k regularization parameter.

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