struct dnnl::prelu_forward::desc

Overview

Descriptor for a PReLU forward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_prelu_desc_t data;

    // methods

    desc(
        prop_kind aprop_kind,
        const memory::desc& data_desc,
        const memory::desc& weight_desc
        );
};

Detailed Documentation

Descriptor for a PReLU forward propagation primitive.

Methods

desc(
    prop_kind aprop_kind,
    const memory::desc& data_desc,
    const memory::desc& weight_desc
    )

Constructs a descriptor for a PReLU forward propagation primitive.

Parameters:

aprop_kind

Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.

data_desc

Source and destination memory descriptors.

weight_desc

Alpha parameters memory descriptor.