src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertAttention/__init__
class TFRemBertAttention:    def __init__(self, config: RemBertConfig, **kwargs):
        super().__init__(**kwargs)

        self.self_attention = TFRemBertSelfAttention(config, name="self")
        self.dense_output = TFRemBertSelfOutput(config, name="output")