aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/IWorkingMemHandle.hpp
diff options
context:
space:
mode:
authorFinn Williams <finwil01@e127804.cambridge.arm.com>2021-10-02 15:06:39 +0100
committerFinn Williams <finwil01@e127804.cambridge.arm.com>2021-10-08 15:42:14 +0100
commit8636bc705cc33fd869f64ebf24b14836d5a40b29 (patch)
treee546319af87596d053c75d65eedd1efb1f946228 /include/armnn/IWorkingMemHandle.hpp
parent521032fd424cf86681eb125afbf5eaee47d8c585 (diff)
downloadarmnn-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.hpp15
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;
};