tf.matrix_band_part have been deprecated
https://github.com/huggingface/transformers/blob/fae0b9faefc371977d139b60202a63dcedbd8ea8/src/transformers/models/xlnet/modeling_tf_xlnet.py#L500

This should be replaced by `tf.linalg.band_part` here?
