src/transformers/models/electra/modeling_tf_electra.py/TFElectraMaskedLMHead/__init__
class TFElectraMaskedLMHead:    def __init__(self, config, input_embeddings, **kwargs):
        super().__init__(**kwargs)

        self.config = config
        self.embedding_size = config.embedding_size
        self.input_embeddings = input_embeddings