From 1b11f32dbfea8383956c5d2c60b034469194f6d9 Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Wed, 13 Oct 2021 11:44:50 +0100 Subject: IVGCVSW-6450 Add Support of Models with Dynamic Batch Tensor to ONNX parser Signed-off-by: Narumol Prangnawarat Change-Id: Ia7dbf0735619d406d6b4e34a71f14f20d92586e6 --- include/armnnOnnxParser/IOnnxParser.hpp | 13 +++++++++++++ 1 file changed, 13 insertions(+) (limited to 'include/armnnOnnxParser/IOnnxParser.hpp') 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& inputShapes); + + /// Create the network from a protobuf text file on disk, with inputShapes specified + armnn::INetworkPtr CreateNetworkFromTextFile(const char* graphFile, + const std::map& 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& 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; -- cgit v1.2.1