提交 a53cc08c authored 作者: Frederic's avatar Frederic

Fix test in FAST_COMPILE.

上级 842732a7
...@@ -2491,6 +2491,7 @@ def test_local_IncSubtensor_serialize(): ...@@ -2491,6 +2491,7 @@ def test_local_IncSubtensor_serialize():
cost = T.sqr(t - y) cost = T.sqr(t - y)
dW = theano.grad(cost, W) dW = theano.grad(cost, W)
mode = theano.compile.mode.get_default_mode().excluding('fusion') mode = theano.compile.mode.get_default_mode().excluding('fusion')
mode = mode.including("local_IncSubtensor_serialize")
f = theano.function([i, j, t], updates=[(W, W - 0.01 * dW)], mode=mode) f = theano.function([i, j, t], updates=[(W, W - 0.01 * dW)], mode=mode)
topo = f.maker.fgraph.toposort() topo = f.maker.fgraph.toposort()
adds = [n for n in topo if isinstance(n.op, T.Elemwise) and adds = [n for n in topo if isinstance(n.op, T.Elemwise) and
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论