diff options
| author | Derek Lamberti <derek.lamberti@arm.com> | 2019-06-13 17:34:19 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-06-14 16:41:17 +0000 |
| commit | c81855f6ca52eb025a303b95eee7a12a5e2f9557 (patch) | |
| tree | 86ac342839f49348b54f29c4588b6a9e4059cd84 /src/backends/neon/NeonTensorHandle.hpp | |
| parent | db48288f54ad441f34108cf270ed3089d3a465ea (diff) | |
| download | armnn-c81855f6ca52eb025a303b95eee7a12a5e2f9557.tar.gz | |
IVGCVSW-3278 Cl and Neon TensorHandles inherit from common base interface
Change-Id: Ia68da09d8f0fb0a04af9cb61062d7edaa5f1b887
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/backends/neon/NeonTensorHandle.hpp')
| -rw-r--r-- | src/backends/neon/NeonTensorHandle.hpp | 16 |
1 files changed, 4 insertions, 12 deletions
diff --git a/src/backends/neon/NeonTensorHandle.hpp b/src/backends/neon/NeonTensorHandle.hpp index b972043827..3bbba78525 100644 --- a/src/backends/neon/NeonTensorHandle.hpp +++ b/src/backends/neon/NeonTensorHandle.hpp @@ -5,6 +5,7 @@ #pragma once #include <backendsCommon/OutputHandler.hpp> +#include <aclCommon/ArmComputeTensorHandle.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> #include <arm_compute/runtime/MemoryGroup.h> @@ -19,16 +20,7 @@ namespace armnn { -class INeonTensorHandle : public ITensorHandle -{ -public: - virtual arm_compute::ITensor& GetTensor() = 0; - virtual arm_compute::ITensor const& GetTensor() const = 0; - virtual arm_compute::DataType GetDataType() const = 0; - virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) = 0; -}; - -class NeonTensorHandle : public INeonTensorHandle +class NeonTensorHandle : public IAclTensorHandle { public: NeonTensorHandle(const TensorInfo& tensorInfo) @@ -131,10 +123,10 @@ private: std::shared_ptr<arm_compute::MemoryGroup> m_MemoryGroup; }; -class NeonSubTensorHandle : public INeonTensorHandle +class NeonSubTensorHandle : public IAclTensorHandle { public: - NeonSubTensorHandle(INeonTensorHandle* parent, + NeonSubTensorHandle(IAclTensorHandle* parent, const arm_compute::TensorShape& shape, const arm_compute::Coordinates& coords) : m_Tensor(&parent->GetTensor(), shape, coords) |