diff options
| author |  Benjamin Kramer <kramerb@google.com> | 2018-09-24 20:54:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-24 20:58:57 -0700 |
| commit | df90003a68bcb813843e447d6fa2c49deccc48b6 (patch) | |
| tree | 22843a0374c4e0a150714dc2368eb1dccfbfb20f /tensorflow/contrib/fused_conv | |
| parent | 626fef2af7d4bc49aeeef7ffd195dc30235bcd1e (diff) | |
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
Diffstat (limited to 'tensorflow/contrib/fused_conv')
| -rw-r--r-- | tensorflow/contrib/fused_conv/python/ops/fused_conv2d_bias_activation_op_test.py | 6 |
1 files changed, 2 insertions, 4 deletions
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( |