IVGCVSW-2914 Add Switch Layer and no-op factory method
Change-Id: I6a6ece708a49e8a97c83a3e7fec11c88af1e1cfa
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index 73db2e8..c1462c0 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -971,6 +971,11 @@
return m_Graph->AddLayer<MergeLayer>(name);
}
+IConnectableLayer* Network::AddSwitchLayer(const char* name)
+{
+ return m_Graph->AddLayer<SwitchLayer>(name);
+}
+
void Network::Accept(ILayerVisitor& visitor) const
{
for (auto layer : GetGraph())