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.
Inputs:
diff_dst
(dnnl::primitive_desc_base::diff_dst_desc(0
))Outputs:
diff_src
(dnnl::primitive_desc_base::diff_src_desc(0
))diff_data_desc | Diff source and diff destination memory descriptor. |
axis | The axis along which the data is shuffled. |
group_size | Shuffle group size. |