src/transformers/models/cvt/modeling_tf_cvt.py/TFCvtAttention/call
class TFCvtAttention:    def call(self, hidden_state: tf.Tensor, height: int, width: int, training: bool = False):
        self_output = self.attention(hidden_state, height, width, training=training)
        attention_output = self.dense_output(self_output, training=training)
        return attention_output