diff options
author | Finn Williams <finwil01@e127804.cambridge.arm.com> | 2021-10-02 15:06:39 +0100 |
---|---|---|
committer | Finn Williams <finwil01@e127804.cambridge.arm.com> | 2021-10-08 15:42:14 +0100 |
commit | 8636bc705cc33fd869f64ebf24b14836d5a40b29 (patch) | |
tree | e546319af87596d053c75d65eedd1efb1f946228 /include/armnn/IWorkingMemHandle.hpp | |
parent | 521032fd424cf86681eb125afbf5eaee47d8c585 (diff) | |
download | armnn-8636bc705cc33fd869f64ebf24b14836d5a40b29.tar.gz |
IVGCVSW-6313 Support pre-importing outputs
* Add ClearImportedInputs/Outputs function to IRuntime
* Add UnImport function to ITensorHandle
* Remove mutex from IWorkingMemHandle
Change-Id: I34c9b6e1618755e10f3b4597afa1d9a9ea97e5fe
Signed-off-by: Finn Williams <finwil01@e127804.cambridge.arm.com>
Diffstat (limited to 'include/armnn/IWorkingMemHandle.hpp')
-rw-r--r-- | include/armnn/IWorkingMemHandle.hpp | 15 |
1 files changed, 5 insertions, 10 deletions
diff --git a/include/armnn/IWorkingMemHandle.hpp b/include/armnn/IWorkingMemHandle.hpp index 171fa3d81c..bbc4913c59 100644 --- a/include/armnn/IWorkingMemHandle.hpp +++ b/include/armnn/IWorkingMemHandle.hpp @@ -5,8 +5,6 @@ #pragma once -#include <mutex> - namespace armnn { @@ -26,23 +24,20 @@ public: virtual NetworkId GetNetworkId() = 0; /// 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. + /// deferred to execution time. virtual void Allocate() = 0; - /// Free the backing memory required for execution. The mutex must be locked. + /// Free the backing memory required for execution. virtual void Free() = 0; - /// IsAllocated returns true if the backing memory is currently allocated. The mutex must be locked. + /// IsAllocated returns true if the backing memory is currently allocated. virtual bool IsAllocated() = 0; - /// Get a mutex which can be used for synchronizing access to the WorkingMemHandle object. - virtual std::mutex& GetMutex() = 0; - - /// Get the WorkingMemDescriptor for a Layer. The mutex must be locked. + /// Get the WorkingMemDescriptor for a Layer. virtual WorkingMemDescriptor& GetWorkingMemDescriptor(LayerGuid id) = 0; /// Get the WorkingMemDescriptor at an index. The WorkingMemDescriptors are stored in the same order as - /// the Workloads in a topologically sorted graph. The mutex must be locked. + /// the Workloads in a topologically sorted graph. virtual WorkingMemDescriptor& GetWorkingMemDescriptorAt(unsigned int id) = 0; }; |