docstr: Tests that RNG can be used as tf.function's argument.
      
code: def testFunctionArg(self):
      """Tests that RNG can be used as tf.function's argument."""
      shape = [2, 3]
  
      @def_function.function
      def f(gen):
          return gen.normal(shape)
  
      g1 = random.Generator.from_seed(1)
      g2 = random.Generator.from_seed(1)
      res1 = f(g1)
      res2 = g2.normal(shape)
      self.assertAllEqual(res1, res2)
      self.assertAllEqual(g1.state.read_value(), g2.state.read_value())
  
imports: import os
  import re
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.checkpoint import checkpoint
  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 errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.kernel_tests.random import util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_random_ops
  from tensorflow.python.ops import gen_stateful_random_ops
  from tensorflow.python.ops import logging_ops
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.ops import stateful_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 34 for calling the StatefulRandomOpsTest obj's underlying member method code with example usage: obj.testFunctionArg() and returns: gen