code: def _get_random_data(x: jax.Array) -> np.ndarray:
      dtype = dtypes.canonicalize_dtype(x.dtype)
      if np.issubdtype(dtype, np.integer):
          return np.random.randint(0, 100, size=x.shape, dtype=dtype)
      elif np.issubdtype(dtype, np.floating):
          return np.array(np.random.uniform(size=x.shape), dtype=dtype)
      elif dtype == bool:
          return np.random.choice(a=[False, True], size=x.shape)
      else:
          raise ValueError(
              f"Unsupported dtype for numerical comparison: {dtype}"
          )
  
file_path: jax/jax/experimental/jax2tf/tests/models_test_main.py
imports: from collections.abc import Sequence
  import datetime
  import functools
  import numpy
  import os
  import re
  import jax
  from absl import app
  from absl import flags
  from jax._src import dtypes
  import tensorflow
  from jax.core import InconclusiveDimensionOperation
  from jax.experimental.jax2tf.tests.model_harness import ALL_HARNESSES
  from jax.experimental.jax2tf.tests.converters import ALL_CONVERTERS
label_desc: Use image node_id 3 for calling a global function with example usage: _get_random_data(x) and returns: np, np, np