[−][src]Function onednn_sys::dnnl_shuffle_backward_desc_init
pub unsafe extern "C" fn dnnl_shuffle_backward_desc_init(
shuffle_desc: *mut dnnl_shuffle_desc_t,
diff_data_desc: *const dnnl_memory_desc_t,
axis: c_int,
group_size: dnnl_dim_t
) -> dnnl_status_t
Initializes a descriptor for shuffle backward propagation primitive.
@param shuffle_desc Output descriptor for a shuffle primitive. @param diff_data_desc Diff source and diff destination memory descriptor. @param axis The axis along which the data is shuffled. @param group_size Shuffle group size. @returns #dnnl_success on success and a status describing the error otherwise.