src/transformers/models/t5/modeling_tf_t5.py/TFT5DenseReluDense/call
class TFT5DenseReluDense:    def call(self, hidden_states, training=False):
        hidden_states = self.wi(hidden_states)
        hidden_states = self.act(hidden_states)
        hidden_states = self.dropout(hidden_states, training=training)
        hidden_states = self.wo(hidden_states)
        return hidden_states