aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefWorkloadUtils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/RefWorkloadUtils.hpp')
-rw-r--r--src/backends/reference/workloads/RefWorkloadUtils.hpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefWorkloadUtils.hpp b/src/backends/reference/workloads/RefWorkloadUtils.hpp
index f1b31571db..a36ed45970 100644
--- a/src/backends/reference/workloads/RefWorkloadUtils.hpp
+++ b/src/backends/reference/workloads/RefWorkloadUtils.hpp
@@ -9,12 +9,12 @@
#include <armnn/Tensor.hpp>
#include <armnn/Types.hpp>
+#include <armnn/utility/PolymorphicDowncast.hpp>
#include <reference/RefTensorHandle.hpp>
#include <BFloat16.hpp>
#include <Half.hpp>
-#include <boost/polymorphic_cast.hpp>
namespace armnn
{
@@ -27,7 +27,7 @@ inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle)
{
// We know that reference workloads use RefTensorHandles for inputs and outputs
const RefTensorHandle* refTensorHandle =
- boost::polymorphic_downcast<const RefTensorHandle*>(tensorHandle);
+ PolymorphicDowncast<const RefTensorHandle*>(tensorHandle);
return refTensorHandle->GetTensorInfo();
}