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/reference/RefLayerSupport.hpp

diff --git a/src/backends/reference/RefLayerSupport.hpp b/src/backends/reference/RefLayerSupport.hpp
index b161f5c..2e86ece 100644
--- a/src/backends/reference/RefLayerSupport.hpp
+++ b/src/backends/reference/RefLayerSupport.hpp
@@ -31,6 +31,11 @@
                                        const BatchNormalizationDescriptor& descriptor,
                                        Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
 
+    bool IsBatchToSpaceNdSupported(const TensorInfo& input,
+                                   const TensorInfo& output,
+                                   const BatchToSpaceNdDescriptor& descriptor,
+                                   Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
+
     bool IsConstantSupported(const TensorInfo& output,
                              Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;