diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/constant_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/constant_op_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/constant_op_test.py b/tensorflow/python/kernel_tests/constant_op_test.py index fe93a30668..128a6529f0 100644 --- a/tensorflow/python/kernel_tests/constant_op_test.py +++ b/tensorflow/python/kernel_tests/constant_op_test.py @@ -771,6 +771,12 @@ class PlaceholderWithDefaultTest(test.TestCase): self.assertAllEqual( [[3, 3], [3, 3]], a.eval(feed_dict={p: [[3, 3], [3, 3]]})) + def testGradient(self): + with self.test_session(): + x = array_ops.placeholder(dtypes_lib.float32, [5, 7]) + y = array_ops.placeholder_with_default(x, None) + err = gradient_checker.compute_gradient_error(x, [5, 7], y, [5, 7]) + self.assertLess(err, 1e-3) if __name__ == "__main__": test.main() |