src/transformers/models/mpnet/modeling_tf_mpnet.py/TFMPNetEncoder/build
class TFMPNetEncoder:    def build(self, input_shape):
        with tf.name_scope("relative_attention_bias"):
            self.relative_attention_bias = self.add_weight(
                name="embeddings",
                shape=[self.relative_attention_num_buckets, self.n_heads],
                initializer=get_initializer(self.initializer_range),
            )

        return super().build(input_shape)