diff options
Diffstat (limited to 'src/backends/cl/workloads/ClRankWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClRankWorkload.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClRankWorkload.hpp b/src/backends/cl/workloads/ClRankWorkload.hpp index b5a856ff07..8a7e2c2078 100644 --- a/src/backends/cl/workloads/ClRankWorkload.hpp +++ b/src/backends/cl/workloads/ClRankWorkload.hpp @@ -5,7 +5,7 @@ #pragma once -#include <armnn/backends/Workload.hpp> +#include "ClBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> #include "ClWorkloadUtils.hpp" @@ -13,10 +13,10 @@ namespace armnn { -struct ClRankWorkload : public BaseWorkload<RankQueueDescriptor> +struct ClRankWorkload : public ClBaseWorkload<RankQueueDescriptor> { public: - using BaseWorkload<RankQueueDescriptor>::BaseWorkload; + using ClBaseWorkload<RankQueueDescriptor>::ClBaseWorkload; virtual void Execute() const override { const ClTensorHandle* clTensorHandle = PolymorphicDowncast<const ClTensorHandle*>(m_Data.m_Inputs[0]); |