aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonConstantWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/workloads/NeonConstantWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonConstantWorkload.hpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonConstantWorkload.hpp b/src/backends/neon/workloads/NeonConstantWorkload.hpp
index f800a45256..3ca0bce5d2 100644
--- a/src/backends/neon/workloads/NeonConstantWorkload.hpp
+++ b/src/backends/neon/workloads/NeonConstantWorkload.hpp
@@ -1,17 +1,19 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
+#include "NeonBaseWorkload.hpp"
+
#include <neon/workloads/NeonWorkloadUtils.hpp>
namespace armnn
{
arm_compute::Status NeonConstantWorkloadValidate(const TensorInfo& output);
-class NeonConstantWorkload : public BaseWorkload<ConstantQueueDescriptor>
+class NeonConstantWorkload : public NeonBaseWorkload<ConstantQueueDescriptor>
{
public:
NeonConstantWorkload(const ConstantQueueDescriptor& descriptor, const WorkloadInfo& info);