src/transformers/models/mobilebert/modeling_tf_mobilebert.py/TFMobileBertLMPredictionHead/set_output_embeddings
class TFMobileBertLMPredictionHead:    def set_output_embeddings(self, value):
        self.decoder = value
        self.vocab_size = shape_list(value)[0]