blob: b51e4d179eeddc225fb54c07fe77fed7c8b963c0 [file] [log] [blame]
//
// 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);