提交 fc2b9fc0 authored 作者: Frederic Bastien's avatar Frederic Bastien

Speed up some tests

上级 60a7c53c
......@@ -380,7 +380,7 @@ class TestRepeatOp(utt.InferShapeTester):
self.numpy_unsupported_dtypes = ('uint32', 'int64', 'uint64')
def test_repeatOp(self):
for ndim in range(3):
for ndim in [1, 3]:
x = T.TensorType(config.floatX, [False] * ndim)()
a = np.random.random((10, ) * ndim).astype(config.floatX)
......@@ -438,13 +438,13 @@ class TestRepeatOp(utt.InferShapeTester):
@attr('slow')
def test_infer_shape(self):
for ndim in range(4):
for ndim in [1, 3]:
x = T.TensorType(config.floatX, [False] * ndim)()
shp = (np.arange(ndim) + 1) * 5
shp = (np.arange(ndim) + 1) * 3
a = np.random.random(shp).astype(config.floatX)
for axis in self._possible_axis(ndim):
for dtype in tensor.integer_dtypes:
for dtype in ["int8", "uint8", "uint64"]:
r_var = T.scalar(dtype=dtype)
r = np.asarray(3, dtype=dtype)
if dtype in self.numpy_unsupported_dtypes:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论