diff options
Diffstat (limited to 'src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp')
-rw-r--r-- | src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp b/src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp index ae93d03656..5d4b483fa7 100644 --- a/src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp +++ b/src/backends/reference/workloads/RefDepthwiseConvolution2dWorkload.hpp @@ -2,7 +2,7 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // -#include <armnn/backends/Workload.hpp> +#include "RefBaseWorkload.hpp" #include <armnn/backends/WorkloadData.hpp> #include "Decoders.hpp" #include "Encoders.hpp" @@ -12,7 +12,7 @@ namespace armnn { -class RefDepthwiseConvolution2dWorkload : public BaseWorkload<DepthwiseConvolution2dQueueDescriptor> { +class RefDepthwiseConvolution2dWorkload : public RefBaseWorkload<DepthwiseConvolution2dQueueDescriptor> { public: explicit RefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info); |