1 files changed, 19 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp
index ec19006be7..a0fa8af31b 100644
--- a/src/armnnOnnxParser/OnnxParser.hpp
+++ b/src/armnnOnnxParser/OnnxParser.hpp
@@ -34,12 +34,25 @@ public:
     /// Create the network from a protobuf binary file on disk
     armnn::INetworkPtr CreateNetworkFromBinaryFile(const char* graphFile);
 
+    /// Create the network from a protobuf binary file on disk, with inputShapes specified
+    armnn::INetworkPtr CreateNetworkFromBinaryFile(const char* graphFile,
+                                                   const std::map<std::string, armnn::TensorShape>& inputShapes);
+
     /// Create the network from a protobuf text file on disk
     armnn::INetworkPtr CreateNetworkFromTextFile(const char* graphFile);
 
+    /// Create the network from a protobuf text file on disk, with inputShapes specified
+    armnn::INetworkPtr CreateNetworkFromTextFile(const char* graphFile,
+                                                 const std::map<std::string, armnn::TensorShape>& inputShapes);
+
     /// Create the network directly from protobuf text in a string. Useful for debugging/testing
     armnn::INetworkPtr CreateNetworkFromString(const std::string& protoText);
 
+     /// Create the network directly from protobuf text in a string, with inputShapes specified.
+     /// Useful for debugging/testing
+    armnn::INetworkPtr CreateNetworkFromString(const std::string& protoText,
+                                               const std::map<std::string, armnn::TensorShape>& inputShapes);
+
     /// Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name
     BindingPointInfo GetNetworkInputBindingInfo(const std::string& name) const;
 
@@ -203,5 +216,11 @@ private:
 
     std::vector<UsageSummary> m_OutputsFusedAndUsed;
 
+    std::map<std::string, armnn::TensorShape> m_InputShapes;
+
+    std::unordered_map<std::string, armnn::TensorInfo> m_InputInfos;
+
+    std::unordered_map<std::string, armnn::TensorInfo> m_OutputInfos;
+
 };
 }