diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-03-27 16:51:31 +0000 |
---|---|---|
committer | Derek Lamberti <derek.lamberti@arm.com> | 2019-03-27 16:51:31 +0000 |
commit | 87acb2733718526fc49e279386e2d85d60a74c8b (patch) | |
tree | 01194e1ab1b9116e5ee8d5f0dc33f9696ab60c35 /src/armnnSerializer/Serializer.cpp | |
parent | a9cca6aa935b3e290181a05fdb2c5f5557a49c09 (diff) | |
download | armnn-87acb2733718526fc49e279386e2d85d60a74c8b.tar.gz |
IVGCVSW-2870 Serialize quantize layer
Change-Id: I2cff85a3cb4d47aa09227a6810812a142f2aedd3
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 2fd840258e..83777c9849 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -681,6 +681,14 @@ void SerializerVisitor::VisitPooling2dLayer(const armnn::IConnectableLayer* laye CreateAnyLayer(fbPooling2dLayer.o, serializer::Layer::Layer_Pooling2dLayer); } +void SerializerVisitor::VisitQuantizeLayer(const armnn::IConnectableLayer *layer, const char *name) +{ + auto fbQuantizeBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Quantize); + auto fbQuantizeLayer = serializer::CreateQuantizeLayer(m_flatBufferBuilder, + fbQuantizeBaseLayer); + CreateAnyLayer(fbQuantizeLayer.o, serializer::Layer::Layer_QuantizeLayer); +} + // Build FlatBuffer for FullyConnected Layer void SerializerVisitor::VisitFullyConnectedLayer(const armnn::IConnectableLayer* layer, const armnn::FullyConnectedDescriptor& fullyConnectedDescriptor, |