diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-11-24 15:47:28 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-12-14 11:02:41 +0000 |
commit | a097d2a0ed8e30d5aaf6d29ec18d0c39201b7b67 (patch) | |
tree | 947e587bc42d07f52c55b155308b5ea5bd3ebacd /src/armnn/test/GraphUtils.cpp | |
parent | bc14881a76699dd942e94265116da68a6466455e (diff) | |
download | armnn-a097d2a0ed8e30d5aaf6d29ec18d0c39201b7b67.tar.gz |
IVGCVSW-6453 'Move the ArmNN Test Utils code to a physically separate directory'
* Created include/armnnTestUtils directory
* Moved Arm NN test utils files into armnnTestUtils directory
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I03ac54c645c41c52650c4c03b6a58fb1481fef5d
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()); -} |