提交 b093421d authored 作者: Frederic's avatar Frederic

make test smaller to be faster in debug mode.

上级 1f2a52e2
...@@ -204,9 +204,9 @@ class TestFourier(utt.InferShapeTester): ...@@ -204,9 +204,9 @@ class TestFourier(utt.InferShapeTester):
def fft_test4(a): def fft_test4(a):
return self.op(a, 4, 0) return self.op(a, 4, 0)
pts = [numpy.random.rand(7, 2, 4, 3), pts = [numpy.random.rand(5, 2, 4, 3),
numpy.random.rand(5, 5, 4), numpy.random.rand(2, 3, 4),
numpy.random.rand(2, 9), numpy.random.rand(2, 5),
numpy.random.rand(5)] numpy.random.rand(5)]
for fft_test in [fft_test1, fft_test2, fft_test3, fft_test4]: for fft_test in [fft_test1, fft_test2, fft_test3, fft_test4]:
for pt in pts: for pt in pts:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论