src/transformers/models/efficientformer/modeling_efficientformer.py/EfficientFormerLastStage/__init__
class EfficientFormerLastStage:    def __init__(self, config: EfficientFormerConfig):
        super().__init__()
        self.meta4D_layers = EfficientFormerMeta4DLayers(config, -1)
        self.flat = EfficientFormerFlat()
        self.meta3D_layers = EfficientFormerMeta3DLayers(config)