struct dnnl::shuffle_backward::desc
Overview
Descriptor for a shuffle primitive backward propagation primitive. More…
#include <dnnl.hpp> struct desc { // fields dnnl_shuffle_desc_t data; // construction desc(const memory::desc& diff_data_desc, int axis, int group_size); };
Detailed Documentation
Descriptor for a shuffle primitive backward propagation primitive.
Construction
desc(const memory::desc& diff_data_desc, int axis, int group_size)
Constructs a descriptor for a shuffle backward propagation primitive.
Parameters:
diff_data_desc | Diff source and diff destination memory descriptor. |
axis | The axis along which the data is shuffled. |
group_size | Shuffle group size. |