1 files changed, 78 insertions, 0 deletions
diff --git a/src/armnnTestUtils/GraphUtils.cpp b/src/armnnTestUtils/GraphUtils.cpp
new file mode 100644
index 0000000000..15dc888e21
--- /dev/null
+++ b/src/armnnTestUtils/GraphUtils.cpp
@@ -0,0 +1,78 @@
+//
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "GraphUtils.hpp"
+
+#include <armnn/utility/PolymorphicDowncast.hpp>
+
+bool GraphHasNamedLayer(const armnn::Graph& graph, const std::string& name)
+{
+    for (auto&& layer : graph)
+    {
+        if (layer->GetName() == name)
+        {
+            return true;
+        }
+    }
+    return false;
+}
+
+armnn::Layer* GetFirstLayerWithName(armnn::Graph& graph, const std::string& name)
+{
+    for (auto&& layer : graph)
+    {
+        if (layer->GetNameStr() == name)
+        {
+            return layer;
+        }
+    }
+    return nullptr;
+}
+
+bool CheckNumberOfInputSlot(armnn::Layer* layer, unsigned int num)
+{
+    return layer->GetNumInputSlots() == num;
+}
+
+bool CheckNumberOfOutputSlot(armnn::Layer* layer, unsigned int num)
+{
+    return layer->GetNumOutputSlots() == num;
+}
+
+bool IsConnected(armnn::Layer* srcLayer, armnn::Layer* destLayer,
+                 unsigned int srcSlot, unsigned int destSlot,
+                 const armnn::TensorInfo& expectedTensorInfo)
+{
+    const armnn::IOutputSlot& outputSlot = srcLayer->GetOutputSlot(srcSlot);
+    const armnn::TensorInfo& tensorInfo = outputSlot.GetTensorInfo();
+    if (expectedTensorInfo != tensorInfo)
+    {
+        return false;
+    }
+    const unsigned int numConnections = outputSlot.GetNumConnections();
+    for (unsigned int c = 0; c < numConnections; ++c)
+    {
+        auto inputSlot = armnn::PolymorphicDowncast<const armnn::InputSlot*>(outputSlot.GetConnection(c));
+        if (inputSlot->GetOwningLayer().GetNameStr() == destLayer->GetNameStr() &&
+            inputSlot->GetSlotIndex() == destSlot)
+        {
+            return true;
+        }
+    }
+    return false;
+}
+
+/// Checks that first comes before second in the order.
+bool CheckOrder(const armnn::Graph& graph, const armnn::Layer* first, const armnn::Layer* second)
+{
+    graph.Print();
+
+    const auto& order = graph.TopologicalSort();
+
+    auto firstPos = std::find(order.begin(), order.end(), first);
+    auto secondPos = std::find(firstPos, order.end(), second);
+
+    return (secondPos != order.end());
+}