src/transformers/models/mt5/modeling_mt5.py/MT5LayerFF/forward
class MT5LayerFF:    def forward(self, hidden_states):
        forwarded_states = self.layer_norm(hidden_states)
        forwarded_states = self.DenseReluDense(forwarded_states)
        hidden_states = hidden_states + self.dropout(forwarded_states)
        return hidden_states