From b348521a8308d374e39c5d8da1a94622e89cf8df Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Mon, 4 Mar 2019 12:35:39 +0000 Subject: IVGCVSW-2710 Add Serializer and Deserializer for StridedSlice Change-Id: I6e8198a2aa5f8c56f00ccf9b4d98fcd208755654 Signed-off-by: Nattapat Chaimanowong --- src/armnnSerializer/Serializer.cpp | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'src/armnnSerializer/Serializer.cpp') diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 93231d0366..9653908493 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -588,6 +588,31 @@ void SerializerVisitor::VisitNormalizationLayer(const armnn::IConnectableLayer* 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); -- cgit v1.2.1