diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-01-17 21:19:52 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-01-18 21:32:27 +0000 |
| commit | 788e2a6c917abe7c5187d9e5c349683d456080e5 (patch) | |
| tree | 1c88b36828a3ebb6ef04c3e8c0bc92bdd330e9cf /include/armnn/backends/Workload.hpp | |
| parent | adeebaa73205bd981ea7e8c8f135f01355cba841 (diff) | |
| download | armnn-788e2a6c917abe7c5187d9e5c349683d456080e5.tar.gz | |
IVGCVSW-6682 Add ReplaceTensorHandle functions to IWorkload and BaseWorkload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I9f80b9f45206db920568e28e363fcb60f5c0819a
Diffstat (limited to 'include/armnn/backends/Workload.hpp')
| -rw-r--r-- | include/armnn/backends/Workload.hpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/include/armnn/backends/Workload.hpp b/include/armnn/backends/Workload.hpp index 7c1bda50bc..07e1abb392 100644 --- a/include/armnn/backends/Workload.hpp +++ b/include/armnn/backends/Workload.hpp @@ -54,6 +54,18 @@ public: profiling::ProfilingGuid GetGuid() const final { return m_Guid; } + // Replace input tensor handle with the given TensorHandle + void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override + { + m_Data.m_Inputs[slot] = tensorHandle; + } + + // Replace output tensor handle with the given TensorHandle + void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override + { + m_Data.m_Outputs[slot] = tensorHandle; + } + protected: QueueDescriptor m_Data; const profiling::ProfilingGuid m_Guid; |