code: def setUp(self):
      super(DTensorRNGTest, self).setUp()
      global_ids = test_util.create_device_ids_array((2, 4))
      local_ids = _LOCAL_IDS
      mesh_dict = {
          device: Mesh(
              [_MESH_DIM_X, _MESH_DIM_Y],
              global_ids,
              local_ids,
              test_util.create_device_list((2, 4), device),
          )
          for device in ("CPU", "GPU", "TPU")
      }
      self.mesh = self.configTestMesh(mesh_dict)
  
      # Creates a bunch of common layouts used by tests later.
      self.replicated_layout_2d = Layout.replicated(self.mesh, rank=2)
      self.shardings = {
          "batch": Layout.batch_sharded,
          "inner": Layout.inner_sharded,
      }
      # Creates a bunch of parameters for rng V2 ops
      self.key = constant_op.constant([123], dtype=dtypes.uint64)
      self.counter = constant_op.constant([1, 1], dtype=dtypes.uint64)
      self.alg = 1
      self.minval = 1
      self.maxval = 100
  
imports: from absl.testing import parameterized
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import d_variable
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python import numpy_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.dtensor.python.tests import test_util_ops
  from tensorflow.python.distribute import tpu_strategy
  from tensorflow.python.distribute.cluster_resolver.tpu import tpu_cluster_resolver
  from tensorflow.python.eager import remote
  from tensorflow.python.eager.polymorphic_function import polymorphic_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import gen_bitwise_ops
  from tensorflow.python.ops import gen_stateful_random_ops
  from tensorflow.python.ops import gen_stateless_random_ops_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import device_assignment
label_desc: Use image node_id 1 for calling the DTensorRNGTest obj's underlying member method code with example usage: obj.setUp() without return types