提交 04f79550 authored 作者: Frederic's avatar Frederic

Make CPU conv2d test use floatX.

上级 96bf815a
...@@ -13,13 +13,13 @@ from theano.tensor.basic import _allclose, NotScalarConstantError ...@@ -13,13 +13,13 @@ from theano.tensor.basic import _allclose, NotScalarConstantError
class TestConv2D(utt.InferShapeTester): class TestConv2D(utt.InferShapeTester):
mode = None mode = None
dtype = 'float64' dtype = theano.config.floatX
def setUp(self): def setUp(self):
super(TestConv2D, self).setUp() super(TestConv2D, self).setUp()
self.input = T.dtensor4('input') self.input = T.tensor4('input', dtype=self.dtype)
self.input.name = 'default_V' self.input.name = 'default_V'
self.filters = T.dtensor4('filters') self.filters = T.tensor4('filters', dtype=self.dtype)
self.filters.name = 'default_filters' self.filters.name = 'default_filters'
if not conv.imported_scipy_signal and theano.config.cxx == "": if not conv.imported_scipy_signal and theano.config.cxx == "":
raise SkipTest("conv2d tests need SciPy or a c++ compiler") raise SkipTest("conv2d tests need SciPy or a c++ compiler")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论