aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LayersFwd.hpp
diff options
context:
space:
mode:
authorMohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com>2018-09-24 13:30:09 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-10 16:16:57 +0100
commit5662c206864df4121eea29c541c24c0f62113809 (patch)
treef80aa8a094dc3ede6d2c7e4a905ac2f278199268 /src/armnn/LayersFwd.hpp
parent351d13d0b5fa698b72130012b2f069d30b911cb3 (diff)
downloadarmnn-5662c206864df4121eea29c541c24c0f62113809.tar.gz
IVGCVSW-1881 Add PadLayer to ArmNN
Change-Id: Ib70883a20fa7b82bfb090aa98cc51a6d645b5722
Diffstat (limited to 'src/armnn/LayersFwd.hpp')
-rw-r--r--src/armnn/LayersFwd.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/LayersFwd.hpp b/src/armnn/LayersFwd.hpp
index c9ee9dbfec..8321be447d 100644
--- a/src/armnn/LayersFwd.hpp
+++ b/src/armnn/LayersFwd.hpp
@@ -27,6 +27,7 @@
#include "layers/MultiplicationLayer.hpp"
#include "layers/NormalizationLayer.hpp"
#include "layers/OutputLayer.hpp"
+#include "layers/PadLayer.hpp"
#include "layers/PermuteLayer.hpp"
#include "layers/Pooling2dLayer.hpp"
#include "layers/ReshapeLayer.hpp"
@@ -83,6 +84,7 @@ DECLARE_LAYER(Merger)
DECLARE_LAYER(Multiplication)
DECLARE_LAYER(Normalization)
DECLARE_LAYER(Output)
+DECLARE_LAYER(Pad)
DECLARE_LAYER(Permute)
DECLARE_LAYER(Pooling2d)
DECLARE_LAYER(Reshape)