[][src]Function onednn_sys::dnnl_batch_normalization_backward_desc_init

pub unsafe extern "C" fn dnnl_batch_normalization_backward_desc_init(
    bnrm_desc: *mut dnnl_batch_normalization_desc_t,
    prop_kind: dnnl_prop_kind_t,
    diff_data_desc: *const dnnl_memory_desc_t,
    data_desc: *const dnnl_memory_desc_t,
    epsilon: f32,
    flags: c_uint
) -> dnnl_status_t

Initializes a descriptor for a batch normalization backward propagation primitive.

@note In-place operation is supported: the diff_dst can refer to the same memory as the diff_src.

@param bnrm_desc Output descriptor for batch normalization primitive. @param prop_kind Propagation kind. Possible values are #dnnl_backward_data and #dnnl_backward (diffs for all parameters are computed in this case). @param diff_data_desc Diff source and diff destination memory descriptor. @param data_desc Source memory descriptor. @param epsilon Batch normalization epsilon parameter. @param flags Batch normalization flags (@ref dnnl_normalization_flags_t). @returns #dnnl_success on success and a status describing the error otherwise.