Function layer_norm 

pub fn layer_norm<C>(
    size: usize,
    config: C,
    vb: VarBuilderArgs<'static, ShardedSafeTensors>,
) -> Result<LayerNorm, Error>
where
    C: Into<LayerNormConfig>,