aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorSaoirse Stewart <saoirse.stewart@arm.com>2019-03-04 13:40:12 +0000
committerSaoirse Stewart <saoirse.stewart@arm.com>2019-03-04 13:40:12 +0000
commita1ed73a7c0780584a957509ddc4e65ad4de22c39 (patch)
tree7a4da5dda5ecea1c7134f63946a557c074bf3384 /src/armnnSerializer/Serializer.hpp
parentb348521a8308d374e39c5d8da1a94622e89cf8df (diff)
downloadarmnn-experimental/matteo.tar.gz
IVGCVSW-2691 Add Serialize/Deseralize Gather layerexperimental/matteo
Change-Id: I589d37c9f65801b701858d6e68e2e3151fac6e16 Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
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 0dcacc8898..1b1a3e998d 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -92,6 +92,9 @@ public:
const armnn::Optional<armnn::ConstTensor>& biases,
const char* name = nullptr) override;
+ void VisitGatherLayer(const armnn::IConnectableLayer* layer,
+ const char* name = nullptr) override;
+
void VisitGreaterLayer(const armnn::IConnectableLayer* layer,
const char* name = nullptr) override;