From 3ea76d5f0d99794cf5f0b60ef3738d0905f10b2a Mon Sep 17 00:00:00 2001
From: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Date: Fri, 9 Nov 2018 14:10:38 +0000
Subject: IVGCVSW-2095 Add reference implementation and unit tests for
 SpaceToBatchNd

Change-Id: I27ffebdece6e68460931a44c15b9b029f9fce638
---
 src/backends/reference/RefLayerSupport.hpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/backends/reference/RefLayerSupport.hpp')

diff --git a/src/backends/reference/RefLayerSupport.hpp b/src/backends/reference/RefLayerSupport.hpp
index 0da59986a9..b161f5c7cf 100644
--- a/src/backends/reference/RefLayerSupport.hpp
+++ b/src/backends/reference/RefLayerSupport.hpp
@@ -159,6 +159,11 @@ public:
                             const SoftmaxDescriptor& descriptor,
                             Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsSpaceToBatchNdSupported(const TensorInfo& input,
+                                   const TensorInfo& output,
+                                   const SpaceToBatchNdDescriptor& descriptor,
+                                   Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
     bool IsSplitterSupported(const TensorInfo& input,
                              const ViewsDescriptor& descriptor,
                              Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
-- 
cgit v1.2.1