From 4e1e136cce3fca73ba49b570cfcb620f4ec574da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89anna=20=C3=93=20Cath=C3=A1in?=
 <eanna.ocathain@arm.com>
Date: Mon, 12 Nov 2018 11:36:34 +0000
Subject: IVGCVSW-2054: BATCH_TO_SPACE_ND Reference implementation and Unit
 tests.

Change-Id: I13c6728dbb60643d0e086d171225c5d802987f92
---
 include/armnn/Descriptors.hpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'include/armnn/Descriptors.hpp')

diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp
index 656afb1756..bda8cf7396 100644
--- a/include/armnn/Descriptors.hpp
+++ b/include/armnn/Descriptors.hpp
@@ -296,6 +296,25 @@ struct BatchNormalizationDescriptor
     DataLayoutIndexed m_DataLayout;
 };
 
+struct BatchToSpaceNdDescriptor
+{
+    BatchToSpaceNdDescriptor()
+        : m_BlockShape()
+        , m_Crops()
+        , m_DataLayout(DataLayout::NCHW)
+    {}
+
+    BatchToSpaceNdDescriptor(std::vector<unsigned int> blockShape, std::vector<std::vector<unsigned int>> crops)
+        : m_BlockShape(blockShape)
+        , m_Crops(crops)
+        , m_DataLayout(DataLayout::NCHW)
+    {}
+
+    std::vector<unsigned int> m_BlockShape;
+    std::vector<std::vector<unsigned int>> m_Crops;
+    DataLayoutIndexed m_DataLayout;
+};
+
 struct FakeQuantizationDescriptor
 {
     FakeQuantizationDescriptor()
-- 
cgit v1.2.1