aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/BatchToSpaceNd.hpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2018-11-28 16:22:22 +0000
committerMatteo Martincigh <matteo.martincigh@arm.com>2018-12-05 11:03:35 +0000
commit2135015779092e259ad5d5df185eda0c34b56359 (patch)
tree744a46d8f6036ac13fc40ab5001ad2ad05fb4525 /src/backends/reference/workloads/BatchToSpaceNd.hpp
parent24df822711e14cd3099a926272d863ed139ed4d7 (diff)
downloadarmnn-2135015779092e259ad5d5df185eda0c34b56359.tar.gz
IVGCVSW-2264 Move DataLayoutIndexed to armnnUtils
* Since DataLayoutIndexed is now required in the TF parser, this changes move it to the armnnUtils library so that it'll be accessible by the armnnTfParser * Modified CMake files and Android.mk files accordingly Change-Id: Ie2620359ef288aeff64cb9e9bec068a466eee0e9
Diffstat (limited to 'src/backends/reference/workloads/BatchToSpaceNd.hpp')
-rw-r--r--src/backends/reference/workloads/BatchToSpaceNd.hpp10
1 files changed, 6 insertions, 4 deletions
diff --git a/src/backends/reference/workloads/BatchToSpaceNd.hpp b/src/backends/reference/workloads/BatchToSpaceNd.hpp
index 5d932cc991..f08df93584 100644
--- a/src/backends/reference/workloads/BatchToSpaceNd.hpp
+++ b/src/backends/reference/workloads/BatchToSpaceNd.hpp
@@ -5,19 +5,21 @@
#pragma once
-#include <backendsCommon/DataLayoutIndexed.hpp>
+#include <armnn/Types.hpp>
+
#include <backendsCommon/Workload.hpp>
#include <backendsCommon/WorkloadData.hpp>
-#include <armnn/Types.hpp>
+
+#include <DataLayoutIndexed.hpp>
namespace armnn
{
-void BatchToSpaceNd(const DataLayoutIndexed& dataLayout,
+void BatchToSpaceNd(const armnnUtils::DataLayoutIndexed& dataLayout,
const TensorInfo& inputTensorInfo,
const TensorInfo& outputTensorInfo,
const std::vector<unsigned int>& blockShape,
const std::vector<std::pair<unsigned int, unsigned int>>& cropsData,
const float* inputData,
float* outputData);
-} // namespace armnn \ No newline at end of file
+} // namespace armnn