提交 9d49e951 编写于 作者: Y yejianwu

fix onnx squeeze

上级 7fa03dd2
......@@ -1395,8 +1395,8 @@ class OnnxConverter(base_converter.ConverterInterface):
op.type = MaceOp.Squeeze.name
axis_arg = op.arg.add()
axis_arg.name = MaceKeyword.mace_axis_str
if 'axis' in node.attrs:
axis_value = node.attrs['axis']
if 'axes' in node.attrs:
axis_value = node.attrs['axes']
else:
axis_value = []
axis_arg.ints.extend(axis_value)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册