diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-03-04 14:34:30 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-03-04 16:06:42 +0000 |
commit | f81edaacc38d6edc4a2dc230460120c6f83e0cda (patch) | |
tree | f2d1833efde82b1254c96b343810c56e80d7eb40 /src/armnnSerializer/Serializer.cpp | |
parent | 5c7ab05555d3abd23a11db403090b7563ce388e6 (diff) | |
download | armnn-f81edaacc38d6edc4a2dc230460120c6f83e0cda.tar.gz |
IVGCVSW-2691 Add Serialize/Deseralize Gather layer
Change-Id: I445c4475e5abfe500b61ce8b7138c45322043c8b
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 38e815d9e0..4e408ae807 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -331,10 +331,8 @@ void SerializerVisitor::VisitMaximumLayer(const armnn::IConnectableLayer* layer, void SerializerVisitor::VisitGatherLayer(const armnn::IConnectableLayer* layer, const char* name) { - auto fbBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Gather); - - auto flatBufferLayer = CreateGatherLayer(m_flatBufferBuilder, - fbBaseLayer); + auto fbGatherBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Gather); + auto flatBufferLayer = serializer::CreateGatherLayer(m_flatBufferBuilder, fbGatherBaseLayer); CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_GatherLayer); } @@ -342,7 +340,7 @@ void SerializerVisitor::VisitGatherLayer(const armnn::IConnectableLayer* layer, void SerializerVisitor::VisitGreaterLayer(const armnn::IConnectableLayer* layer, const char* name) { auto fbGreaterBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Greater); - auto fbGreaterLayer = serializer::CreateGreaterLayer(m_flatBufferBuilder, fbGreaterBaseLayer); + auto fbGreaterLayer = serializer::CreateGreaterLayer(m_flatBufferBuilder, fbGreaterBaseLayer); CreateAnyLayer(fbGreaterLayer.o, serializer::Layer::Layer_GreaterLayer); } |