src/transformers/models/fnet/modeling_fnet.py/FNetLayer/feed_forward_chunk
class FNetLayer:    def feed_forward_chunk(self, fourier_output):
        intermediate_output = self.intermediate(fourier_output)
        layer_output = self.output(intermediate_output, fourier_output)
        return layer_output