src/transformers/models/reformer/modeling_reformer.py/ChunkReformerFeedForward/forward_chunk
class ChunkReformerFeedForward:    def forward_chunk(self, hidden_states):
        hidden_states = self.layer_norm(hidden_states)
        hidden_states = self.dense(hidden_states)
        return self.output(hidden_states)