提交 d05a2005 编写于 作者: 刘托

Merge branch 'fix_onnx_squeeze' into 'master'

fix onnx squeeze

See merge request !1118
......@@ -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.
先完成此消息的编辑!
想要评论请 注册