Call eval for reshape

36464c0c · Paul · f67fc560 · 36464c0c
Commit 36464c0c authored Feb 21, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

src/onnx/onnx.cpp src/onnx/onnx.cpp +3 -1

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -354,7 +354,9 @@ struct onnx_parser
        }
        if(args.size() == 2)
        {
-            literal s = args[1]->get_literal();
+            literal s = args[1]->eval();
+            if (s.empty())
+                MIGRAPHX_THROW("Dynamic shape is not supported.");
            s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
        }
        return prog.add_instruction(op, args[0]);