remove some code to have better code coverage

3107fda5 · Shucai Xiao · 74b05172 · 3107fda5
Commit 3107fda5 authored Jul 24, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 10 deletions

src/onnx/onnx.cpp src/onnx/onnx.cpp +2 -10

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -461,16 +461,8 @@ struct onnx_parser
        op::reshape op;
        if(args.size() == 1)
        {
-            if(contains(attributes, "shape"))
+            literal s = parse_value(attributes.at("shape"));
-            {
+            s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
-                literal s = parse_value(attributes.at("shape"));
-                s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
-            }
-            else
-            {
-                MIGRAPHX_THROW(
-                    "Parse_reshape: shape attribute is needed when only one argument is provided!");
-            }
        }
        if(args.size() == 2)
        {