experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Increase tolerance in TensorfFlow test of lgamma

Browse Source 
Relax tolerance in favor of the new MLIR-generated `lgamma` kernel. The new
kernel uses the same approximation as XLA.

PiperOrigin-RevId: 355356258
Change-Id: I9179f843903a061e517960729720c0949aa71a76
This commit is contained in:
A. Unique TensorFlower 2021-02-03 02:46:44 -08:00 committed by TensorFlower Gardener
parent eed09a5ab0
commit b137fe2ad5
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tensorflow/python/kernel_tests/distributions/beta_test.py
View File

@ -112,7 +112,7 @@ class BetaTest(test.TestCase):
x = [.5, .5]
dist = beta_lib.Beta(a, b)
pdf = dist.prob(x)
self.assertAllClose([1., 3. / 2], self.evaluate(pdf))
self.assertAllClose([1., 3. / 2], self.evaluate(pdf), rtol=1e-5, atol=1e-5)
self.assertEqual((2,), pdf.get_shape())
def testPdfTwoBatchesNontrivialX(self):
@ -121,7 +121,7 @@ class BetaTest(test.TestCase):
x = [.3, .7]
dist = beta_lib.Beta(a, b)
pdf = dist.prob(x)
self.assertAllClose([1, 63. / 50], self.evaluate(pdf))
self.assertAllClose([1, 63. / 50], self.evaluate(pdf), rtol=1e-5, atol=1e-5)
self.assertEqual((2,), pdf.get_shape())
def testPdfUniformZeroBatch(self):
@ -140,7 +140,10 @@ class BetaTest(test.TestCase):
x = [[.5, .5], [.3, .7]]
dist = beta_lib.Beta(a, b)
pdf = dist.prob(x)
self.assertAllClose([[1., 3. / 2], [1., 63. / 50]], self.evaluate(pdf))
self.assertAllClose([[1., 3. / 2], [1., 63. / 50]],
self.evaluate(pdf),
rtol=1e-5,
atol=1e-5)
self.assertEqual((2, 2), pdf.get_shape())
def testPdfAlphaStretchedInBroadcastWhenLowerRank(self):
@ -148,7 +151,10 @@ class BetaTest(test.TestCase):
b = [1., 2]
x = [[.5, .5], [.2, .8]]
pdf = beta_lib.Beta(a, b).prob(x)
self.assertAllClose([[1., 3. / 2], [1., 24. / 25]], self.evaluate(pdf))
self.assertAllClose([[1., 3. / 2], [1., 24. / 25]],
self.evaluate(pdf),
rtol=1e-5,
atol=1e-5)
self.assertEqual((2, 2), pdf.get_shape())
def testPdfXStretchedInBroadcastWhenSameRank(self):
@ -156,7 +162,10 @@ class BetaTest(test.TestCase):
b = [[1., 2], [2., 3]]
x = [[.5, .5]]
pdf = beta_lib.Beta(a, b).prob(x)
self.assertAllClose([[1., 3. / 2], [3. / 2, 15. / 8]], self.evaluate(pdf))
self.assertAllClose([[1., 3. / 2], [3. / 2, 15. / 8]],
self.evaluate(pdf),
rtol=1e-5,
atol=1e-5)
self.assertEqual((2, 2), pdf.get_shape())
def testPdfXStretchedInBroadcastWhenLowerRank(self):
@ -164,7 +173,10 @@ class BetaTest(test.TestCase):
b = [[1., 2], [2., 3]]
x = [.5, .5]
pdf = beta_lib.Beta(a, b).prob(x)
self.assertAllClose([[1., 3. / 2], [3. / 2, 15. / 8]], self.evaluate(pdf))
self.assertAllClose([[1., 3. / 2], [3. / 2, 15. / 8]],
self.evaluate(pdf),
rtol=1e-5,
atol=1e-5)
self.assertEqual((2, 2), pdf.get_shape())
def testLogPdfOnBoundaryIsFiniteWhenAlphaIsOne(self):