diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-10-13 11:44:50 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-10-18 08:32:02 +0000 |
commit | 1b11f32dbfea8383956c5d2c60b034469194f6d9 (patch) | |
tree | 3bd3f73e9af499778db894c3db18dc7b5f4ee668 /include/armnnOnnxParser/IOnnxParser.hpp | |
parent | ea0712e72080b794fa864e67d073d3bfe2eda0f1 (diff) | |
download | armnn-1b11f32dbfea8383956c5d2c60b034469194f6d9.tar.gz |
IVGCVSW-6450 Add Support of Models with Dynamic Batch Tensor to ONNX parser
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ia7dbf0735619d406d6b4e34a71f14f20d92586e6
Diffstat (limited to 'include/armnnOnnxParser/IOnnxParser.hpp')
-rw-r--r-- | include/armnnOnnxParser/IOnnxParser.hpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/include/armnnOnnxParser/IOnnxParser.hpp b/include/armnnOnnxParser/IOnnxParser.hpp index f9d692738d..ba7fc83f93 100644 --- a/include/armnnOnnxParser/IOnnxParser.hpp +++ b/include/armnnOnnxParser/IOnnxParser.hpp @@ -36,6 +36,19 @@ public: /// Create the network directly from protobuf text in a string. Useful for debugging/testing armnn::INetworkPtr CreateNetworkFromString(const std::string& protoText); + /// 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, 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, 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; |