diff options
| author | James Conroy <james.conroy@arm.com> | 2022-06-21 11:31:47 +0000 |
|---|---|---|
| committer | James Conroy <james.conroy@arm.com> | 2022-06-21 11:31:47 +0000 |
| commit | a0f8b15d4ddb5075f380003ff31b271d389d3b66 (patch) | |
| tree | 5a4f351ee688bd760449c82c455b0e324b11f88d /delegate/src/armnn_delegate.cpp | |
| parent | 03bf98a8bc51ad20eef4b9ca5fbf6ce15e063721 (diff) | |
| download | armnn-a0f8b15d4ddb5075f380003ff31b271d389d3b66.tar.gz | |
Revert "IVGCVSW-6873 Import inputs but don't export outputs fails."
This reverts commit 03bf98a8bc51ad20eef4b9ca5fbf6ce15e063721.
Reason for revert: Caused failures in tests located in internal repo.
Change-Id: If35cb0ede349b270e4e7827324382e09455d8cfa
Diffstat (limited to 'delegate/src/armnn_delegate.cpp')
| -rw-r--r-- | delegate/src/armnn_delegate.cpp | 14 |
1 files changed, 4 insertions, 10 deletions
diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp index 1b6d68eb7a..bb2f3c319a 100644 --- a/delegate/src/armnn_delegate.cpp +++ b/delegate/src/armnn_delegate.cpp @@ -394,20 +394,14 @@ ArmnnSubgraph* ArmnnSubgraph::Create(TfLiteContext* tfLiteContext, // Load graph into runtime std::string errorMessage; armnn::Status loadingStatus; - armnn::MemorySource inputSource = armnn::MemorySource::Undefined; - armnn::MemorySource outputSource = armnn::MemorySource::Undefined; - // There's a bit of an assumption here that the delegate will only support Malloc memory source. + armnn::MemorySource memorySource = armnn::MemorySource::Undefined; if (delegate->m_Options.GetOptimizerOptions().m_ImportEnabled) { - inputSource = armnn::MemorySource::Malloc; - } - if (delegate->m_Options.GetOptimizerOptions().m_ExportEnabled) - { - outputSource = armnn::MemorySource::Malloc; + memorySource = armnn::MemorySource::Malloc; } armnn::INetworkProperties networkProperties(false, - inputSource, - outputSource, + memorySource, + memorySource, delegate->m_Options.GetInternalProfilingState(), delegate->m_Options.GetInternalProfilingDetail()); loadingStatus = delegate->m_Runtime->LoadNetwork(networkId, |