Descriptor for a shuffle primitive backward propagation primitive. More...
#include <dnnl.hpp>
Public Member Functions | |
desc (const memory::desc &diff_data_desc, int axis, int group_size) | |
Constructs a descriptor for a shuffle backward propagation primitive. More... | |
Descriptor for a shuffle primitive backward propagation primitive.
|
inline |
Constructs a descriptor for a shuffle backward propagation primitive.
diff_data_desc | Diff source and diff destination memory descriptor. |
axis | The axis along which the data is shuffled. |
group_size | Shuffle group size. |