diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-01-29 15:38:54 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-02-02 13:53:38 +0000 |
commit | 74a3cf5755b801cf258177e8e55b4cda64a0c351 (patch) | |
tree | 01246003e2b336e75095d148000370a17e790511 /delegate/src/armnn_delegate.cpp | |
parent | bd4fcc30a3732ce63e100d556ee5c9ea9e556c05 (diff) | |
download | armnn-74a3cf5755b801cf258177e8e55b4cda64a0c351.tar.gz |
IVGCVSW-5619 Enable OptimizerOptions for the python external delegate
* Add reduce-fp32-to-fp16, reduce-fp32-to-bf16, debug-data, memory-import
options to external delegate
* Simplify DelegateOptions
* Add test mock models
* Unit tests
* Configure lfs to manage tflite files
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I1e4db468862ba03d4cb031347bc307cf940b3cb1
Diffstat (limited to 'delegate/src/armnn_delegate.cpp')
-rw-r--r-- | delegate/src/armnn_delegate.cpp | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp index 6f72d864b9..639e514a78 100644 --- a/delegate/src/armnn_delegate.cpp +++ b/delegate/src/armnn_delegate.cpp @@ -353,10 +353,21 @@ ArmnnSubgraph* ArmnnSubgraph::Create(TfLiteContext* tfLiteContext, { // Load graph into runtime std::string errorMessage; - auto loadingStatus = delegate->m_Runtime->LoadNetwork(networkId, - std::move(optNet), - errorMessage, - delegate->m_Options.GetNetworkProperties()); + armnn::Status loadingStatus; + if (delegate->m_Options.GetOptimizerOptions().m_ImportEnabled) + { + armnn::INetworkProperties networkProperties(true, true); + loadingStatus = delegate->m_Runtime->LoadNetwork(networkId, + std::move(optNet), + errorMessage, + networkProperties); + } + else + { + loadingStatus = delegate->m_Runtime->LoadNetwork(networkId, + std::move(optNet), + errorMessage); + } if (loadingStatus != armnn::Status::Success) { // Optimize failed |