src/transformers/models/data2vec/modeling_data2vec_text.py/Data2VecTextLayer/feed_forward_chunk
class Data2VecTextLayer:    def feed_forward_chunk(self, attention_output):
        intermediate_output = self.intermediate(attention_output)
        layer_output = self.output(intermediate_output, attention_output)
        return layer_output