src/transformers/models/segformer/modeling_tf_segformer.py/TFSegformerForSemanticSegmentation/__init__
class TFSegformerForSemanticSegmentation:    def __init__(self, config: SegformerConfig, **kwargs):
        super().__init__(config, **kwargs)
        self.segformer = TFSegformerMainLayer(config, name="segformer")
        self.decode_head = TFSegformerDecodeHead(config, name="decode_head")