[][src]Function onednn_sys::dnnl_logsoftmax_forward_desc_init

pub unsafe extern "C" fn dnnl_logsoftmax_forward_desc_init(
    logsoftmax_desc: *mut dnnl_logsoftmax_desc_t,
    prop_kind: dnnl_prop_kind_t,
    data_desc: *const dnnl_memory_desc_t,
    logsoftmax_axis: c_int
) -> dnnl_status_t

Initializes a descriptor for logsoftmax forward propagation primitive.

@param logsoftmax_desc Output descriptor for a logsoftmax primitive. @param prop_kind Propagation kind. Possible values are #dnnl_forward_training and #dnnl_forward_inference. @param data_desc Source and destination memory descriptor. @param logsoftmax_axis Axis over which logsoftmax is computed. @returns #dnnl_success on success and a status describing the error otherwise.