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

Fix crash when scan has only non_sequences inputs

上级 9147b12e
...@@ -1347,12 +1347,6 @@ class Scan(PureOp): ...@@ -1347,12 +1347,6 @@ class Scan(PureOp):
return self.outputs[s:e] return self.outputs[s:e]
def _get_inner_inps(iidx): def _get_inner_inps(iidx):
s = 0
if self.n_seqs > 0:
e = 1
else:
e = len(self.tap_array[0])
p = iidx
if node.inputs[iidx + 1] in self.outer_nitsot(node): if node.inputs[iidx + 1] in self.outer_nitsot(node):
return None return None
if node.inputs[iidx + 1] in self.outer_non_seqs(node): if node.inputs[iidx + 1] in self.outer_non_seqs(node):
...@@ -1360,6 +1354,11 @@ class Scan(PureOp): ...@@ -1360,6 +1354,11 @@ class Scan(PureOp):
node.inputs[iidx + 1]) node.inputs[iidx + 1])
return [self.inner_non_seqs(self.inputs)[loc_idx]] return [self.inner_non_seqs(self.inputs)[loc_idx]]
s = 0
if self.n_seqs > 0:
e = 1
else:
e = len(self.tap_array[0])
for p in xrange(iidx): for p in xrange(iidx):
s = e s = e
if p < self.n_seqs: if p < self.n_seqs:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论