src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTAttention/call
class TFMobileViTAttention:    def call(self, hidden_states: tf.Tensor, training: bool = False) -> tf.Tensor:
        self_outputs = self.attention(hidden_states, training=training)
        attention_output = self.dense_output(self_outputs, training=training)
        return attention_output