code: def testSampleXLA(self):
      self.skip_if_no_xla()
      if not tf.executing_eagerly():
          return  # jit_compile is eager-only.
      concentration = np.exp(np.random.rand(4, 3).astype(np.float32))
      rate = np.exp(np.random.rand(4, 3).astype(np.float32))
      dist = exp_gamma_lib.ExpGamma(
          concentration=concentration, rate=rate, validate_args=True
      )
      # Verify the compile succeeds going all the way through the distribution.
      self.evaluate(
          tf.function(
              lambda: dist.sample(5, seed=test_util.test_seed()),
              jit_compile=True,
          )()
      )
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import misc
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import exp
  from tensorflow_probability.python.distributions import exp_gamma
  from tensorflow_probability.python.distributions import gamma
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 2 for calling the GammaSamplingTest obj's underlying member method code with example usage: obj.testSampleXLA() without return types