diff options
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index a8d9c23c04..595573c417 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -107,7 +107,10 @@ void SerializerVisitor::VisitOutputLayer(const armnn::IConnectableLayer* layer, void SerializerVisitor::VisitAbsLayer(const armnn::IConnectableLayer* layer, const char* name) { - throw UnimplementedException("SerializerVisitor::VisitAbsLayer is not implemented"); + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_Abs); + auto flatBufferAbsLayer = serializer::CreateAbsLayer(m_flatBufferBuilder, flatBufferBaseLayer); + + CreateAnyLayer(flatBufferAbsLayer.o, serializer::Layer::Layer_AbsLayer); } // Build FlatBuffer for Activation Layer |