src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertPreTrainingHeads/forward
class MobileBertPreTrainingHeads:    def forward(self, sequence_output, pooled_output):
        prediction_scores = self.predictions(sequence_output)
        seq_relationship_score = self.seq_relationship(pooled_output)
        return prediction_scores, seq_relationship_score