src/transformers/models/vit_mae/modeling_tf_vit_mae.py/TFViTMAEAttention/__init__
class TFViTMAEAttention:    def __init__(self, config: ViTMAEConfig, **kwargs):
        super().__init__(**kwargs)

        self.self_attention = TFViTMAESelfAttention(config, name="attention")
        self.dense_output = TFViTMAESelfOutput(config, name="output")