commit b348521a8308d374e39c5d8da1a94622e89cf8df
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Mon Mar 04 12:35:39 2019 +0000
committer Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Mon Mar 04 12:35:39 2019 +0000
tree 9a9fb1fc81591eec09c3b5301bd72d0fc1fa7be3
parent da1f975558a96499ea48cfca54e727a04b175271

IVGCVSW-2710 Add Serializer and Deserializer for StridedSlice

Change-Id: I6e8198a2aa5f8c56f00ccf9b4d98fcd208755654
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>

src/armnnSerializer/Serializer.cpp

diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index 93231d0..9653908 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -588,6 +588,31 @@
     CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_NormalizationLayer);
 }
 
+void SerializerVisitor::VisitStridedSliceLayer(const armnn::IConnectableLayer* layer,
+                                               const armnn::StridedSliceDescriptor& stridedSliceDescriptor,
+                                               const char* name)
+{
+    auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_StridedSlice);
+
+    auto flatBufferDescriptor =
+        CreateStridedSliceDescriptor(m_flatBufferBuilder,
+                                     m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_Begin),
+                                     m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_End),
+                                     m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_Stride),
+                                     stridedSliceDescriptor.m_BeginMask,
+                                     stridedSliceDescriptor.m_EndMask,
+                                     stridedSliceDescriptor.m_ShrinkAxisMask,
+                                     stridedSliceDescriptor.m_EllipsisMask,
+                                     stridedSliceDescriptor.m_NewAxisMask,
+                                     GetFlatBufferDataLayout(stridedSliceDescriptor.m_DataLayout));
+
+    auto flatBufferLayer = serializer::CreateStridedSliceLayer(m_flatBufferBuilder,
+                                                               flatBufferBaseLayer,
+                                                               flatBufferDescriptor);
+
+    CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_StridedSliceLayer);
+}
+
 void SerializerVisitor::VisitSubtractionLayer(const armnn::IConnectableLayer* layer, const char* name)
 {
     auto fbSubtractionBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Subtraction);