struct dnnl::lrn_forward::desc
Overview
Descriptor for an LRN forward propagation primitive. More…
#include <dnnl.hpp> struct desc { // fields dnnl_lrn_desc_t data; // construction desc( prop_kind aprop_kind, algorithm aalgorithm, const memory::desc& data_desc, memory::dim local_size, float alpha, float beta, float k = 1.f ); };
Detailed Documentation
Descriptor for an LRN forward propagation primitive.
Construction
desc( prop_kind aprop_kind, algorithm aalgorithm, const memory::desc& data_desc, memory::dim local_size, float alpha, float beta, float k = 1.f )
Constructs a descriptor for a LRN forward propagation primitive.
Parameters:
aprop_kind | Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference. |
aalgorithm | LRN algorithm kind: either dnnl::algorithm::lrn_across_channels, or dnnl::algorithm::lrn_within_channel. |
data_desc | Source and destination memory descriptors. |
local_size | Regularization local size. |
alpha | The alpha regularization parameter. |
beta | The beta regularization parameter. |
k | The k regularization parameter. |