diff options
author | Tracy Narine <tracy.narine@arm.com> | 2023-07-04 15:08:57 +0100 |
---|---|---|
committer | Tracy Narine <tracy.narine@arm.com> | 2023-07-07 10:05:28 +0000 |
commit | 944fb508b1c30415e423b8916849c66a13867ea4 (patch) | |
tree | b4a3c96425184965415f0adc3a9cd7a0011e97cd /src/armnnSerializer/Serializer.cpp | |
parent | 8e9747ab5211460e8dcc5e81d97e567152a0b710 (diff) | |
download | armnn-944fb508b1c30415e423b8916849c66a13867ea4.tar.gz |
IVGCVSW-7832 Add REVERSE_V2 to Serializer and Deserializer
* Support for ReverseV2 for the serializer and deserializer added
* Tests added
* CMake files updated for the build
* Fixed an issue with the operator_list documentation for Resize and ReverseV2
Signed-off-by: Tracy Narine <tracy.narine@arm.com>
Change-Id: If396f55ecdd763d6f91c51707809f4bd58715cec
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); |