diff options
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
| -rw-r--r-- | src/armnnSerializer/Serializer.cpp | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 2d5877db63..57d674095b 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -660,7 +660,20 @@ void SerializerVisitor::VisitResizeLayer(const armnn::IConnectableLayer* layer, const armnn::ResizeDescriptor& resizeDescriptor, const char* name) { - throw armnn::Exception("SerializerVisitor::VisitResizeLayer is not yet implemented"); + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Resize); + + auto flatBufferDescriptor = + CreateResizeDescriptor(m_flatBufferBuilder, + resizeDescriptor.m_TargetHeight, + resizeDescriptor.m_TargetWidth, + GetFlatBufferResizeMethod(resizeDescriptor.m_Method), + GetFlatBufferDataLayout(resizeDescriptor.m_DataLayout)); + + auto flatBufferLayer = serializer::CreateResizeLayer(m_flatBufferBuilder, + flatBufferBaseLayer, + flatBufferDescriptor); + + CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ResizeLayer); } void SerializerVisitor::VisitRsqrtLayer(const armnn::IConnectableLayer* layer, const char* name) |