From 3d6c8b8aae8433b16af61097641b9958e679fb3d Mon Sep 17 00:00:00 2001 From: Smit Hinsu Date: Thu, 19 Jul 2018 11:49:44 -0700 Subject: [PATCH] Assert closeness of output values instead of equality in layout optimizer tests using convolutions Convolution output may differ across convolution algorithms and is not guaranteed to match exactly. PiperOrigin-RevId: 205276671 --- tensorflow/python/grappler/layout_optimizer_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/grappler/layout_optimizer_test.py b/tensorflow/python/grappler/layout_optimizer_test.py index 7d07c77c797..8cc971c61d5 100644 --- a/tensorflow/python/grappler/layout_optimizer_test.py +++ b/tensorflow/python/grappler/layout_optimizer_test.py @@ -1340,7 +1340,7 @@ class LayoutOptimizerTest(test.TestCase): expected_num_transposes = 2 self.assertEqual(expected_num_transposes, num_transposes) self._assert_trans_nhwc_to_nchw('Conv2D-0', nodes) - self.assertAllEqual(output_val_ref, output_val) + self.assertAllClose(output_val_ref, output_val, atol=1e-3) def testLoop(self): if test.is_gpu_available(cuda_only=True):