src/transformers/models/gpt2/modeling_flax_gpt2.py/FlaxGPT2MLP/__call__
class FlaxGPT2MLP:    def __call__(self, hidden_states, deterministic: bool = True):
        hidden_states = self.c_fc(hidden_states)
        hidden_states = self.act(hidden_states)
        hidden_states = self.c_proj(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        return hidden_states