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