code: def skip_if_no_xla(skip_test_fn):
      try:
          tf.function(lambda: tf.constant(0), jit_compile=True)()
      except tf.errors.UnimplementedError as e:
          if "Could not find compiler" in str(e):
              skip_test_fn("XLA not available")
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.linalg import linear_operator_psd_kernel
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.psd_kernels import exponentiated_quadratic
  from tensorflow_probability.python.math.psd_kernels import polynomial
label_desc: Use image node_id 1 for calling a global function with example usage: skip_if_no_xla(skip_test_fn) without return types