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

Make CPU conv2d test use floatX.

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