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