[][src]Function onednn_sys::dnnl_softmax_backward_desc_init

pub unsafe extern "C" fn dnnl_softmax_backward_desc_init(
    softmax_desc: *mut dnnl_softmax_desc_t,
    diff_data_desc: *const dnnl_memory_desc_t,
    data_desc: *const dnnl_memory_desc_t,
    softmax_axis: c_int
) -> dnnl_status_t

Initializes a descriptor for softmax backward propagation primitive.

@param softmax_desc Output descriptor for a softmax primitive. @param diff_data_desc Diff source and diff destination memory descriptors. @param data_desc Destination memory descriptor. @param softmax_axis Axis over which softmax is computed. @returns #dnnl_success on success and a status describing the error otherwise.