diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnnOnnxParser/IOnnxParser.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/include/armnnOnnxParser/IOnnxParser.hpp b/include/armnnOnnxParser/IOnnxParser.hpp index ba7fc83f93..89c22c03de 100644 --- a/include/armnnOnnxParser/IOnnxParser.hpp +++ b/include/armnnOnnxParser/IOnnxParser.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017,2022 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -27,6 +27,13 @@ public: static IOnnxParserPtr Create(); static void Destroy(IOnnxParser* parser); + /// Create the network from a protobuf binary vector + armnn::INetworkPtr CreateNetworkFromBinary(const std::vector<uint8_t>& binaryContent); + + /// Create the network from a protobuf binary vector, with inputShapes specified + armnn::INetworkPtr CreateNetworkFromBinary(const std::vector<uint8_t>& binaryContent, + const std::map<std::string, armnn::TensorShape>& inputShapes); + /// Create the network from a protobuf binary file on disk armnn::INetworkPtr CreateNetworkFromBinaryFile(const char* graphFile); |