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

Fix test in FAST_COMPILE and floatX=float32

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