src/transformers/models/electra/modeling_electra.py/ElectraGeneratorPredictions/forward
class ElectraGeneratorPredictions:    def forward(self, generator_hidden_states):
        hidden_states = self.dense(generator_hidden_states)
        hidden_states = get_activation("gelu")(hidden_states)
        hidden_states = self.LayerNorm(hidden_states)

        return hidden_states