From c74b1750fe8cf7affdbc59edd53357e0ea4efa53 Mon Sep 17 00:00:00 2001 From: Colm Donelan Date: Fri, 12 Mar 2021 15:58:48 +0000 Subject: IVGCVSW-5724 Add import tensor handling as ClImportTensorHandleFactory. * Add new ClImportTensorHandlefactory for tensor import. * Add unit tests. Signed-off-by: Colm Donelan Change-Id: I61884fed65e764ebd6985fe0833e43a7296d0641 --- src/backends/cl/ClTensorHandleFactory.hpp | 22 +++++++++++++--------- 1 file changed, 13 insertions(+), 9 deletions(-) (limited to 'src/backends/cl/ClTensorHandleFactory.hpp') 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 #include -#include #include +#include +#include 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 mgr) - : m_MemoryManager(mgr), - m_ImportFlags(static_cast(MemorySource::Undefined)), - m_ExportFlags(static_cast(MemorySource::Undefined)) - {} + : m_MemoryManager(mgr) + , m_ImportFlags(static_cast(MemorySource::Undefined)) + , m_ExportFlags(static_cast(MemorySource::Undefined)) + {} std::unique_ptr 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 -- cgit v1.2.1