dnnl::softmax_backward::desc Struct Reference

Descriptor for a softmax backward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::softmax_backward::desc:

Public Member Functions
	desc ()=default
	Default constructor. Produces an empty object.
	desc (const memory::desc &diff_data_desc, const memory::desc &data_desc, int softmax_axis)
	Constructs a descriptor for a softmax backward propagation primitive. More...

Detailed Description

Descriptor for a softmax backward propagation primitive.

Constructor & Destructor Documentation

desc()

dnnl::softmax_backward::desc::desc ( const memory::desc &  diff_data_desc, const memory::desc &  data_desc, int  softmax_axis  )

inline

Constructs a descriptor for a softmax backward propagation primitive.

Inputs:

• dst (dnnl::primitive_desc_base::dst_desc (0))
• diff_dst (dnnl::primitive_desc_base::diff_dst_desc (0))

Outputs:

• diff_src (dnnl::primitive_desc_base::diff_src_desc (0))

Parameters

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

---

The documentation for this struct was generated from the following file:

• include/dnnl.hpp