src/transformers/models/funnel/modeling_tf_funnel.py/TFFunnelDiscriminatorPredictions/call
class TFFunnelDiscriminatorPredictions:    def call(self, discriminator_hidden_states):
        hidden_states = self.dense(discriminator_hidden_states)
        hidden_states = self.activation_function(hidden_states)
        logits = tf.squeeze(self.dense_prediction(hidden_states))
        return logits