Increases rtol for a few assertAllClose so that OSS build can pass.

Change: 112917524
This commit is contained in:
A. Unique TensorFlower 2016-01-24 21:36:40 -08:00 committed by Vijay Vasudevan
parent 9b8fe3cec9
commit 87af568b0b

View File

@ -404,9 +404,9 @@ class UnrollLSTMTest(tf.test.TestCase):
mv1 = RunForward("cell")
mv2 = RunForward("loop")
mv3 = RunForward("loop10")
self.assertAllClose(mv0, mv1)
self.assertAllClose(mv0, mv2)
self.assertAllClose(mv0, mv3)
self.assertAllClose(mv0, mv1, rtol=1e-4)
self.assertAllClose(mv0, mv2, rtol=1e-4)
self.assertAllClose(mv0, mv3, rtol=1e-4)
def testUnrollLSTMGrad(self):
# Run one step of the unrolled lstm graph.
@ -430,9 +430,9 @@ class UnrollLSTMTest(tf.test.TestCase):
d1 = RunForwardBackward("cell")
d2 = RunForwardBackward("loop")
d3 = RunForwardBackward("loop10")
self.assertAllClose(d0, d1)
self.assertAllClose(d0, d2)
self.assertAllClose(d0, d3)
self.assertAllClose(d0, d1, rtol=1e-4)
self.assertAllClose(d0, d2, rtol=1e-4)
self.assertAllClose(d0, d3, rtol=1e-4)
if __name__ == "__main__":