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