src/transformers/models/convbert/modeling_tf_convbert.py/TFConvBertGeneratorPredictions/__init__
class TFConvBertGeneratorPredictions:    def __init__(self, config, **kwargs):
        super().__init__(**kwargs)

        self.LayerNorm = tf.keras.layers.LayerNormalization(epsilon=config.layer_norm_eps, name="LayerNorm")
        self.dense = tf.keras.layers.Dense(config.embedding_size, name="dense")