TestUtils.cpp File Reference

#include "TestUtils.hpp"
#include <boost/assert.hpp>

Go to the source code of this file.

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

Function Documentation

Connect()

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

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);
}