commit 300ad5695e2a577d2a9292b3cd6d182aae3298a3
author Keith Davis <keith.davis@arm.com> Thu Jun 04 16:34:23 2020 +0100
committer KeithARM <keith.davis@arm.com> Wed Jun 17 14:54:08 2020 +0000
tree 3fb34c3dc50b62630538592e6964fc263d078921
parent 6398a98ac273931cc0b3ab33222d255d1edf48b0

IVGCVSW-4908 Add Serializer/Deserializer Support for FILL operator

Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: Icae26505d0e378ee5ffb3e92b35d78d48b369d2e

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 532c12c..18415ce 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -156,7 +156,8 @@
     StandIn = 53,
     ElementwiseUnary = 54,
     Transpose = 55,
-    QLstm = 56
+    QLstm = 56,
+    Fill = 57
 }
 
 // Base layer table to be used as part of other layers
@@ -284,6 +285,15 @@
     base:LayerBase;
 }
 
+table FillLayer {
+    base:LayerBase;
+    descriptor:FillDescriptor;
+}
+
+table FillDescriptor {
+    value:float;
+}
+
 table FloorLayer{
     base:LayerBase;
 }
@@ -901,7 +911,8 @@
     StandInLayer,
     ElementwiseUnaryLayer,
     TransposeLayer,
-    QLstmLayer
+    QLstmLayer,
+    FillLayer
 }
 
 table AnyLayer {
@@ -920,4 +931,4 @@
     featureVersions:FeatureCompatibilityVersions;
 }
 
-root_type SerializedGraph;
+root_type SerializedGraph;
\ No newline at end of file

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index ddd38e1..17076c6 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -472,10 +472,15 @@
                                        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 @@
     return !stream.bad();
 }
 
-} // namespace armnnSerializer
+} // namespace armnnSerializer
\ No newline at end of file

src/armnnSerializer/SerializerSupport.md

diff --git a/src/armnnSerializer/SerializerSupport.md b/src/armnnSerializer/SerializerSupport.md
index f8d8655..4f7868b 100644
--- a/src/armnnSerializer/SerializerSupport.md
+++ b/src/armnnSerializer/SerializerSupport.md
@@ -21,6 +21,7 @@
 * DetectionPostProcess
 * Division
 * ElementwiseUnary
+* Fill
 * Floor
 * FullyConnected
 * Gather

src/armnnSerializer/test/SerializerTests.cpp

diff --git a/src/armnnSerializer/test/SerializerTests.cpp b/src/armnnSerializer/test/SerializerTests.cpp
index e7f93c6..fa43e09 100644
--- a/src/armnnSerializer/test/SerializerTests.cpp
+++ b/src/armnnSerializer/test/SerializerTests.cpp
@@ -1199,6 +1199,35 @@
     deserializedNetwork->Accept(verifier);
 }
 
+BOOST_AUTO_TEST_CASE(SerializeFill)
+{
+    DECLARE_LAYER_VERIFIER_CLASS_WITH_DESCRIPTOR(Fill)
+
+    const std::string layerName("fill");
+    const armnn::TensorInfo inputInfo({4}, armnn::DataType::Float32);
+    const armnn::TensorInfo outputInfo({1, 3, 3, 1}, armnn::DataType::Float32);
+
+    armnn::FillDescriptor descriptor(1.0f);
+
+    armnn::INetworkPtr network = armnn::INetwork::Create();
+    armnn::IConnectableLayer* const inputLayer = network->AddInputLayer(0);
+    armnn::IConnectableLayer* const fillLayer = network->AddFillLayer(descriptor, layerName.c_str());
+    armnn::IConnectableLayer* const outputLayer = network->AddOutputLayer(0);
+
+    inputLayer->GetOutputSlot(0).Connect(fillLayer->GetInputSlot(0));
+    fillLayer->GetOutputSlot(0).Connect(outputLayer->GetInputSlot(0));
+
+    inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
+    fillLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
+
+    armnn::INetworkPtr deserializedNetwork = DeserializeNetwork(SerializeNetwork(*network));
+    BOOST_CHECK(deserializedNetwork);
+
+    FillLayerVerifier verifier(layerName, {inputInfo}, {outputInfo}, descriptor);
+
+    deserializedNetwork->Accept(verifier);
+}
+
 BOOST_AUTO_TEST_CASE(SerializeFloor)
 {
     DECLARE_LAYER_VERIFIER_CLASS(Floor)