code: def testXLA(self):
      self.skip_if_no_xla()
      gs = tf.function(gram_schmidt, jit_compile=True)
      mat = self.evaluate(
          tf.random.normal([10, 5], seed=test_util.test_seed())
      )
      self.assertAllClose(gs(mat), gram_schmidt(mat))
  
file_path: probability/tensorflow_probability/python/math/gram_schmidt_test.py
imports: import numpy
  import scipy.linalg
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.gram_schmidt import gram_schmidt
label_desc: Use image node_id 4 for calling the GramSchmidtTest obj's underlying member method code with example usage: obj.testXLA() without return types