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

Annotate a few more tests to run in graph mode

Browse Source 
PiperOrigin-RevId: 236505757
This commit is contained in:
Gaurav Jain 2019-03-02 21:50:08 -08:00 committed by TensorFlower Gardener
parent 5ca2639d07
commit bfa3fcead3
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/kernel_tests/record_input_test.py
View File

@ -125,6 +125,7 @@ class RecordInputOpTest(test.TestCase):
self.assertTrue(r[0] not in epoch_set) self.assertTrue(r[0] not in epoch_set)
epoch_set.add(r[0]) epoch_set.add(r[0])
@test_util.run_deprecated_v1
def testDoesNotDeadlock(self): def testDoesNotDeadlock(self):
# Iterate multiple times to cause deadlock if there is a chance it can occur # Iterate multiple times to cause deadlock if there is a chance it can occur
for _ in range(30): for _ in range(30):

1
tensorflow/python/kernel_tests/signal/fft_ops_test.py
View File

@ -399,6 +399,7 @@ class RFFTOpsTest(BaseFFTOpsTest):
rank, (size,) * rank, rank, (size,) * rank,
use_placeholder=True) use_placeholder=True)
@test_util.run_deprecated_v1
def testFftLength(self): def testFftLength(self):
if test.is_gpu_available(cuda_only=True): if test.is_gpu_available(cuda_only=True):
with spectral_ops_test_util.fft_kernel_label_map(): with spectral_ops_test_util.fft_kernel_label_map():