src/transformers/models/bert/modeling_tf_bert.py/TFBertMLMHead/call
class TFBertMLMHead:    def call(self, sequence_output: tf.Tensor) -> tf.Tensor:
        prediction_scores = self.predictions(hidden_states=sequence_output)

        return prediction_scores