提交 65db8694 authored 作者: Harm de Vries's avatar Harm de Vries

fix opt max optimization

上级 ddc0e308
...@@ -63,7 +63,7 @@ def local_max_and_argmax(node): ...@@ -63,7 +63,7 @@ def local_max_and_argmax(node):
axis = get_scalar_constant_value(node.inputs[1]) axis = get_scalar_constant_value(node.inputs[1])
except NotScalarConstantError: except NotScalarConstantError:
axis = node.inputs[1] axis = node.inputs[1]
if not isinstance(axis, TensorConstant): if not isinstance(axis, T.TensorConstant):
return False return False
axis = axis.data axis = axis.data
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论