diff options
Diffstat (limited to 'src/backends/reference/workloads/RefConstantWorkload.hpp')
| -rw-r--r-- | src/backends/reference/workloads/RefConstantWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefConstantWorkload.hpp b/src/backends/reference/workloads/RefConstantWorkload.hpp index c158983d7a..181d79d320 100644 --- a/src/backends/reference/workloads/RefConstantWorkload.hpp +++ b/src/backends/reference/workloads/RefConstantWorkload.hpp @@ -5,7 +5,7 @@ #pragma once -#include <armnn/backends/Workload.hpp> +#include "RefBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> #include <armnn/Types.hpp> @@ -14,7 +14,7 @@ namespace armnn { // Base class template providing an implementation of the Constant layer common to all data types. -class RefConstantWorkload : public BaseWorkload<ConstantQueueDescriptor> +class RefConstantWorkload : public RefBaseWorkload<ConstantQueueDescriptor> { public: RefConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info); |