Skip to content

T
tic
Commit 42d81cca by Wenhao Hu Committed by Tianqi Chen
Options

raise error when cannot get shape in reshape (#467) 

* raise error when cannot get shape in reshape

* fix pylint
parent 73775f45
Showing with 7 additions and 3 deletions
nnvm/python/nnvm/frontend/onnx.py
...@@ -246,6 +246,8 @@ class Reciprocal(OnnxOpConverter): ...@@ -246,6 +246,8 @@ class Reciprocal(OnnxOpConverter):
class Reshape(OnnxOpConverter): class Reshape(OnnxOpConverter):
""" Operator converter for Reshape.
"""
@classmethod @classmethod
def _impl_v1(cls, inputs, attr, params): def _impl_v1(cls, inputs, attr, params):
...@@ -253,9 +255,11 @@ class Reshape(OnnxOpConverter): ...@@ -253,9 +255,11 @@ class Reshape(OnnxOpConverter):
@classmethod @classmethod
def _impl_v5(cls, inputs, attr, params): def _impl_v5(cls, inputs, attr, params):
return _sym.reshape( if inputs[1].list_output_names()[0] in params:
inputs[0], shape = tuple(params[inputs[1].list_output_names()[0]].asnumpy())
shape=tuple(params[inputs[1].list_output_names()[0]].asnumpy())) else:
raise RuntimeError('Shape is not contained in graph initializer.')
return _sym.reshape(inputs[0], shape=shape)
class Scale(OnnxOpConverter): class Scale(OnnxOpConverter):
......
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