Added code to read initializer data from ONNX

src/include/migraph/shape.hpp

@@ -80,11 +80,13 @@ struct shape
 
     /// Returns true if the shape is packed with no padding
     bool packed() const;
-    /// Returns true is the shape has been transposed. That is the strides are not in descending order
+    /// Returns true is the shape has been transposed. That is the strides are not in descending
+    /// order
     bool transposed() const;
     /// Returns true if the shape is broadcasting a dimension. That is, one of the strides are zero
     bool broadcasted() const;
-    /// Returns true if the shape is in its standard format. That is, the shape is both packed and not transposed.
+    /// Returns true if the shape is in its standard format. That is, the shape is both packed and
+    /// not transposed.
     bool standard() const;
 
     friend bool operator==(const shape& x, const shape& y);

src/onnx/CMakeLists.txt

@@ -20,6 +20,10 @@ add_executable(mnist mnist.cpp)
 rocm_clang_tidy_check(mnist)
 target_link_libraries(mnist migraph_cpu migraph_onnx)
 
+add_executable(resnet18 resnet18.cpp)
+rocm_clang_tidy_check(resnet18)
+target_link_libraries(resnet18 migraph_cpu migraph_onnx)
+
 if(MIGRAPH_ENABLE_GPU)
 add_executable(verify_onnx verify_onnx.cpp)
 rocm_clang_tidy_check(verify_onnx)

src/onnx/onnx.cpp

@@ -285,13 +285,29 @@ struct onnx_parser
     void parse_graph(const onnx::GraphProto& graph)
     {
         nodes = get_nodes(graph);
+        std::unordered_map<std::string, size_t> initializer_data;
+        auto cnt = 0;
+        for(auto&& f : graph.initializer())
+        {
+            initializer_data[f.name()] = cnt++;
+        }
         for(auto&& input : graph.input())
         {
             const std::string& name = input.name();
+            // Does the input have an initializer?
+            if(initializer_data.find(name) != initializer_data.end())
+            {
+                auto idx           = initializer_data[name];
+                auto t             = graph.initializer()[idx];
+                instructions[name] = prog.add_literal(parse_tensor(t));
+            }
+            else
+            {
                 // TODO: Get shape of input parameter
                 shape s            = parse_type(input.type());
                 instructions[name] = prog.add_parameter(name, s);
             }
+        }
         for(auto&& p : nodes)
         {
             this->parse_node(get_name(p.second));