src/transformers/models/roberta_prelayernorm/modeling_tf_roberta_prelayernorm.py/TFRobertaPreLayerNormForMaskedLM/__init__
class TFRobertaPreLayerNormForMaskedLM:    def __init__(self, config, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)

        self.roberta_prelayernorm = TFRobertaPreLayerNormMainLayer(
            config, add_pooling_layer=False, name="roberta_prelayernorm"
        )
        self.lm_head = TFRobertaPreLayerNormLMHead(config, self.roberta_prelayernorm.embeddings, name="lm_head")