src/transformers/models/tapas/modeling_tf_tapas.py/TFTapasMainLayer/__init__
class TFTapasMainLayer:    def __init__(self, config: TapasConfig, add_pooling_layer: bool = True, **kwargs):
        requires_backends(self, "tensorflow_probability")
        super().__init__(**kwargs)

        self.config = config

        self.embeddings = TFTapasEmbeddings(config, name="embeddings")
        self.encoder = TFTapasEncoder(config, name="encoder")
        self.pooler = TFTapasPooler(config, name="pooler") if add_pooling_layer else None