.rst .pdf

Contents

struct dnnl_prelu_desc_t

Overview

struct dnnl_prelu_desc_t
{
    // fields

    dnnl_primitive_kind_t primitive_kind;
    dnnl_prop_kind_t prop_kind;
    dnnl_memory_desc_t data_desc;
    dnnl_memory_desc_t weights_desc;
    dnnl_memory_desc_t diff_data_desc;
    dnnl_memory_desc_t diff_weights_desc;
};

Detailed Documentation

Fields

dnnl_primitive_kind_t primitive_kind

The kind of primitive.

Used for self-identifying the primitive descriptor. Must be dnnl_prelu.

dnnl_prop_kind_t prop_kind

The kind of propagation.

Possible values: dnnl_forward_training, dnnl_forward_inference, dnnl_backward

dnnl_memory_desc_t data_desc

Source and destination memory descriptor.

dnnl_memory_desc_t weights_desc

Learnable parameter alpha memory descriptor.

Alpha describes negative slope.

dnnl_memory_desc_t diff_data_desc

Source and destination gradient memory descriptor.

dnnl_memory_desc_t diff_weights_desc

Learnable parameter alpha gradient memory descriptor.

PReLU struct dnnl::prelu_backward