Only parse output nodes from onnx

a6e0400d · Paul · e15b8333 · a6e0400d
Commit a6e0400d authored Feb 28, 2019 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -1149,9 +1149,9 @@ struct onnx_parser
                instructions[name] = prog.add_parameter(name, s);
            }
        }
-        for(auto&& p : nodes)
+        for(auto&& output:graph.output())
        {
-            this->parse_node(p.first);
+            this->parse_node(output.name());
        }
    }