Descriptor for a softmax backward propagation primitive. More...
#include <dnnl.hpp>

Public Member Functions | |
| desc ()=default | |
| Default constructor. Produces an empty object. | |
| desc (const memory::desc &diff_data_desc, const memory::desc &data_desc, int softmax_axis) | |
| Constructs a descriptor for a softmax backward propagation primitive. More... | |
Descriptor for a softmax backward propagation primitive.
|
inline |
Constructs a descriptor for a softmax backward propagation primitive.
| diff_data_desc | Diff source and diff destination memory descriptor. |
| data_desc | Destination memory descriptor. |
| softmax_axis | Axis over which softmax is computed. |