ArmNN
NotReleased
|
Go to the source code of this file.
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) |
bool CheckRelatedLayers | ( | armnn::Graph & | graph, |
const std::list< std::string > & | testRelatedLayers | ||
) |
Definition at line 33 of file TestUtils.hpp.
|
inline |
Definition at line 20 of file TestUtils.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), and CheckSequence().
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().
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().
bool IsLayerOfType | ( | const armnn::Layer *const | layer | ) |