Skip to content

T
tic
Commit 075b3677 by nhynes Committed by Yizhi Liu
Options

Rename axis parameter in onnx squeeze (#1683) 

* Rename axis parameter in onnx squeeze

* Add test
parent 5fec9adb
Showing with 2 additions and 2 deletions
nnvm/python/nnvm/frontend/onnx.py
...@@ -684,7 +684,7 @@ def _get_convert_map(opset): ...@@ -684,7 +684,7 @@ def _get_convert_map(opset):
'Slice': Slice.get_converter(opset), 'Slice': Slice.get_converter(opset),
'Transpose': AttrCvt('transpose', {'perm': 'axes'}), 'Transpose': AttrCvt('transpose', {'perm': 'axes'}),
'Gather': Gather.get_converter(opset), 'Gather': Gather.get_converter(opset),
'Squeeze': Renamer('squeeze'), 'Squeeze': AttrCvt('squeeze', {'axes': 'axis'}),
'Unsqueeze': Unsqueeze.get_converter(opset), 'Unsqueeze': Unsqueeze.get_converter(opset),
'Pad': Pad.get_converter(opset), 'Pad': Pad.get_converter(opset),
'Shape': Shape.get_converter(opset), 'Shape': Shape.get_converter(opset),
......
nnvm/tests/python/frontend/onnx/test_forward.py
...@@ -178,7 +178,7 @@ def test_power(): ...@@ -178,7 +178,7 @@ def test_power():
def test_squeeze(): def test_squeeze():
in_shape = (1, 3, 1, 3, 1, 1) in_shape = (1, 3, 1, 3, 1, 1)
out_shape = (3, 3) out_shape = (3, 3)
y = helper.make_node("Squeeze", ['in'], ['out']) y = helper.make_node("Squeeze", ['in'], ['out'], axes=[0, 2, 4, 5])
graph = helper.make_graph([y], graph = helper.make_graph([y],
'squeeze_test', 'squeeze_test',
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment