src/transformers/models/vit/modeling_tf_vit.py/TFViTAttention/__init__
class TFViTAttention:    def __init__(self, config: ViTConfig, **kwargs):
        super().__init__(**kwargs)

        self.self_attention = TFViTSelfAttention(config, name="attention")
        self.dense_output = TFViTSelfOutput(config, name="output")