code: def uniform(rng, shape, dtype):
      if np.issubdtype(dtype, np.integer):
          minval = None
      else:
          minval = 0
      return tf_np.asarray(
          rng.uniform(shape=shape, dtype=dtype, minval=minval)
      )
  
imports: import functools
  import itertools
  from absl import flags
  from absl.testing import parameterized
  from jax import lax
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy import extensions
  import trax.tf_numpy.numpy
label_desc: Use image node_id 4 for calling a global function with example usage: uniform(rng, shape, dtype) and returns: tf_np