code: def setUp(self):
      super().setUp()
  
      if FLAGS.compute_on == "tpu" and not JAX_MODE:
          resolver = tf.distribute.cluster_resolver.TPUClusterResolver(
              ""
          )
          tf.config.experimental_connect_to_cluster(resolver)
          tf.tpu.experimental.initialize_tpu_system(resolver)
          self.strategy = tf.distribute.experimental.TPUStrategy(
              resolver
          )
      else:
          self.strategy = None
  
file_path: probability/tensorflow_probability/python/experimental/math/manual_special_functions_test.py
imports: import functools
  from absl import flags
  from absl.testing import parameterized
  import mock
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.math import manual_special_functions
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 1 for calling the ManualSpecialFunctionsTest obj's underlying member method code with example usage: obj.setUp() without return types