.rst .pdf

Contents

struct dnnl::logsoftmax_backward::desc

Overview

Descriptor for a logsoftmax backward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_logsoftmax_desc_t data;

    // construction

    desc();

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

Detailed Documentation

Descriptor for a logsoftmax backward propagation primitive.

Construction

desc()

Default constructor. Produces an empty object.

desc(
    const memory::desc& diff_data_desc,
    const memory::desc& data_desc,
    int logsoftmax_axis
    )

Constructs a descriptor for a logsoftmax backward propagation primitive.

Parameters:

diff_data_desc	Diff source and diff destination memory descriptors.
data_desc	Destination memory descriptor.
logsoftmax_axis	Axis over which softmax is computed.

struct dnnl::logsoftmax_backward struct dnnl::logsoftmax_backward::primitive_desc