Descriptor for softmax forward propagation. More...

#include <mkldnn.hpp>

Collaboration diagram for mkldnn::softmax_forward::desc:

[legend]

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 mkldnn::forward_training and mkldnn::forward_inference) and memory descriptor `data_desc`. More...

Detailed Description

Descriptor for softmax forward propagation.

Constructor & Destructor Documentation

mkldnn::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 mkldnn::forward_training and mkldnn::forward_inference) and memory descriptor data_desc.

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