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