Descriptor for softmax backward propagation. More...
#include <dnnl.hpp>
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... | |
Descriptor for softmax backward propagation.
|
inline |
Initializes a softmax descriptor for backward propagation using memory descriptors diff_desc
and data_desc
.