src/transformers/models/t5/modeling_flax_t5.py/FlaxT5DenseActDense/__call__
class FlaxT5DenseActDense:    def __call__(self, hidden_states, deterministic=True):
        hidden_states = self.wi(hidden_states)
        hidden_states = self.act(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        hidden_states = self.wo(hidden_states)
        return hidden_states