src/transformers/models/electra/modeling_tf_electra.py/TFElectraGeneratorPredictions/__init__
class TFElectraGeneratorPredictions:    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")