diff options
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 |