Remove shape dict definition in tvmc.

apache · Dec 16, 2020 · 895954f · 895954f
1 parent 389820c
commit 895954f
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/python/tvm/driver/tvmc/frontends.py b/python/tvm/driver/tvmc/frontends.py
@@ -161,16 +161,7 @@ def load(self, path):
         # pylint: disable=E1101
         model = onnx.load(path)
 
-        # pylint: disable=E1101
-        name = model.graph.input[0].name
-
-        # pylint: disable=E1101
-        proto_shape = model.graph.input[0].type.tensor_type.shape.dim
-        shape = [d.dim_value for d in proto_shape]
-
-        shape_dict = {name: shape}
-
-        return relay.frontend.from_onnx(model, shape_dict)
+        return relay.frontend.from_onnx(model)
 
 
 class TensorflowFrontend(Frontend):