#include <armnn/INetwork.hpp>
#include <Graph.hpp>

Functions
void	Connect (armnn::IConnectableLayer from, armnn::IConnectableLayer to, const armnn::TensorInfo &tensorInfo, unsigned int fromIndex=0, unsigned int toIndex=0)

template<typename LayerT >
bool	IsLayerOfType (const armnn::Layer *const layer)

bool	CheckSequence (const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last)

template<typename U , typename... Us>
bool	CheckSequence (const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last, U &&u, Us &&... us)
	Checks each unary function in Us evaluates true for each correspondent layer in the sequence [first, last). More...

template<typename LayerT >
bool	CheckRelatedLayers (armnn::Graph &graph, const std::list< std::string > &testRelatedLayers)

Function Documentation

◆ CheckRelatedLayers()

bool CheckRelatedLayers	(	armnn::Graph &	graph,
		const std::list< std::string > &	testRelatedLayers
	)

Definition at line 33 of file TestUtils.hpp.

 {
     for (auto& layer : graph)
     {
         if (layer->GetType() == armnn::LayerEnumOf<LayerT>())
         {
             auto& relatedLayers = layer->GetRelatedLayerNames();
             if (!std::equal(relatedLayers.begin(), relatedLayers.end(), testRelatedLayers.begin(),
                             testRelatedLayers.end()))
             {
                 return false;
             }
         }
     }
 
     return true;
 }

◆ CheckSequence() [1/2]

bool CheckSequence	(	const armnn::Graph::ConstIterator	first,
		const armnn::Graph::ConstIterator	last
	)

inline

Definition at line 20 of file TestUtils.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), and CheckSequence().

 {
     return (first == last);
 }

◆ CheckSequence() [2/2]

bool CheckSequence	(	const armnn::Graph::ConstIterator	first,
		const armnn::Graph::ConstIterator	last,
		U &&	u,
		Us &&...	us
	)

Checks each unary function in Us evaluates true for each correspondent layer in the sequence [first, last).

Definition at line 27 of file TestUtils.hpp.

References CheckSequence().

 {
     return u(*first) && CheckSequence(std::next(first), last, us...);
 }

◆ Connect()

void Connect	(	armnn::IConnectableLayer *	from,
		armnn::IConnectableLayer *	to,
		const armnn::TensorInfo &	tensorInfo,
		unsigned int	fromIndex = `0`,
		unsigned int	toIndex = `0`
	)

Definition at line 12 of file TestUtils.cpp.

References IOutputSlot::Connect(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), and IOutputSlot::SetTensorInfo().

Referenced by BOOST_AUTO_TEST_CASE(), OutputSlot::Connect(), CreatePreluLayerHelper(), CreateStackLayerHelper(), OutputSlot::GetOutputHandler(), CaffeParserBase::ParseBatchNormLayer(), CaffeParserBase::ParseEltwiseLayer(), CaffeParserBase::ParseInnerProductLayer(), CaffeParserBase::ParseLRNLayer(), CaffeParserBase::ParsePoolingLayer(), CaffeParserBase::ParseReluLayer(), CaffeParserBase::ParseScaleLayer(), and CaffeParserBase::ParseSoftmaxLayer().

 {
     BOOST_ASSERT(from);
     BOOST_ASSERT(to);
 
     from->GetOutputSlot(fromIndex).Connect(to->GetInputSlot(toIndex));
     from->GetOutputSlot(fromIndex).SetTensorInfo(tensorInfo);
 }

◆ IsLayerOfType()

bool IsLayerOfType ( const armnn::Layer *const layer )

Definition at line 15 of file TestUtils.hpp.

References Layer::GetType().