message: Increase rtol and atol for flaky test.
diff: @@ -259,8 +259,8 @@ class GRUV2Test(keras_parameterized.TestCase):
  canonical_model.set_weights(weights)
  y_3 = canonical_model.predict(x_train)
  - self.assertAllClose(y_1, y_2)
  - self.assertAllClose(y_2, y_3)
  + self.assertAllClose(y_1, y_2, rtol=1e-5, atol=1e-5)
  + self.assertAllClose(y_2, y_3, rtol=1e-5, atol=1e-5)
  @parameterized.named_parameters(
  # test_name, time_major, go_backwards
  