Fixed extra reshape parameter bug. (#4524)

f221844b · Josh Fromm · Tianqi Chen · f5e2c44e · f221844b
Commit f221844b authored Dec 15, 2019 by Josh Fromm Committed by Tianqi Chen Dec 15, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletions

python/tvm/relay/frontend/onnx.py
+2 -1

No files found.
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -463,7 +463,8 @@ class Reshape(OnnxOpConverter):
    @classmethod
    def _impl_v5(cls, inputs, attr, params):
        if get_name(inputs[1]) in params:
-            shape = tuple(params[inputs[1].name_hint].asnumpy())
+            # pop shape out of parameters since it wont be needed later.
+            shape = tuple(params.pop(inputs[1].name_hint).asnumpy())
            out = _op.reshape(inputs[0], shape)
        else:
            data, shape = inputs