src/transformers/models/perceiver/modeling_perceiver.py/PerceiverLayer/feed_forward_chunk
class PerceiverLayer:    def feed_forward_chunk(self, attention_output):
        layer_output = self.layernorm(attention_output)
        layer_output = self.mlp(layer_output)
        return layer_output