提交 d05a2005 编写于 作者: 刘托

Merge branch 'fix_onnx_squeeze' into 'master'

fix onnx squeeze

See merge request !1118
...@@ -1395,8 +1395,8 @@ class OnnxConverter(base_converter.ConverterInterface): ...@@ -1395,8 +1395,8 @@ class OnnxConverter(base_converter.ConverterInterface):
op.type = MaceOp.Squeeze.name op.type = MaceOp.Squeeze.name
axis_arg = op.arg.add() axis_arg = op.arg.add()
axis_arg.name = MaceKeyword.mace_axis_str axis_arg.name = MaceKeyword.mace_axis_str
if 'axis' in node.attrs: if 'axes' in node.attrs:
axis_value = node.attrs['axis'] axis_value = node.attrs['axes']
else: else:
axis_value = [] axis_value = []
axis_arg.ints.extend(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.
先完成此消息的编辑!
想要评论请 注册