src/transformers/models/electra/modeling_flax_electra.py/FlaxElectraDiscriminatorPredictions/__call__
class FlaxElectraDiscriminatorPredictions:    def __call__(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = ACT2FN[self.config.hidden_act](hidden_states)
        hidden_states = self.dense_prediction(hidden_states).squeeze(-1)
        return hidden_states