[][src]Function onednn_sys::dnnl_layer_normalization_forward_desc_init

pub unsafe extern "C" fn dnnl_layer_normalization_forward_desc_init(
    lnrm_desc: *mut dnnl_layer_normalization_desc_t,
    prop_kind: dnnl_prop_kind_t,
    data_desc: *const dnnl_memory_desc_t,
    stat_desc: *const dnnl_memory_desc_t,
    epsilon: f32,
    flags: c_uint
) -> dnnl_status_t

Initializes a descriptor for layer normalization forward propagation primitive.

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

@param lnrm_desc Output descriptor for layer normalization 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 stat_desc Memory descriptor for mean and variance. If this parameter is NULL, a zero memory descriptor, or a memory descriptor with format_kind set to #dnnl_format_kind_undef, then the memory descriptor for stats is derived from @p data_desc by removing the last dimension. @param epsilon Layer normalization epsilon parameter. @param flags Layer normalization flags (@ref dnnl_normalization_flags_t). @returns #dnnl_success on success and a status describing the error otherwise.