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 1830d186b6..0481307972 100644 --- a/src/backends/cl/ClTensorHandle.hpp +++ b/src/backends/cl/ClTensorHandle.hpp @@ -9,6 +9,8 @@ #include <Half.hpp> +#include <armnn/utility/PolymorphicDowncast.hpp> + #include <arm_compute/runtime/CL/CLTensor.h> #include <arm_compute/runtime/CL/CLSubTensor.h> #include <arm_compute/runtime/IMemoryGroup.h> @@ -16,8 +18,6 @@ #include <arm_compute/core/TensorShape.h> #include <arm_compute/core/Coordinates.h> -#include <boost/polymorphic_pointer_cast.hpp> - namespace armnn { @@ -71,7 +71,7 @@ public: virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) override { - m_MemoryGroup = boost::polymorphic_pointer_downcast<arm_compute::MemoryGroup>(memoryGroup); + m_MemoryGroup = PolymorphicPointerDowncast<arm_compute::MemoryGroup>(memoryGroup); } TensorShape GetStrides() const override |