src/transformers/models/bert/modeling_tf_bert.py/TFBertAttention/__init__
class TFBertAttention:    def __init__(self, config: BertConfig, **kwargs):
        super().__init__(**kwargs)

        self.self_attention = TFBertSelfAttention(config, name="self")
        self.dense_output = TFBertSelfOutput(config, name="output")