aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.cpp
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2020-06-04 16:34:23 +0100
committerKeithARM <keith.davis@arm.com>2020-06-17 14:54:08 +0000
commit300ad5695e2a577d2a9292b3cd6d182aae3298a3 (patch)
tree3fb34c3dc50b62630538592e6964fc263d078921 /src/armnnSerializer/Serializer.cpp
parent6398a98ac273931cc0b3ab33222d255d1edf48b0 (diff)
downloadarmnn-300ad5695e2a577d2a9292b3cd6d182aae3298a3.tar.gz
IVGCVSW-4908 Add Serializer/Deserializer Support for FILL operator
Signed-off-by: Keith Davis <keith.davis@arm.com> Change-Id: Icae26505d0e378ee5ffb3e92b35d78d48b369d2e
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r--src/armnnSerializer/Serializer.cpp13
1 files changed, 9 insertions, 4 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index ddd38e18ef..17076c62ab 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -472,10 +472,15 @@ void SerializerVisitor::VisitFillLayer(const armnn::IConnectableLayer* layer,
const armnn::FillDescriptor& fillDescriptor,
const char* name)
{
- throw UnimplementedException("SerializerVisitor::VisitFillLayer is not implemented");
IgnoreUnused(name);
- IgnoreUnused(layer);
- IgnoreUnused(fillDescriptor);
+
+ auto fbFillBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Fill);
+
+ auto fbDescriptor = serializer::CreateFillDescriptor(m_flatBufferBuilder, fillDescriptor.m_Value);
+
+ auto fbFillLayer = serializer::CreateFillLayer(m_flatBufferBuilder, fbFillBaseLayer, fbDescriptor);
+
+ CreateAnyLayer(fbFillLayer.o, serializer::Layer::Layer_FillLayer);
}
void SerializerVisitor::VisitFloorLayer(const armnn::IConnectableLayer *layer, const char *name)
@@ -1726,4 +1731,4 @@ bool Serializer::SaveSerializedToStream(std::ostream& stream)
return !stream.bad();
}
-} // namespace armnnSerializer
+} // namespace armnnSerializer \ No newline at end of file