diff options
author | 2017-06-23 11:21:46 -0700 | |
---|---|---|
committer | 2017-06-23 11:25:23 -0700 | |
commit | 53ac1ea05a61e4e2d35cdbdf67727d3d7571faa4 (patch) | |
tree | 8713c0cb8ed9e350711ae34401f49deef8ab548b | |
parent | 4cc35257f1f3613aae65a42b96e5ea5bba785f66 (diff) |
Increase rtol for util_test.
PiperOrigin-RevId: 159971136
-rw-r--r-- | tensorflow/python/kernel_tests/distributions/util_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/kernel_tests/distributions/util_test.py b/tensorflow/python/kernel_tests/distributions/util_test.py index 7bce16c8a0..fb1960f672 100644 --- a/tensorflow/python/kernel_tests/distributions/util_test.py +++ b/tensorflow/python/kernel_tests/distributions/util_test.py @@ -162,8 +162,8 @@ class GetLogitsAndProbsTest(test.TestCase): new_logits, new_p = distribution_util.get_logits_and_probs( logits=logits, validate_args=True) - self.assertAllClose(p, new_p.eval(), rtol=1e-6, atol=0.) - self.assertAllClose(logits, new_logits.eval(), rtol=1e-6, atol=0.) + self.assertAllClose(p, new_p.eval(), rtol=1e-5, atol=0.) + self.assertAllClose(logits, new_logits.eval(), rtol=1e-5, atol=0.) def testLogitsMultidimensional(self): p = np.array([0.2, 0.3, 0.5], dtype=np.float32) |