diff options
author | nikraj01 <nikhil.raj@arm.com> | 2019-05-30 17:29:32 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-30 19:29:20 +0000 |
commit | 22f0f2bc3d8d8e460200ae353dcea42157d9ad39 (patch) | |
tree | 153a081f76b1a0b64bf2cabbaeb483b54c124b71 /src/backends/reference/workloads/SpaceToBatchNd.hpp | |
parent | 25339c31a829111fce691311cb84100e8591f5da (diff) | |
download | armnn-22f0f2bc3d8d8e460200ae353dcea42157d9ad39.tar.gz |
IVGCVSW-3194 Refactor SpaceToBatchNd workload
Change-Id: Iac2ded9b20c37299e3de51465dcbfb5a7bfc52d5
Signed-off-by: nikraj01 <nikhil.raj@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/SpaceToBatchNd.hpp')
-rw-r--r-- | src/backends/reference/workloads/SpaceToBatchNd.hpp | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/SpaceToBatchNd.hpp b/src/backends/reference/workloads/SpaceToBatchNd.hpp index e74e457a72..57c9b6bc25 100644 --- a/src/backends/reference/workloads/SpaceToBatchNd.hpp +++ b/src/backends/reference/workloads/SpaceToBatchNd.hpp @@ -5,17 +5,20 @@ #pragma once +#include "BaseIterator.hpp" +#include "Decoders.hpp" +#include "Encoders.hpp" + #include <armnn/Descriptors.hpp> #include "armnn/Tensor.hpp" namespace armnn { -template <typename T> void SpaceToBatchNd(const TensorInfo& inputInfo, const TensorInfo& outputInfo, const SpaceToBatchNdDescriptor& params, - const T* inputData, - T* outputData); + Decoder<float>& inputData, + Encoder<float>& outputData); } //namespace armnn |