src/transformers/models/cvt/modeling_cvt.py/CvtAttention/forward
class CvtAttention:    def forward(self, hidden_state, height, width):
        self_output = self.attention(hidden_state, height, width)
        attention_output = self.output(self_output, hidden_state)
        return attention_output