diff options
Diffstat (limited to 'src/armnnSerializer/Serializer.cpp')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index 5a095d9887..cf098eb5f8 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1035,6 +1035,25 @@ void SerializerStrategy::SerializeResizeLayer(const armnn::IConnectableLayer* la CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ResizeLayer); } +void SerializerStrategy::SerializeReverseV2Layer(const armnn::IConnectableLayer* layer, + const armnn::ReverseV2Descriptor& reverseV2Descriptor, + const char* name) +{ + IgnoreUnused(name); + + auto flatBufferBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_ReverseV2); + + auto flatBufferDescriptor = + CreateReverseV2Descriptor(m_flatBufferBuilder, + m_flatBufferBuilder.CreateVector(reverseV2Descriptor.m_Axis)); + + auto flatBufferLayer = serializer::CreateReverseV2Layer(m_flatBufferBuilder, + flatBufferBaseLayer, + flatBufferDescriptor); + + CreateAnyLayer(flatBufferLayer.o, serializer::Layer::Layer_ReverseV2Layer); +} + void SerializerStrategy::SerializeSliceLayer(const armnn::IConnectableLayer* layer, const armnn::SliceDescriptor& sliceDescriptor, const char* name) @@ -2332,6 +2351,13 @@ void SerializerStrategy::ExecuteStrategy(const armnn::IConnectableLayer* layer, SerializeResizeLayer(layer, layerDescriptor, name); break; } + case armnn::LayerType::ReverseV2: + { + const armnn::ReverseV2Descriptor& layerDescriptor = + static_cast<const armnn::ReverseV2Descriptor&>(descriptor); + SerializeReverseV2Layer(layer, layerDescriptor, name); + break; + } case armnn::LayerType::Shape: { SerializeShapeLayer(layer, name); |