Commit 3107fda5 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

remove some code to have better code coverage

parent 74b05172
...@@ -460,18 +460,10 @@ struct onnx_parser ...@@ -460,18 +460,10 @@ struct onnx_parser
{ {
op::reshape op; op::reshape op;
if(args.size() == 1) if(args.size() == 1)
{
if(contains(attributes, "shape"))
{ {
literal s = parse_value(attributes.at("shape")); literal s = parse_value(attributes.at("shape"));
s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); }); 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) if(args.size() == 2)
{ {
auto s = args[1]->eval(); auto s = args[1]->eval();
......
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