aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LayersFwd.hpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2020-10-08 11:42:30 +0100
committerJim Flynn <jim.flynn@arm.com>2020-10-08 14:24:58 +0000
commit3a40ea5e1efd61ebb21a174ef7bf9adecac8ade9 (patch)
tree2bc713946514f7c3d551641a66bee30dd2a4a7c8 /src/armnn/LayersFwd.hpp
parente7ba17ec386ac5e6ddca02f49e9b8e06faa95d4e (diff)
downloadarmnn-3a40ea5e1efd61ebb21a174ef7bf9adecac8ade9.tar.gz
IVGCVSW-5363 Add Unmap layer and Unmap workload
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Signed-off-by: Jim Flynn <jim.flynn@arm.com> Change-Id: Ie5ecfa67e4763d0c058905592fe2e2fd7315f85c
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 f22110d3ac..ccc5ef2b4c 100644
--- a/src/armnn/LayersFwd.hpp
+++ b/src/armnn/LayersFwd.hpp
@@ -69,6 +69,7 @@
#include "layers/SwitchLayer.hpp"
#include "layers/TransposeConvolution2dLayer.hpp"
#include "layers/TransposeLayer.hpp"
+#include "layers/UnmapLayer.hpp"
namespace armnn
{
@@ -160,5 +161,6 @@ DECLARE_LAYER(Subtraction)
DECLARE_LAYER(Switch)
DECLARE_LAYER(Transpose)
DECLARE_LAYER(TransposeConvolution2d)
+DECLARE_LAYER(Unmap)
}