class_docstr: Test cases for stateless random-number generator operators.
class_docstr_tok: ["Test","cases","for","stateless","random-number","generator","operators","."]
docstr: Tests an OOM bug of StatelessRandomNormalV2 on TPU.
docstr_tok: ["Tests","an","OOM","bug","of","StatelessRandomNormalV2","on","TPU","."]
code: def testLargeNormal(self):
      """Tests an OOM bug of StatelessRandomNormalV2 on TPU."""
      with self.session() as sess, self.test_scope():
          seed_t = array_ops.placeholder(dtypes.int32, shape=[2])
          (
              key,
              counter,
              alg,
          ) = gen_stateless_random_ops_v2.stateless_random_get_key_counter_alg(
              seed_t
          )
          x = gen_stateless_random_ops_v2.stateless_random_normal_v2(
              shape=[1024, 32000],
              key=key,
              counter=counter,
              dtype=dtypes.float32,
              alg=alg,
          )
          y = sess.run(x, {seed_t: [0x12345678, 0xABCDEF1]})
          self.assertAllEqual([1024, 32000], y.shape)
          (
              key,
              counter,
          ) = gen_stateless_random_ops_v2.stateless_random_get_key_counter(
              seed_t
          )
          alg = gen_stateless_random_ops_v2.stateless_random_get_alg()
          x = gen_stateless_random_ops_v2.stateless_random_normal_v2(
              shape=[1024, 32000],
              key=key,
              counter=counter,
              dtype=dtypes.float32,
              alg=alg,
          )
          y = sess.run(x, {seed_t: [0x12345678, 0xABCDEF1]})
          self.assertAllEqual([1024, 32000], y.shape)
  
imports: import functools
  import os
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.client import device_lib
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import dtypes
  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_stateless_random_ops_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 5 for calling the StatelessRandomOpsTest obj's underlying member method code with example usage: obj.testLargeNormal() without return types