aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/workloads/NeonGatherWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/workloads/NeonGatherWorkload.hpp')
-rw-r--r--src/backends/neon/workloads/NeonGatherWorkload.hpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonGatherWorkload.hpp b/src/backends/neon/workloads/NeonGatherWorkload.hpp
index b1b47a5069..e5b7b57629 100644
--- a/src/backends/neon/workloads/NeonGatherWorkload.hpp
+++ b/src/backends/neon/workloads/NeonGatherWorkload.hpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2020 Arm Ltd. All rights reserved.
+// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
@@ -13,7 +13,8 @@ namespace armnn
{
arm_compute::Status NeonGatherWorkloadValidate(const TensorInfo& input,
const TensorInfo& indices,
- const TensorInfo& output);
+ const TensorInfo& output,
+ const GatherDescriptor& descriptor);
class NeonGatherWorkload : public BaseWorkload<GatherQueueDescriptor>
{