.rst .pdf

Contents

struct dnnl::logsoftmax_forward::desc

Overview

Descriptor for a logsoftmax forward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_logsoftmax_desc_t data;

    // construction

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

Detailed Documentation

Descriptor for a logsoftmax forward propagation primitive.

Construction

desc()

Default constructor. Produces an empty object.

desc(prop_kind aprop_kind, const memory::desc& data_desc, int logsoftmax_axis)

Constructs a descriptor for a logsoftmax 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 descriptor.
logsoftmax_axis	Axis over which softmax is computed.

struct dnnl::logsoftmax_forward struct dnnl::logsoftmax_forward::primitive_desc