src/transformers/models/albert/modeling_tf_albert.py/TFAlbertSOPHead/call
class TFAlbertSOPHead:    def call(self, pooled_output: tf.Tensor, training: bool) -> tf.Tensor:
        dropout_pooled_output = self.dropout(inputs=pooled_output, training=training)
        logits = self.classifier(inputs=dropout_pooled_output)

        return logits