commit 4e1e136cce3fca73ba49b570cfcb620f4ec574da
author Éanna Ó Catháin <eanna.ocathain@arm.com> Mon Nov 12 11:36:34 2018 +0000
committer Les Bell <les.bell@arm.com> Mon Nov 12 13:08:37 2018 +0000
tree 1fe9fcbb6a9dbafc12aa99ac543bc0da636a1cd1
parent f97debb95cbc7e0bbc60e66e5463ede517cac61b

IVGCVSW-2054: BATCH_TO_SPACE_ND Reference implementation and Unit tests.

Change-Id: I13c6728dbb60643d0e086d171225c5d802987f92

src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp

diff --git a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp
index 2c992bc..2507905 100644
--- a/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp
+++ b/src/backends/backendsCommon/test/IsLayerSupportedTestImpl.hpp
@@ -92,6 +92,20 @@
 };
 
 template<>
+struct DummyLayer<armnn::BatchToSpaceNdLayer>
+{
+    DummyLayer()
+    {
+        m_Layer = dummyGraph.AddLayer<armnn::BatchToSpaceNdLayer>(armnn::BatchToSpaceNdDescriptor(), "");
+    }
+    ~DummyLayer()
+    {
+        dummyGraph.EraseLayer(m_Layer);
+    }
+    armnn::BatchToSpaceNdLayer* m_Layer;
+};
+
+template<>
 struct DummyLayer<armnn::ConstantLayer, void>
 {
     DummyLayer()
@@ -306,6 +320,8 @@
 
 DECLARE_LAYER_POLICY_2_PARAM(BatchNormalization)
 
+DECLARE_LAYER_POLICY_2_PARAM(BatchToSpaceNd)
+
 DECLARE_LAYER_POLICY_1_PARAM(Constant)
 
 DECLARE_LAYER_POLICY_1_PARAM(ConvertFp16ToFp32)