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.