diff options
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 |