aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/Network.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 /src/armnn/Network.hpp
parent6b9658239d377372523fe49c71fde31701d986e3 (diff)
downloadarmnn-207ef9a6b8b3ea0afe9a095639f67b5dedd095d7.tar.gz
IVGCVSW-2093 Add SpaceToBatchNd layer and corresponding no-op factory implementations
Change-Id: Ibd457f3a2d4342c4d6335bd3c471282a14ab6b14
Diffstat (limited to 'src/armnn/Network.hpp')
-rw-r--r--src/armnn/Network.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/Network.hpp b/src/armnn/Network.hpp
index 129513d657..4a93dd1ee4 100644
--- a/src/armnn/Network.hpp
+++ b/src/armnn/Network.hpp
@@ -106,6 +106,9 @@ public:
IConnectableLayer* AddReshapeLayer(const ReshapeDescriptor& reshapeDescriptor,
const char* name = nullptr) override;
+ IConnectableLayer* AddSpaceToBatchNdLayer(const SpaceToBatchNdDescriptor& spaceToBatchNdDescriptor,
+ const char* name = nullptr) override;
+
IConnectableLayer* AddFloorLayer(const char* name = nullptr) override;
IConnectableLayer* AddOutputLayer(LayerBindingId id, const char* name = nullptr) override;