diff options
author | Mike Kelly <mike.kelly@arm.com> | 2022-11-25 13:55:24 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-01-03 14:45:41 +0000 |
commit | 2ae3224c559dcd3033be1bfd41be08113048dc50 (patch) | |
tree | e130d940fb6d080807dc29bf0c9602e1bd30af4a /src/armnnOnnxParser/OnnxParser.hpp | |
parent | a09e0336d3e89931e8e0d43010197155a45d2ec7 (diff) | |
download | armnn-2ae3224c559dcd3033be1bfd41be08113048dc50.tar.gz |
GitHub #709 Provide a CreateNetworkFromBinary method for the ONNX parser
* Added CreateNetworkFromBinary to the ONNX parser
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I5ca72ee49c7b098f9fb4aaf55a8bc077230cb30e
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index bb94472c6d..c9f321a5b5 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -38,6 +38,13 @@ public: armnn::INetworkPtr CreateNetworkFromBinaryFile(const char* graphFile, const std::map<std::string, armnn::TensorShape>& inputShapes); + /// Create the network from a protobuf binary + armnn::INetworkPtr CreateNetworkFromBinary(const std::vector<uint8_t>& binaryContent); + + /// Create the network from a protobuf binary, 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 text file on disk armnn::INetworkPtr CreateNetworkFromTextFile(const char* graphFile); @@ -64,6 +71,7 @@ public: OnnxParserImpl(); ~OnnxParserImpl() = default; + static ModelPtr LoadModelFromBinary(const std::vector<uint8_t>& binaryContent); static ModelPtr LoadModelFromBinaryFile(const char * fileName); static ModelPtr LoadModelFromTextFile(const char * fileName); static ModelPtr LoadModelFromString(const std::string& inputString); |