diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-06-20 12:07:19 +0100 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-06-20 12:07:30 +0100 |
commit | 47ea3c0e8d8d10906d04a0e7c537ffee68b0f819 (patch) | |
tree | ab160c7126820dd2b5766974256167fdc66d183a /src/backends/reference/workloads/BatchToSpaceNd.hpp | |
parent | 51982472bfedf12e7d82cde6614617f94b2c86d0 (diff) | |
download | armnn-47ea3c0e8d8d10906d04a0e7c537ffee68b0f819.tar.gz |
IVGCVSW-3248 Refactor reference BatchToSpace workload
* Add Decoders and Encoders to workload to make it data type agnostic
* Merge float32 and Uint8 into single workload
Change-Id: I8adfa1898a63f13889eaaf55a31c26fd1e2d7ee8
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/backends/reference/workloads/BatchToSpaceNd.hpp')
-rw-r--r-- | src/backends/reference/workloads/BatchToSpaceNd.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/reference/workloads/BatchToSpaceNd.hpp b/src/backends/reference/workloads/BatchToSpaceNd.hpp index f08df93584..b757d3709c 100644 --- a/src/backends/reference/workloads/BatchToSpaceNd.hpp +++ b/src/backends/reference/workloads/BatchToSpaceNd.hpp @@ -9,6 +9,9 @@ #include <backendsCommon/Workload.hpp> #include <backendsCommon/WorkloadData.hpp> +#include "BaseIterator.hpp" +#include "Decoders.hpp" +#include "Encoders.hpp" #include <DataLayoutIndexed.hpp> @@ -20,6 +23,6 @@ void BatchToSpaceNd(const armnnUtils::DataLayoutIndexed& dataLayout, const TensorInfo& outputTensorInfo, const std::vector<unsigned int>& blockShape, const std::vector<std::pair<unsigned int, unsigned int>>& cropsData, - const float* inputData, - float* outputData); + Decoder<float>& inputDecoder, + Encoder<float>& outputEncoder); } // namespace armnn |