diff options
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 19 |
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; + }; } |