diff options
author | Kevin May <kevin.may@arm.com> | 2024-02-26 09:12:23 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-03-12 16:59:29 +0000 |
commit | e3cc71663e39a2d46bad3ef7e6c146ff9a9e5b0e (patch) | |
tree | 2559295d707eb88b0370370ccca36f250856a6a5 /src/armnnSerializer/Serializer.hpp | |
parent | 70dc5e9dc375c2feb306c7e709078c9f61ef8b1a (diff) | |
download | armnn-e3cc71663e39a2d46bad3ef7e6c146ff9a9e5b0e.tar.gz |
IVGCVSW-8230 Add ScatterNd to Serializer and Deserializer
* Added parsing functions to the serializer and deserializer
* Added ScatterNd and its Descriptor to the ArmnnSchema.fbs
* Added Unittest for Serializer and Deserializer
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I1ed674dc32d2e2d0d84dca4c7018984ea367ea50
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index afff66e21a..7434d63dd6 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017,2019-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017,2019-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -312,6 +312,10 @@ private: const armnn::NormalizationDescriptor& normalizationDescriptor, const char* name = nullptr); + void SerializeScatterNdLayer(const armnn::IConnectableLayer* layer, + const armnn::ScatterNdDescriptor& descriptor, + const char* name); + void SerializeShapeLayer(const armnn::IConnectableLayer* layer, const char* name = nullptr); |