diff options
Diffstat (limited to 'src/backends/reference/workloads/Gather.hpp')
| -rw-r--r-- | src/backends/reference/workloads/Gather.hpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/Gather.hpp b/src/backends/reference/workloads/Gather.hpp index 0ad4f8ceb6..16c983eec4 100644 --- a/src/backends/reference/workloads/Gather.hpp +++ b/src/backends/reference/workloads/Gather.hpp @@ -7,15 +7,18 @@ #include "armnn/Tensor.hpp" +#include "BaseIterator.hpp" +#include "Decoders.hpp" +#include "Encoders.hpp" + namespace armnn { -template <typename T> void Gather(const TensorInfo& paramsInfo, const TensorInfo& indicesInfo, const TensorInfo& outputInfo, - const T* params, + Decoder<float>& params, const int32_t* indices, - T* output); + Encoder<float>& output); } //namespace armnn |