diff --git a/tensorflow/python/kernel_tests/record_input_test.py b/tensorflow/python/kernel_tests/record_input_test.py index ad8188b372f..a37eca69add 100644 --- a/tensorflow/python/kernel_tests/record_input_test.py +++ b/tensorflow/python/kernel_tests/record_input_test.py @@ -125,6 +125,7 @@ class RecordInputOpTest(test.TestCase): self.assertTrue(r[0] not in epoch_set) epoch_set.add(r[0]) + @test_util.run_deprecated_v1 def testDoesNotDeadlock(self): # Iterate multiple times to cause deadlock if there is a chance it can occur for _ in range(30): diff --git a/tensorflow/python/kernel_tests/signal/fft_ops_test.py b/tensorflow/python/kernel_tests/signal/fft_ops_test.py index a9fbe2975ed..9b66f3eb2a0 100644 --- a/tensorflow/python/kernel_tests/signal/fft_ops_test.py +++ b/tensorflow/python/kernel_tests/signal/fft_ops_test.py @@ -399,6 +399,7 @@ class RFFTOpsTest(BaseFFTOpsTest): rank, (size,) * rank, use_placeholder=True) + @test_util.run_deprecated_v1 def testFftLength(self): if test.is_gpu_available(cuda_only=True): with spectral_ops_test_util.fft_kernel_label_map():