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

Fix gh-3663

上级 52185277
......@@ -693,12 +693,9 @@ class PushOutScanOutput(gof.Optimizer):
args = scan_args(node.inputs, node.outputs,
op.inputs, op.outputs, op.info)
local_fgraph = gof.FunctionGraph(args.inner_inputs,
args.inner_outputs,
clone=False)
new_scan_node = None
local_fgraph_topo = local_fgraph.toposort()
local_fgraph_topo = theano.gof.graph.io_toposort(op.inputs, op.outputs)
for nd in local_fgraph_topo:
if (isinstance(nd.op, theano.tensor.Dot) and
nd.out in args.inner_out_nit_sot):
......@@ -860,7 +857,6 @@ class PushOutScanOutput(gof.Optimizer):
reason="scanOp_pushout_output")
break
return new_scan_node
def inner_sitsot_only_last_step_used(self, var, scan_args):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论