diff options
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 |