[][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.