src/transformers/models/segformer/modeling_tf_segformer.py/TFSegformerMainLayer/__init__
class TFSegformerMainLayer:    def __init__(self, config: SegformerConfig, **kwargs):
        super().__init__(**kwargs)

        self.config = config
        # hierarchical Transformer encoder
        self.encoder = TFSegformerEncoder(config, name="encoder")