aboutsummaryrefslogtreecommitdiff
path: root/include/armnn/ILayerSupport.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2018-11-02 10:57:25 +0000
committernattapat.chaimanowong <nattapat.chaimanowong@arm.com>2018-11-02 12:35:59 +0000
commit207ef9a6b8b3ea0afe9a095639f67b5dedd095d7 (patch)
treefb6251cfbc28cc2bfdeb5f8f85cc994672e706eb /include/armnn/ILayerSupport.hpp
parent6b9658239d377372523fe49c71fde31701d986e3 (diff)
downloadarmnn-207ef9a6b8b3ea0afe9a095639f67b5dedd095d7.tar.gz
IVGCVSW-2093 Add SpaceToBatchNd layer and corresponding no-op factory implementations
Change-Id: Ibd457f3a2d4342c4d6335bd3c471282a14ab6b14
Diffstat (limited to 'include/armnn/ILayerSupport.hpp')
-rw-r--r--include/armnn/ILayerSupport.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/include/armnn/ILayerSupport.hpp b/include/armnn/ILayerSupport.hpp
index d63c3a7063..0e7adff0af 100644
--- a/include/armnn/ILayerSupport.hpp
+++ b/include/armnn/ILayerSupport.hpp
@@ -161,6 +161,11 @@ public:
virtual bool IsReshapeSupported(const TensorInfo& input,
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
+ virtual bool IsSpaceToBatchNdSupported(const TensorInfo& input,
+ const TensorInfo& output,
+ const SpaceToBatchNdDescriptor& descriptor,
+ Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;
+
virtual bool IsResizeBilinearSupported(const TensorInfo& input,
Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const;