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