dnnl::softmax_forward::desc Struct Reference

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

#include <dnnl.hpp>

Collaboration diagram for dnnl::softmax_forward::desc:

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

Detailed Description

Descriptor for a softmax forward propagation primitive.

Examples:

softmax.cpp.

Constructor & Destructor Documentation

desc()

dnnl::softmax_forward::desc::desc ( prop_kind  aprop_kind, const memory::desc &  data_desc, int  softmax_axis  )

inline

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

---

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

• include/oneapi/dnnl/dnnl.hpp