pub fn layer_norm<C: Into<LayerNormConfig>>( size: usize, config: C, vb: VarBuilder<'_>, ) -> Result<LayerNorm>