diff options
| author |  Gunhan Gulsoy <gunan@google.com> | 2017-08-28 20:46:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-28 20:50:29 -0700 |
| commit | 2fa844c88047afcc519ec60a1f74f1cbc03e99cc (patch) | |
| tree | 058340a1d6b0fab6065392620b9be0179be69e50 | |
| parent | ccad7da043fb10af3467bac989bff4b73db34210 (diff) | |
Double the tolerance in nn_grad_test to avoid flakes.
PiperOrigin-RevId: 166798376
| -rw-r--r-- | tensorflow/cc/gradients/nn_grad_test.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/cc/gradients/nn_grad_test.cc b/tensorflow/cc/gradients/nn_grad_test.cc index a02d36549b..f9a512f29e 100644 --- a/tensorflow/cc/gradients/nn_grad_test.cc +++ b/tensorflow/cc/gradients/nn_grad_test.cc @@ -36,7 +36,7 @@ class NNGradTest : public ::testing::Test { float max_error; TF_ASSERT_OK(ComputeGradientError(scope_, {x}, {x_shape}, {y}, {y_shape}, &max_error)); - EXPECT_LT(max_error, 1e-4); + EXPECT_LT(max_error, 2e-4); } void RunTest(const Output& x, const Tensor& x_init_value, const Output& y, @@ -44,7 +44,7 @@ class NNGradTest : public ::testing::Test { float max_error; TF_ASSERT_OK( ComputeGradientError(scope_, x, x_init_value, y, y_shape, &max_error)); - EXPECT_LT(max_error, 1e-4); + EXPECT_LT(max_error, 2e-4); } void RunTest(const OutputList& xs, const std::vector<TensorShape>& x_shapes, @@ -53,7 +53,7 @@ class NNGradTest : public ::testing::Test { float max_error; TF_ASSERT_OK( ComputeGradientError(scope_, xs, x_shapes, ys, y_shapes, &max_error)); - EXPECT_LT(max_error, 1e-4); + EXPECT_LT(max_error, 2e-4); } Scope scope_; |