From df90003a68bcb813843e447d6fa2c49deccc48b6 Mon Sep 17 00:00:00 2001 From: Benjamin Kramer Date: Mon, 24 Sep 2018 20:54:47 -0700 Subject: Increase the tolerance to deflake fused_conv2d test For testInceptionFwd I see 8.482029 != 8.48317 when comparing GPU vs. CPU. testFusedConvInt8 has off-by-one errors. Both occur flakily. PiperOrigin-RevId: 214378820 --- .../fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'tensorflow/contrib/fused_conv') diff --git a/tensorflow/contrib/fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py b/tensorflow/contrib/fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py index e47342bc7d..4894298694 100644 --- a/tensorflow/contrib/fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py +++ b/tensorflow/contrib/fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py @@ -268,7 +268,7 @@ class FusedConv2DBiasActivationTest(test.TestCase): with self.cached_session() as sess: values = sess.run(tensors) for i in range(1, len(values)): - self.assertAllClose(values[0], values[i], rtol=1e-5, atol=1e-5) + self.assertAllClose(values[0], values[i], rtol=1e-3, atol=1e-3) def _VerifyValues(self, tensor_in_sizes, filter_in_sizes, bias, strides, padding): @@ -873,9 +873,7 @@ class FusedConvInt8Tests(test.TestCase): with self.test_session(use_gpu=True) as sess: actual_y, expected_y = sess.run([actual, expected]) - tf_logging.info("actual_y = ", actual_y) - tf_logging.info("expected_y = ", expected_y) - self.assertTrue(np.array_equal(actual_y, expected_y)) + self.assertAllClose(actual_y, expected_y, rtol=0, atol=1) def testFusedConvInt8(self): if not test.is_gpu_available( -- cgit v1.2.3