aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2022-04-25 17:14:50 +0100
committerTeresa Charlin <teresa.charlinreyes@arm.com>2022-05-04 12:11:42 +0100
commit6966bfa643305fde25e96bb938cad811cd3b4f31 (patch)
tree7c6e377e0e0ba74d9e963a94c6cdc8f03ce4a407 /src/armnnSerializer/Serializer.hpp
parentb2d3ec5b1e938ef34facfdbcff83fc8e845d5f7c (diff)
downloadarmnn-6966bfa643305fde25e96bb938cad811cd3b4f31.tar.gz
IVGCVSW-6856 Add GATHERNd Serializer and Deserializer
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Ibab3525d53edbdf6a48e43b2bf668fcd2efaba58
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 3905e49cd1..98c1984cd2 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -191,6 +191,9 @@ private:
const armnn::GatherDescriptor& gatherDescriptor,
const char* name = nullptr);
+ void SerializeGatherNdLayer(const armnn::IConnectableLayer* layer,
+ const char* name = nullptr);
+
void SerializeInputLayer(const armnn::IConnectableLayer* layer,
armnn::LayerBindingId id,
const char* name = nullptr);