diff options
Diffstat (limited to 'src/backends/reference/workloads/RefGatherWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefGatherWorkload.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/RefGatherWorkload.hpp b/src/backends/reference/workloads/RefGatherWorkload.hpp index a2698e3a25..ec880a5109 100644 --- a/src/backends/reference/workloads/RefGatherWorkload.hpp +++ b/src/backends/reference/workloads/RefGatherWorkload.hpp @@ -5,7 +5,7 @@ #pragma once -#include <armnn/backends/Workload.hpp> +#include "RefBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> #include <armnn/TypesUtils.hpp> @@ -16,10 +16,10 @@ namespace armnn { -class RefGatherWorkload : public BaseWorkload<GatherQueueDescriptor> +class RefGatherWorkload : public RefBaseWorkload<GatherQueueDescriptor> { public: - using BaseWorkload<GatherQueueDescriptor>::BaseWorkload; + using RefBaseWorkload<GatherQueueDescriptor>::RefBaseWorkload; void Execute() const override; void ExecuteAsync(WorkingMemDescriptor& workingMemDescriptor) override; private: |