aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/WorkingMemHandle.hpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2021-04-07 20:10:49 +0100
committerfinn.williams <finn.williams@arm.com>2021-04-08 11:23:47 +0000
commit55a8ffda24fff5515803df10fb4863d46a1effdf (patch)
treee314dea48f22ae88d452527b2decaca61df108ad /src/armnn/WorkingMemHandle.hpp
parentb76eaed55a89330b3b448c4f4522b3fc94a4f38d (diff)
downloadarmnn-55a8ffda24fff5515803df10fb4863d46a1effdf.tar.gz
IVGCVSW-5823 Refactor Async Network API
* Moved IAsyncNetwork into IRuntime. * All LoadedNetworks can be executed Asynchronously. Signed-off-by: Mike Kelly <mike.kelly@arm.com> Change-Id: Ibbc901ab9110dc2f881425b75489bccf9ad54169
Diffstat (limited to 'src/armnn/WorkingMemHandle.hpp')
-rw-r--r--src/armnn/WorkingMemHandle.hpp12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/armnn/WorkingMemHandle.hpp b/src/armnn/WorkingMemHandle.hpp
index 090f180206..cef6fb6fd3 100644
--- a/src/armnn/WorkingMemHandle.hpp
+++ b/src/armnn/WorkingMemHandle.hpp
@@ -24,10 +24,17 @@ class WorkingMemHandle final : public IWorkingMemHandle
{
public:
- WorkingMemHandle(std::vector<WorkingMemDescriptor> workingMemDescriptors,
+ WorkingMemHandle(NetworkId networkId,
+ std::vector<WorkingMemDescriptor> workingMemDescriptors,
std::unordered_map<LayerGuid, WorkingMemDescriptor> workingMemDescriptorMap);
- ~WorkingMemHandle() { FreeWorkingMemory(); }
+ ~WorkingMemHandle()
+ { FreeWorkingMemory(); }
+
+ NetworkId GetNetworkId() override
+ {
+ return m_NetworkId;
+ }
/// Allocate the backing memory required for execution. If this is not called, then allocation will be
/// deferred to execution time. The mutex must be locked.
@@ -106,6 +113,7 @@ public:
private:
void FreeWorkingMemory();
+ NetworkId m_NetworkId;
std::shared_ptr<ProfilerImpl> m_Profiler;
std::vector<WorkingMemDescriptor> m_WorkingMemDescriptors;