src/transformers/models/mobilebert/modeling_tf_mobilebert.py/TFMobileBertMLMHead/__init__
class TFMobileBertMLMHead:    def __init__(self, config, **kwargs):
        super().__init__(**kwargs)
        self.predictions = TFMobileBertLMPredictionHead(config, name="predictions")