diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-04 12:35:39 +0000 |
---|---|---|
committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-03-04 12:35:39 +0000 |
commit | b348521a8308d374e39c5d8da1a94622e89cf8df (patch) | |
tree | 9a9fb1fc81591eec09c3b5301bd72d0fc1fa7be3 /src/armnnSerializer/Serializer.cpp | |
parent | da1f975558a96499ea48cfca54e727a04b175271 (diff) | |
download | armnn-b348521a8308d374e39c5d8da1a94622e89cf8df.tar.gz |
IVGCVSW-2710 Add Serializer and Deserializer for StridedSlice
Change-Id: I6e8198a2aa5f8c56f00ccf9b4d98fcd208755654
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 25 |
1 files changed, 25 insertions, 0 deletions
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); |