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