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

Change contrib/bayesflow/custom_grad tests to assertAllClose.

Browse Source 
PiperOrigin-RevId: 162681035
This commit is contained in:
Joshua V. Dillon 2017-07-20 16:19:22 -07:00 committed by TensorFlower Gardener
parent d2f8e98650
commit e4d6cb9744
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/contrib/bayesflow/python/kernel_tests/custom_grad_test.py
View File

@ -48,8 +48,8 @@ class CustomGradientTest(test.TestCase):
gx = gradients_impl.gradients(fx, x)[0] gx = gradients_impl.gradients(fx, x)[0]
[fx_, gx_] = sess.run([fx, gx]) [fx_, gx_] = sess.run([fx, gx])
self.assertAllEqual(f(x_), fx_) self.assertAllClose(f(x_), fx_)
self.assertAllEqual(g(x_), gx_) self.assertAllClose(g(x_), gx_)
def test_works_correctly_both_f_g_zero(self): def test_works_correctly_both_f_g_zero(self):
with self.test_session() as sess: with self.test_session() as sess:
@ -62,8 +62,8 @@ class CustomGradientTest(test.TestCase):
gx = gradients_impl.gradients(fx, x)[0] gx = gradients_impl.gradients(fx, x)[0]
[fx_, gx_] = sess.run([fx, gx]) [fx_, gx_] = sess.run([fx, gx])
self.assertAllEqual(f(x_), fx_) self.assertAllClose(f(x_), fx_)
self.assertAllEqual(g(x_), gx_) self.assertAllClose(g(x_), gx_)
def test_works_correctly_vector_of_vars(self): def test_works_correctly_vector_of_vars(self):
with self.test_session() as sess: with self.test_session() as sess: