diff options
| author | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-06-21 15:50:00 +0100 |
|---|---|---|
| committer | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-06-24 11:14:48 +0100 |
| commit | 6858d3fb714cc20d5fcfd814c35ed3a84dc82145 (patch) | |
| tree | db6a6651fff0dce4095be58bc155e63175abe33e /src/backends/reference/workloads/Gather.hpp | |
| parent | 389aa70c8a24fa2faf33df5f8cd9a99b0fabe971 (diff) | |
| download | armnn-6858d3fb714cc20d5fcfd814c35ed3a84dc82145.tar.gz | |
IVGCVSW-3247: Refactor reference Gather workload
* Refactored Gather reference workload to not use templates for different types
* Added quantization values in Gather unit tests for Quantized types
Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: Ibe5d655aa1c287824e45b83818c5862bda7f92b0
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 |