提交 faa77c77 authored 作者: lamblin's avatar lamblin

Merge pull request #1338 from pascanur/get_scalar_constant

fix for get_constant_value
......@@ -1347,9 +1347,9 @@ def scan_merge_inouts(node):
if equal_computations([sh], [ssh]):
return so
try:
vsh = int(opt.get_constant_value(sh))
vssh = int(opt.get_constant_value(ssh))
except TypeError:
vsh = int(opt.get_scalar_constant_value(sh))
vssh = int(opt.get_scalar_constant_value(ssh))
except tensor.NotScalarConstantError:
return o
if vsh == vssh:
return so
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论