src/transformers/models/mobilevit/modeling_mobilevit.py/MobileViTAttention/forward
class MobileViTAttention:    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
        self_outputs = self.attention(hidden_states)
        attention_output = self.output(self_outputs)
        return attention_output