code: def test_jitted_sampling(self):
      self.skip_if_no_xla()
      shape = np.int32([2, 3])
      seed = test_util.test_seed()
      dimension = np.int32(10)
  
      @tf.function(jit_compile=True)
      def sample():
          return random_ops.spherical_uniform(
              dimension=dimension,
              shape=shape,
              seed=seed,
              dtype=self.dtype,
          )
  
      samples = self.evaluate(sample())
      self.assertAllEqual([2, 3, 10], samples.shape)
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.random import random_ops
  from tensorflow_probability.python.stats import sample_stats
label_desc: Use image node_id 7 for calling the _RandomSphericalUniform obj's underlying member method code with example usage: obj.test_jitted_sampling() and returns: random_ops