GraphUtils.cpp File Reference

#include "GraphUtils.hpp"

Go to the source code of this file.

Functions
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)

Function Documentation

CheckNumberOfInputSlot()

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

Definition at line 32 of file GraphUtils.cpp.

References Layer::GetNumInputSlots().

Referenced by BOOST_FIXTURE_TEST_CASE().

{
    return layer->GetNumInputSlots() == num;
}

CheckNumberOfOutputSlot()

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

Definition at line 37 of file GraphUtils.cpp.

References Layer::GetNumOutputSlots().

Referenced by BOOST_FIXTURE_TEST_CASE().

{
    return layer->GetNumOutputSlots() == num;
}

GetFirstLayerWithName()

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

Definition at line 20 of file GraphUtils.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and BOOST_FIXTURE_TEST_CASE().

{
    for (auto&& layer : graph)
    {
        if (layer->GetNameStr() == name)
        {
            return layer;
        }
    }
    return nullptr;
}

GraphHasNamedLayer()

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

Definition at line 8 of file GraphUtils.cpp.

Referenced by BOOST_AUTO_TEST_CASE().

{
    for (auto&& layer : graph)
    {
        if (layer->GetName() == name)
        {
            return true;
        }
    }
    return false;
}

IsConnected()

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

Definition at line 42 of file GraphUtils.cpp.

References IOutputSlot::GetConnection(), Layer::GetNameStr(), IOutputSlot::GetNumConnections(), Layer::GetOutputSlot(), and IOutputSlot::GetTensorInfo().

Referenced by BOOST_FIXTURE_TEST_CASE().

{
    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 = boost::polymorphic_downcast<const armnn::InputSlot*>(outputSlot.GetConnection(c));
        if (inputSlot->GetOwningLayer().GetNameStr() == destLayer->GetNameStr() &&
            inputSlot->GetSlotIndex() == destSlot)
        {
            return true;
        }
    }
    return false;
}