code: def testJitterFn(self):
      cholesky_fn = cholesky_util.make_cholesky_with_jitter_fn(
          jitter=0.0
      )
  
      x = tf.random.normal(shape=[2, 4, 4], seed=test_util.test_seed())
      x = tf.linalg.matmul(x, x, transpose_b=True)
      actual_chol, expected_chol = self.evaluate(
          [cholesky_fn(x), tf.linalg.cholesky(x)]
      )
      self.assertAllClose(expected_chol, actual_chol)
  
      cholesky_fn = cholesky_util.make_cholesky_with_jitter_fn(
          jitter=1.0
      )
      x = 3.0 * tf.linalg.eye(3)
      self.assertAllClose(
          self.evaluate(2.0 * tf.linalg.eye(3)),
          self.evaluate(cholesky_fn(x)),
      )
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import cholesky_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling the CholeskyUtilsTest obj's underlying member method code with example usage: obj.testJitterFn() without return types