aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LayersFwd.hpp
diff options
context:
space:
mode:
authorDerek Lamberti <derek.lamberti@arm.com>2019-03-25 15:41:58 +0000
committerderek.lamberti <derek.lamberti@arm.com>2019-03-27 16:06:46 +0000
commita9cca6aa935b3e290181a05fdb2c5f5557a49c09 (patch)
tree60493eeada1357fa2334f7f952879561480e192f /src/armnn/LayersFwd.hpp
parent045ea78ab73b1c88a38d3ecb501583a38161ca0e (diff)
downloadarmnn-a9cca6aa935b3e290181a05fdb2c5f5557a49c09.tar.gz
IVGCVSW-2870 Support QuantizeLayer on frontend
Change-Id: I2014a8d801f1f222d27a80dddf4f188ddcb3a5c9 Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
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 497b517f28..9300a75a07 100644
--- a/src/armnn/LayersFwd.hpp
+++ b/src/armnn/LayersFwd.hpp
@@ -39,6 +39,7 @@
#include "layers/PermuteLayer.hpp"
#include "layers/Pooling2dLayer.hpp"
#include "layers/PreCompiledLayer.hpp"
+#include "layers/QuantizeLayer.hpp"
#include "layers/ReshapeLayer.hpp"
#include "layers/ResizeBilinearLayer.hpp"
#include "layers/RsqrtLayer.hpp"
@@ -108,6 +109,7 @@ DECLARE_LAYER(Pad)
DECLARE_LAYER(Permute)
DECLARE_LAYER(Pooling2d)
DECLARE_LAYER(PreCompiled)
+DECLARE_LAYER(Quantize)
DECLARE_LAYER(Reshape)
DECLARE_LAYER(ResizeBilinear)
DECLARE_LAYER(Rsqrt)