提交 0cd92a60 authored 作者: Frederic's avatar Frederic

Fix test in FAST_COMPILE and floatX=float32

上级 10dcdb38
...@@ -802,7 +802,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -802,7 +802,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
MaxPoolGrad, MaxPoolGrad,
warn=False) warn=False)
# checking with broadcastable input # checking with broadcastable input
image = tensor.tensor(dtype=theano.config.floatX, image = tensor.tensor(dtype='float64',
broadcastable=(False, False, True, True)) broadcastable=(False, False, True, True))
image_val = rng.rand(4, 6, 1, 1) image_val = rng.rand(4, 6, 1, 1)
self._compile_and_check( self._compile_and_check(
......
...@@ -4021,7 +4021,8 @@ class T_Rebroadcast(unittest.TestCase): ...@@ -4021,7 +4021,8 @@ class T_Rebroadcast(unittest.TestCase):
class T_useless_elemwise(unittest.TestCase): class T_useless_elemwise(unittest.TestCase):
def setUp(self): def setUp(self):
self.mode = theano.compile.get_default_mode().including('canonicalize') self.mode = theano.compile.get_default_mode().including(
'canonicalize', 'local_fill_to_alloc')
def test_eq(self): def test_eq(self):
x = T.dmatrix() x = T.dmatrix()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论