dnnl::softmax_forward::desc Struct Reference

Descriptor for softmax forward propagation. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::softmax_forward::desc:

Public Member Functions
	desc (prop_kind aprop_kind, const memory::desc &data_desc, int softmax_axis)
	Initializes a softmax descriptor for forward propagation using prop_kind (possible values are dnnl::forward_training and dnnl::forward_inference) and memory descriptor data_desc. More...

Detailed Description

Descriptor for softmax forward propagation.

Constructor & Destructor Documentation

desc()

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

inline

Initializes a softmax descriptor for forward propagation using prop_kind (possible values are dnnl::forward_training and dnnl::forward_inference) and memory descriptor data_desc.

---

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

• include/dnnl.hpp