commit ac2879be7c5213264880198c0529c4e6036e8113
author David Majnemer <majnemer@google.com> Thu Jul 01 09:24:35 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Jul 01 09:35:41 2021 -0700
tree dc980565ee02307de162849e65a9b0d9d2a427bd
parent 5fb32a753e1d6b3540805e8cfb64d773a85f0d4e

[XLA] Adjust test size

PiperOrigin-RevId: 382539747
Change-Id: I4887d202826b89a72c9eebcd23acdf322552fa0b

tensorflow/compiler/tests/random_ops_test.py

diff --git a/tensorflow/compiler/tests/random_ops_test.py b/tensorflow/compiler/tests/random_ops_test.py
index bd1411d..9ea2b22 100644
--- a/tensorflow/compiler/tests/random_ops_test.py
+++ b/tensorflow/compiler/tests/random_ops_test.py
@@ -81,7 +81,7 @@
       'stddev': 2.0
   })
   def testRandomNormal(self, mean, stddev):
-    num_elts = 10000000
+    num_elts = 1000000
     for dtype in self._random_types() & self.float_types:
       with self.session():
         with self.test_scope():
@@ -156,7 +156,7 @@
     if x.dtype == dtypes.bfloat16:
       atol = rtol = 1e-1
     else:
-      atol = rtol = 2e-3
+      atol = rtol = 2e-2
     self.assertAllClose(actual_mean, expected_mean, atol=atol, rtol=rtol)
 
     expected_median = mu + probit(