src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertLMPredictionHead/build
class TFRemBertLMPredictionHead:    def build(self, input_shape: tf.TensorShape):
        self.decoder = self.add_weight(
            name="decoder/weight",
            shape=[self.config.vocab_size, self.output_embedding_size],
            initializer=get_initializer(self.initializer_range),
        )
        self.decoder_bias = self.add_weight(
            shape=(self.config.vocab_size,), initializer="zeros", trainable=True, name="decoder/bias"
        )

        super().build(input_shape)