diff options
Diffstat (limited to 'src/backends/cl/ClTensorHandle.hpp')
| -rw-r--r-- | src/backends/cl/ClTensorHandle.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/cl/ClTensorHandle.hpp b/src/backends/cl/ClTensorHandle.hpp index d08b79f9a6..f5583c0b97 100644 --- a/src/backends/cl/ClTensorHandle.hpp +++ b/src/backends/cl/ClTensorHandle.hpp @@ -12,8 +12,8 @@ #include <arm_compute/runtime/CL/CLTensor.h> #include <arm_compute/runtime/CL/CLSubTensor.h> -#include <arm_compute/runtime/CL/CLMemoryGroup.h> #include <arm_compute/runtime/IMemoryGroup.h> +#include <arm_compute/runtime/MemoryGroup.h> #include <arm_compute/core/TensorShape.h> #include <arm_compute/core/Coordinates.h> @@ -72,7 +72,7 @@ public: virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) override { - m_MemoryGroup = boost::polymorphic_pointer_downcast<arm_compute::CLMemoryGroup>(memoryGroup); + m_MemoryGroup = boost::polymorphic_pointer_downcast<arm_compute::MemoryGroup>(memoryGroup); } TensorShape GetStrides() const override @@ -151,7 +151,7 @@ private: } arm_compute::CLTensor m_Tensor; - std::shared_ptr<arm_compute::CLMemoryGroup> m_MemoryGroup; + std::shared_ptr<arm_compute::MemoryGroup> m_MemoryGroup; }; class ClSubTensorHandle : public IClTensorHandle |