diff options
Diffstat (limited to 'src/armnn/test/GraphUtils.cpp')
-rw-r--r-- | src/armnn/test/GraphUtils.cpp | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/src/armnn/test/GraphUtils.cpp b/src/armnn/test/GraphUtils.cpp deleted file mode 100644 index bc6b562c9d..0000000000 --- a/src/armnn/test/GraphUtils.cpp +++ /dev/null @@ -1,78 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. 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()); -} |