code: def shuffle_cases():
      for dtype in np.int32, np.int64, np.float32, np.float64:
          # [], [0, ...] and [1, ...] are important corner cases
          for shape in (
              [],
              [0],
              [1],
              [100],
              [0, 0],
              [1, 0],
              [0, 1],
              [1, 2],
              [5, 3],
              [7, 5, 3, 2],
          ):
              value = (
                  np.arange(np.prod(shape)).reshape(shape).astype(dtype)
              )
              yield (
                  "shuffle",
                  functools.partial(stateless.stateless_shuffle, value),
                  functools.partial(random_ops.random_shuffle, value),
              )
  
imports: import functools
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.compat import compat
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import random_seed
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import gen_stateless_random_ops_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 8 for calling a global function with example usage: shuffle_cases() without return types