diff --git a/tensorflow/compiler/tests/image_ops_test.py b/tensorflow/compiler/tests/image_ops_test.py index 12791ef8ac1..5b19e993ece 100644 --- a/tensorflow/compiler/tests/image_ops_test.py +++ b/tensorflow/compiler/tests/image_ops_test.py @@ -37,6 +37,10 @@ from tensorflow.python.platform import test class RGBToHSVTest(XLATestCase): def testBatch(self): + # TODO(b/78230407): Reenable the test on GPU. + if self.device == "XLA_GPU": + return + # Build an arbitrary RGB image np.random.seed(7) batch_size = 5