src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTAttention/__init__
class TFMobileViTAttention:    def __init__(self, config: MobileViTConfig, hidden_size: int, **kwargs) -> None:
        super().__init__(**kwargs)
        self.attention = TFMobileViTSelfAttention(config, hidden_size, name="attention")
        self.dense_output = TFMobileViTSelfOutput(config, hidden_size, name="output")