src/transformers/models/t5/modeling_flax_t5.py/FlaxT5DenseGatedGeluDense/__call__
class FlaxT5DenseGatedGeluDense:    def __call__(self, hidden_states, deterministic):
        hidden_gelu = self.gelu_act(self.wi_0(hidden_states))
        hidden_linear = self.wi_1(hidden_states)
        hidden_states = hidden_gelu * hidden_linear
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        hidden_states = self.wo(hidden_states)
        return hidden_states