src/transformers/models/transfo_xl/modeling_tf_transfo_xl.py/TFTransfoEmbeddings/build
class TFTransfoEmbeddings:    def build(self, input_shape):
        self.weight = self.add_weight(
            shape=(self.vocab_size, self.emb_size),
            initializer=get_initializer(self.init_std),
            name="embeddings",
        )

        super().build(input_shape)