src/transformers/models/mobilebert/modeling_tf_mobilebert.py/TFMobileBertMLMHead/call
class TFMobileBertMLMHead:    def call(self, sequence_output):
        prediction_scores = self.predictions(sequence_output)
        return prediction_scores