IVGCVSW-2531 Serialize a simple ArmNN Network
Change-Id: I68cf5072aca6e3a8b3b8c57e19b6d417cd5813fc
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 5c70003..cad1690 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -917,6 +917,14 @@
return m_Graph->AddLayer<GatherLayer>(name);
}
+void Network::Accept(ILayerVisitor& visitor) const
+{
+ for (auto layer : GetGraph())
+ {
+ layer->Accept(visitor);
+ };
+}
+
OptimizedNetwork::OptimizedNetwork(std::unique_ptr<Graph> graph)
: m_Graph(std::move(graph))
{