Deep Neural Network Library (DNNL)  1.90.1
Performance library for Deep Learning
Public Member Functions | List of all members
dnnl::softmax_forward::desc Struct Reference

Descriptor for softmax forward propagation. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::softmax_forward::desc:
Collaboration graph
[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 dnnl::forward_training and dnnl::forward_inference) and memory descriptor data_desc.
 

Detailed Description

Descriptor for softmax forward propagation.


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