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

Fix fft_ops_test.py for CPU

Browse Source
This commit is contained in:
Jonathan Hseu 2017-06-08 11:58:21 -07:00
parent 88d648f3be
commit 50d80ddf92
1 changed files with 27 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -298,6 +298,7 @@ class RFFTOpsTest(BaseFFTOpsTest):
use_placeholder=True) use_placeholder=True)
def testFftLength(self): def testFftLength(self):
if test.is_gpu_available(cuda_only=True):
for rank in VALID_FFT_RANKS: for rank in VALID_FFT_RANKS:
for dims in xrange(rank, rank + 3): for dims in xrange(rank, rank + 3):
for size in (5, 6): for size in (5, 6):