src/transformers/models/deit/modeling_tf_deit.py/TFDeiTAttention/__init__
class TFDeiTAttention:    def __init__(self, config: DeiTConfig, **kwargs):
        super().__init__(**kwargs)

        self.self_attention = TFDeiTSelfAttention(config, name="attention")
        self.dense_output = TFDeiTSelfOutput(config, name="output")