From 207ef9a6b8b3ea0afe9a095639f67b5dedd095d7 Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Fri, 2 Nov 2018 10:57:25 +0000
Subject: IVGCVSW-2093 Add SpaceToBatchNd layer and corresponding no-op factory
 implementations

Change-Id: Ibd457f3a2d4342c4d6335bd3c471282a14ab6b14
---
 src/armnn/layers/SpaceToBatchNdLayer.hpp | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 src/armnn/layers/SpaceToBatchNdLayer.hpp

(limited to 'src/armnn/layers/SpaceToBatchNdLayer.hpp')

diff --git a/src/armnn/layers/SpaceToBatchNdLayer.hpp b/src/armnn/layers/SpaceToBatchNdLayer.hpp
new file mode 100644
index 0000000000..090fb35989
--- /dev/null
+++ b/src/armnn/layers/SpaceToBatchNdLayer.hpp
@@ -0,0 +1,30 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include "LayerWithParameters.hpp"
+
+namespace armnn
+{
+
+class SpaceToBatchNdLayer : public LayerWithParameters<SpaceToBatchNdDescriptor>
+{
+public:
+    virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph&            graph,
+                                                      const IWorkloadFactory& factory) const override;
+
+    SpaceToBatchNdLayer* Clone(Graph& graph) const override;
+
+    std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
+
+    void ValidateTensorShapesFromInputs() override;
+
+protected:
+    SpaceToBatchNdLayer(const SpaceToBatchNdDescriptor param, const char* name);
+    ~SpaceToBatchNdLayer() = default;
+};
+
+} // namespace
-- 
cgit v1.2.1