// structs struct dnnl_shuffle_desc_t; struct dnnl::shuffle_backward; struct dnnl::shuffle_forward; // global functions dnnl_status_t DNNL_API dnnl_shuffle_forward_desc_init( dnnl_shuffle_desc_t* shuffle_desc, dnnl_prop_kind_t prop_kind, const dnnl_memory_desc_t* data_desc, int axis, dnnl_dim_t group_size ); dnnl_status_t DNNL_API dnnl_shuffle_backward_desc_init( dnnl_shuffle_desc_t* shuffle_desc, const dnnl_memory_desc_t* diff_data_desc, int axis, dnnl_dim_t group_size );
Output descriptor for a shuffle primitive.
Diff source and diff destination memory descriptor.
The axis along which the data is shuffled.
Shuffle group size.