src/transformers/models/vit_mae/modeling_tf_vit_mae.py/TFViTMAEForPreTraining/__init__
class TFViTMAEForPreTraining:    def __init__(self, config):
        super().__init__(config)
        self.config = config

        self.vit = TFViTMAEMainLayer(config, name="vit")
        self.decoder = TFViTMAEDecoder(
            config,
            num_patches=self.vit.embeddings.num_patches,
            name="decoder",
        )