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

Descriptor for softmax backward propagation. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::softmax_backward::desc:
Collaboration graph
[legend]

Public Member Functions

 desc (const memory::desc &diff_desc, const memory::desc &data_desc, int softmax_axis)
 Initializes a softmax descriptor for backward propagation using memory descriptors diff_desc and data_desc. More...
 

Detailed Description

Descriptor for softmax backward propagation.

Constructor & Destructor Documentation

◆ desc()

dnnl::softmax_backward::desc::desc ( const memory::desc diff_desc,
const memory::desc data_desc,
int  softmax_axis 
)
inline

Initializes a softmax descriptor for backward propagation using memory descriptors diff_desc and data_desc.


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