aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Gather.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/workloads/Gather.cpp')
-rw-r--r--src/backends/reference/workloads/Gather.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/Gather.cpp b/src/backends/reference/workloads/Gather.cpp
index c23edcd3bd..3e2190c81b 100644
--- a/src/backends/reference/workloads/Gather.cpp
+++ b/src/backends/reference/workloads/Gather.cpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
@@ -20,9 +20,12 @@ void Gather(const TensorInfo& paramsInfo,
const TensorInfo& outputInfo,
Decoder<float>& params,
const int32_t* indices,
- Encoder<float>& output)
+ Encoder<float>& output,
+ const int32_t axis)
{
IgnoreUnused(outputInfo);
+ IgnoreUnused(axis);
+
const TensorShape& paramsShape = paramsInfo.GetShape();
unsigned int paramsProduct = 1;