diff options
| author | Colm Donelan <Colm.Donelan@arm.com> | 2021-03-12 15:58:48 +0000 |
|---|---|---|
| committer | Colm Donelan <colm.donelan@arm.com> | 2021-03-23 13:59:15 +0000 |
| commit | c74b1750fe8cf7affdbc59edd53357e0ea4efa53 (patch) | |
| tree | fc925b928147f70016605a0c123066cfdf9c15a1 /src/backends/cl/ClTensorHandleFactory.hpp | |
| parent | 4441d94fa0a97d4137e49315d69d32fdc0bbcd03 (diff) | |
| download | armnn-c74b1750fe8cf7affdbc59edd53357e0ea4efa53.tar.gz | |
IVGCVSW-5724 Add import tensor handling as ClImportTensorHandleFactory.
* Add new ClImportTensorHandlefactory for tensor import.
* Add unit tests.
Signed-off-by: Colm Donelan <Colm.Donelan@arm.com>
Change-Id: I61884fed65e764ebd6985fe0833e43a7296d0641
Diffstat (limited to 'src/backends/cl/ClTensorHandleFactory.hpp')
| -rw-r--r-- | src/backends/cl/ClTensorHandleFactory.hpp | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/src/backends/cl/ClTensorHandleFactory.hpp b/src/backends/cl/ClTensorHandleFactory.hpp index 13c97c0b3c..3acab0bce7 100644 --- a/src/backends/cl/ClTensorHandleFactory.hpp +++ b/src/backends/cl/ClTensorHandleFactory.hpp @@ -4,25 +4,29 @@ // #pragma once -#include <armnn/backends/ITensorHandleFactory.hpp> #include <aclCommon/BaseMemoryManager.hpp> -#include <armnn/backends/IMemoryManager.hpp> #include <armnn/MemorySources.hpp> +#include <armnn/backends/IMemoryManager.hpp> +#include <armnn/backends/ITensorHandleFactory.hpp> namespace armnn { -constexpr const char* ClTensorHandleFactoryId() { return "Arm/Cl/TensorHandleFactory"; } +constexpr const char* ClTensorHandleFactoryId() +{ + return "Arm/Cl/TensorHandleFactory"; +} -class ClTensorHandleFactory : public ITensorHandleFactory { +class ClTensorHandleFactory : public ITensorHandleFactory +{ public: static const FactoryId m_Id; ClTensorHandleFactory(std::shared_ptr<ClMemoryManager> mgr) - : m_MemoryManager(mgr), - m_ImportFlags(static_cast<MemorySourceFlags>(MemorySource::Undefined)), - m_ExportFlags(static_cast<MemorySourceFlags>(MemorySource::Undefined)) - {} + : m_MemoryManager(mgr) + , m_ImportFlags(static_cast<MemorySourceFlags>(MemorySource::Undefined)) + , m_ExportFlags(static_cast<MemorySourceFlags>(MemorySource::Undefined)) + {} std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& parent, const TensorShape& subTensorShape, @@ -56,4 +60,4 @@ private: MemorySourceFlags m_ExportFlags; }; -} // namespace armnn
\ No newline at end of file +} // namespace armnn
\ No newline at end of file |