GraphUtils.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd and Contributors. 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);
 
bool CheckOrder(const armnn::Graph& graph, const armnn::Layer* first, const armnn::Layer* second);