IVGCVSW-2367 Add Equal Operator to TfParser
* Unit tests in Equal.cpp
* Fixed error in Network::AddEqualLayer
* Refactored TfParser::Minimum/Equal to get rid of duplicate code
Change-Id: I0ed6f888eb391c995b88be20dc0c1b916dd14c3c
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp
index d798c84..0d95954 100644
--- a/src/armnn/Network.cpp
+++ b/src/armnn/Network.cpp
@@ -724,7 +724,7 @@
IConnectableLayer* Network::AddEqualLayer(const char* name)
{
- return m_Graph->AddLayer<GreaterLayer>(name);
+ return m_Graph->AddLayer<EqualLayer>(name);
}
OptimizedNetwork::OptimizedNetwork(std::unique_ptr<Graph> graph)