diff options
Diffstat (limited to 'src/armnn/WorkingMemHandle.cpp')
| -rw-r--r-- | src/armnn/WorkingMemHandle.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/WorkingMemHandle.cpp b/src/armnn/WorkingMemHandle.cpp index 7a901b296b..c1a48d482f 100644 --- a/src/armnn/WorkingMemHandle.cpp +++ b/src/armnn/WorkingMemHandle.cpp @@ -13,8 +13,10 @@ namespace armnn namespace experimental { -WorkingMemHandle::WorkingMemHandle(std::vector<WorkingMemDescriptor> workingMemDescriptors, +WorkingMemHandle::WorkingMemHandle(NetworkId networkId, + std::vector<WorkingMemDescriptor> workingMemDescriptors, std::unordered_map<LayerGuid, WorkingMemDescriptor> workingMemDescriptorMap) : + m_NetworkId(networkId), m_WorkingMemDescriptors(workingMemDescriptors), m_WorkingMemDescriptorMap(workingMemDescriptorMap), m_IsAllocated(false), |