提交 dd5cfff5 authored 作者: sebastien-j's avatar sebastien-j

Use optimization with FAST_COMPILE too

上级 162306f8
...@@ -807,11 +807,15 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -807,11 +807,15 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
maxout = theano.tensor.tensor4() maxout = theano.tensor.tensor4()
grad = theano.tensor.tensor4() grad = theano.tensor.tensor4()
compilation_mode=theano.compile.get_default_mode().including(
'local_average_pool_grad')
for mode in ['max', 'sum', 'average_inc_pad', 'average_exc_pad']: for mode in ['max', 'sum', 'average_inc_pad', 'average_exc_pad']:
f = theano.function([im, maxout, grad], f = theano.function([im, maxout, grad],
DownsampleFactorMaxGrad(ds=(3,3), DownsampleFactorMaxGrad(ds=(3,3),
ignore_border=False, ignore_border=False,
mode=mode)(im, maxout, grad)) mode=mode)(im, maxout, grad),
mode=compilation_mode)
if mode == 'max': if mode == 'max':
assert any(isinstance(n.op, MaxPoolGrad) assert any(isinstance(n.op, MaxPoolGrad)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论