diff options
| author | David Monahan <david.monahan@arm.com> | 2019-09-04 09:22:10 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-09-16 10:41:06 +0000 |
| commit | 4f1e8e47e29d42ed5862cccb29cf183853c4a86c (patch) | |
| tree | d3f758b6647f3ba5a065d87ea3db4e5fc989d1bd /src/armnn/Runtime.cpp | |
| parent | 0718ee9f26219da5a1362c1060e7db3feae8b3ce (diff) | |
| download | armnn-4f1e8e47e29d42ed5862cccb29cf183853c4a86c.tar.gz | |
IVGCVSW-3687 Add INetworkProperties to LoadNetwork
* Allows users to specify if Import/Export should be used
Signed-off-by: David Monahan <david.monahan@arm.com>
Change-Id: I64da26a6acbeb91ef72d31b6ccc01bb1447f624d
Diffstat (limited to 'src/armnn/Runtime.cpp')
| -rw-r--r-- | src/armnn/Runtime.cpp | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp index 9e874848ec..e47835687d 100644 --- a/src/armnn/Runtime.cpp +++ b/src/armnn/Runtime.cpp @@ -49,7 +49,16 @@ Status Runtime::LoadNetwork(NetworkId& networkIdOut, IOptimizedNetworkPtr inNetw Status Runtime::LoadNetwork(NetworkId& networkIdOut, IOptimizedNetworkPtr inNetwork, - std::string & errorMessage) + std::string& errorMessage) +{ + INetworkProperties networkProperties; + return LoadNetwork(networkIdOut, std::move(inNetwork), errorMessage, networkProperties); +} + +Status Runtime::LoadNetwork(NetworkId& networkIdOut, + IOptimizedNetworkPtr inNetwork, + std::string& errorMessage, + const INetworkProperties& networkProperties) { IOptimizedNetwork* rawNetwork = inNetwork.release(); @@ -62,7 +71,8 @@ Status Runtime::LoadNetwork(NetworkId& networkIdOut, unique_ptr<LoadedNetwork> loadedNetwork = LoadedNetwork::MakeLoadedNetwork( std::unique_ptr<OptimizedNetwork>(boost::polymorphic_downcast<OptimizedNetwork*>(rawNetwork)), - errorMessage); + errorMessage, + networkProperties); if (!loadedNetwork) { |