diff options
Diffstat (limited to 'src/backends/neon/workloads/NeonRankWorkload.hpp')
-rw-r--r-- | src/backends/neon/workloads/NeonRankWorkload.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/neon/workloads/NeonRankWorkload.hpp b/src/backends/neon/workloads/NeonRankWorkload.hpp index b8b83af886..22bf7af5e5 100644 --- a/src/backends/neon/workloads/NeonRankWorkload.hpp +++ b/src/backends/neon/workloads/NeonRankWorkload.hpp @@ -5,7 +5,7 @@ #pragma once -#include <armnn/backends/Workload.hpp> +#include "NeonBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> #include "NeonWorkloadUtils.hpp" @@ -13,10 +13,10 @@ namespace armnn { -struct NeonRankWorkload : public BaseWorkload<RankQueueDescriptor> +struct NeonRankWorkload : public NeonBaseWorkload<RankQueueDescriptor> { public: - using BaseWorkload<RankQueueDescriptor>::BaseWorkload; + using NeonBaseWorkload<RankQueueDescriptor>::NeonBaseWorkload; virtual void Execute() const override { const NeonTensorHandle* neonTensorHandle = PolymorphicDowncast<const NeonTensorHandle*>(m_Data.m_Inputs[0]); |