From ffe13572cefdacc470ea9f83117db94718563d3a Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 1 Mar 2019 15:59:38 -0800 Subject: [PATCH] Gets functional ops tests to pass in tf2 on gpu PiperOrigin-RevId: 236394682 --- tensorflow/python/kernel_tests/functional_ops_test.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tensorflow/python/kernel_tests/functional_ops_test.py b/tensorflow/python/kernel_tests/functional_ops_test.py index 0ab2b4bdfb5..62e0bac9493 100644 --- a/tensorflow/python/kernel_tests/functional_ops_test.py +++ b/tensorflow/python/kernel_tests/functional_ops_test.py @@ -475,6 +475,7 @@ class FunctionalOpsTest(test.TestCase): mul = self.evaluate(remote_op) self.assertEqual(mul, [6]) + @test_util.run_deprecated_v1 def testRemoteFunctionCPUGPU(self): if not test_util.is_gpu_available(): self.skipTest("No GPU available") @@ -499,6 +500,7 @@ class FunctionalOpsTest(test.TestCase): mul = self.evaluate(remote_op) self.assertEqual(mul, 9.0) + @test_util.run_deprecated_v1 def testRemoteFunctionGPUCPU(self): if not test_util.is_gpu_available(): self.skipTest("No GPU available") @@ -523,6 +525,7 @@ class FunctionalOpsTest(test.TestCase): mul = self.evaluate(remote_op) self.assertEqual(mul, 9.0) + @test_util.run_deprecated_v1 def testRemoteFunctionGPUCPUStrings(self): if not test_util.is_gpu_available(): self.skipTest("No GPU available") @@ -981,6 +984,7 @@ class PartitionedCallTest(test.TestCase): constant_op.constant(2.)], f=Body) self.assertEqual(output.eval(), 12.) + @test_util.run_deprecated_v1 def testBasicMultiDeviceGPU(self): if not test_util.is_gpu_available(): return @@ -1061,6 +1065,7 @@ class PartitionedCallTest(test.TestCase): value = self.evaluate(v.read_value()) self.assertEqual(value, 2.0) + @test_util.run_deprecated_v1 def testFunctionWithResourcesOnDifferentDevices(self): if not test_util.is_gpu_available(): self.skipTest("No GPUs available.")