code: def _test_match_stateless_cpu_gpu(self, case, seed):
      # Stateless ops should produce the same result on CPUs and GPUs.
      _, stateless_op, _ = case
  
      with ops.device("CPU"):
          result_cpu = stateless_op(seed=seed)
  
      with ops.device(get_device().name):
          result_gpu = stateless_op(seed=seed)
          self.assertAllClose(result_cpu, result_gpu)
  
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 2 for calling the StatelessOpsTest obj's underlying member method code with example usage: obj._test_match_stateless_cpu_gpu(case, seed) without return types