src/armnn/test/GraphUtils.hpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once

#include <Graph.hpp>

#include <string>


bool GraphHasNamedLayer(const armnn::Graph& graph, const std::string& name);

armnn::Layer* GetFirstLayerWithName(armnn::Graph& graph, const std::string& name);

bool CheckNumberOfInputSlot(armnn::Layer* layer, unsigned int num);

bool CheckNumberOfOutputSlot(armnn::Layer* layer, unsigned int num);

bool IsConnected(armnn::Layer* srcLayer, armnn::Layer* destLayer,
                 unsigned int srcSlot, unsigned int destSlot,
                 const armnn::TensorInfo& expectedTensorInfo);