experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Slightly relax test tolerance to fix failure in xla_gpu version of image preprocessing test.

Browse Source 
PiperOrigin-RevId: 317495267
Change-Id: I47cc7556f207e7f8061d68320b764c2663bcccc1
This commit is contained in:
A. Unique TensorFlower 2020-06-20 17:06:03 -07:00 committed by TensorFlower Gardener
parent 793c5ca70b
commit 443e2ae284
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/layers/preprocessing/image_preprocessing_test.py
View File

@ -976,7 +976,7 @@ class RandomRotationTest(keras_parameterized.TestCase):
output = strat.run(lambda: layer(input_images, training=True))
values = output.values
self.assertAllEqual(2, len(values))
self.assertAllClose(values[0], values[1])
self.assertAllClose(values[0], values[1], rtol=1e-5)
@tf_test_util.run_v2_only
def test_config_with_custom_name(self):