提交 bdcb0c0d authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Correct bug in PushOutScanOutput unittests where optimization was not properly excluded

上级 58dc930e
......@@ -160,9 +160,9 @@ class TestPushOutScanOutputDot(object):
f_opt = theano.function([v, m], T.jacobian(output, v))
default_mode = theano.compile.get_default_mode()
default_mode.excluding("scanOp_pushout_output")
new_mode = default_mode.excluding("scanOp_pushout_output")
f_no_opt = theano.function([v, m], T.jacobian(output, v),
mode=default_mode)
mode=new_mode)
# Ensure that the optimization was performed correctly in f_opt
# The inner function of scan should have only one output and it should
......@@ -251,8 +251,8 @@ class TestPushOutScanOutputDot(object):
f_opt = theano.function([a, b], outputs)
default_mode = theano.compile.get_default_mode()
default_mode.excluding("scanOp_pushout_output")
f_no_opt = theano.function([a, b], outputs, mode=default_mode)
new_mode = default_mode.excluding("scanOp_pushout_output")
f_no_opt = theano.function([a, b], outputs, mode=new_mode)
# Ensure that the optimization was performed correctly in f_opt
# The inner function of scan should have only one output and it should
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论