commit 831faedd5d2f6306b23202fa4e450c6d241a44a0
author Mike Kelly <mike.kelly@arm.com> Wed Nov 28 11:52:08 2018 +0000
committer Mike Kelly <mike.kelly@arm.com> Wed Nov 28 11:52:08 2018 +0000
tree ad025f07969618a054144aa45333209399656cb5
parent 7e7261ed9a4f8d48c86e57044312e3df82189713

IVGCVSW-1199: BATCH_TO_SPACE_ND integrate Arm Compute CL

Change-Id: Ic772acf481caed6e8a6b99f68a63dfc2a34c24dc

src/backends/cl/ClLayerSupport.hpp

diff --git a/src/backends/cl/ClLayerSupport.hpp b/src/backends/cl/ClLayerSupport.hpp
index b4d2789..cd1b722 100644
--- a/src/backends/cl/ClLayerSupport.hpp
+++ b/src/backends/cl/ClLayerSupport.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;