src/transformers/models/reformer/modeling_reformer.py/ReformerFeedForwardOutput/forward
class ReformerFeedForwardOutput:    def forward(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = nn.functional.dropout(hidden_states, p=self.dropout, training=self.training)
        return hidden_states