提交 c5f73396 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

fix bug in scan_savemem optimization

the index goes from 0 to n_mit_mot-1 for mit_mot sequnces.
上级 e2537ddc
...@@ -475,7 +475,7 @@ class ScanSaveMem(gof.Optimizer): ...@@ -475,7 +475,7 @@ class ScanSaveMem(gof.Optimizer):
break break
# 2.3.2 extract the begin/end of the first dimension # 2.3.2 extract the begin/end of the first dimension
if i > op.n_mit_mot: if i >= op.n_mit_mot:
try: try:
length = shape_of[out][0] length = shape_of[out][0]
except KeyError: except KeyError:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论