diff options
Diffstat (limited to 'src/backends/cl/ClTensorHandle.hpp')
| -rw-r--r-- | src/backends/cl/ClTensorHandle.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/cl/ClTensorHandle.hpp b/src/backends/cl/ClTensorHandle.hpp index 59a6bee7f5..c0773a4ea4 100644 --- a/src/backends/cl/ClTensorHandle.hpp +++ b/src/backends/cl/ClTensorHandle.hpp @@ -5,6 +5,7 @@ #pragma once #include <backendsCommon/OutputHandler.hpp> +#include <aclCommon/ArmComputeTensorHandle.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> #include <Half.hpp> @@ -22,7 +23,7 @@ namespace armnn { -class IClTensorHandle : public ITensorHandle +class IClTensorHandle : public IAclTensorHandle { public: virtual arm_compute::ICLTensor& GetTensor() = 0; |