20.02
|
#include "GraphUtils.hpp"
#include <Graph.hpp>
#include <Layer.hpp>
#include <armnn/TypesUtils.hpp>
#include <armnn/Exceptions.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
#include <backendsCommon/TensorHandleFactoryRegistry.hpp>
#include <boost/cast.hpp>
#include <boost/test/unit_test.hpp>
Go to the source code of this file.
Functions | |
bool | CheckOrder (const armnn::Graph &graph, const armnn::Layer *first, const armnn::Layer *second) |
Checks that first comes before second in the order. More... | |
BOOST_AUTO_TEST_CASE (ClassGraph) | |
BOOST_AUTO_TEST_CASE (TopologicalSort) | |
BOOST_AUTO_TEST_CASE (InsertNewLayerBefore) | |
BOOST_AUTO_TEST_CASE (InsertNewLayerAfter) | |
BOOST_FIXTURE_TEST_CASE (AddCopyLayers, CopyLayersFixture) | |
BOOST_FIXTURE_TEST_CASE (AddCopyLayersSeveralTimes, CopyLayersFixture) | |
BOOST_FIXTURE_TEST_CASE (CopyLayersAddedBetweenSameLayersHaveDifferentNames, CopyLayersFixture) | |
BOOST_AUTO_TEST_CASE (DuplicateLayerNames) | |
BOOST_AUTO_TEST_CASE | ( | ClassGraph | ) |
Definition at line 36 of file GraphTests.cpp.
References Graph::AddLayer(), and GraphHasNamedLayer().
BOOST_AUTO_TEST_CASE | ( | TopologicalSort | ) |
Definition at line 43 of file GraphTests.cpp.
References Graph::AddLayer(), CheckOrder(), OutputSlot::Connect(), GetFirstLayerWithName(), Layer::GetInputSlot(), and Layer::GetOutputSlot().
BOOST_AUTO_TEST_CASE | ( | InsertNewLayerBefore | ) |
Definition at line 86 of file GraphTests.cpp.
References Graph::AddLayer(), CheckOrder(), OutputSlot::Connect(), armnn::Float32, GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), Graph::InsertNewLayer(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | InsertNewLayerAfter | ) |
Definition at line 172 of file GraphTests.cpp.
References Graph::AddLayer(), CheckOrder(), OutputSlot::Connect(), armnn::Float32, GetFirstLayerWithName(), Layer::GetInputSlot(), Layer::GetOutputSlot(), Graph::InsertNewLayer(), and OutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | DuplicateLayerNames | ) |
Definition at line 588 of file GraphTests.cpp.
References Graph::AddLayer(), Graph::begin(), BOOST_AUTO_TEST_SUITE_END(), OutputSlot::Connect(), armnn::CpuRef, Layer::GetInputSlot(), Layer::GetOutputSlot(), armnn::Input, armnn::Output, Layer::SetBackendId(), and Graph::TopologicalSort().
BOOST_FIXTURE_TEST_CASE | ( | AddCopyLayers | , |
CopyLayersFixture | |||
) |
Definition at line 528 of file GraphTests.cpp.
BOOST_FIXTURE_TEST_CASE | ( | AddCopyLayersSeveralTimes | , |
CopyLayersFixture | |||
) |
Definition at line 537 of file GraphTests.cpp.
BOOST_FIXTURE_TEST_CASE | ( | CopyLayersAddedBetweenSameLayersHaveDifferentNames | , |
CopyLayersFixture | |||
) |
Definition at line 552 of file GraphTests.cpp.
References Graph::AddCompatibilityLayers(), Graph::AddLayer(), OutputSlot::Connect(), armnn::CopyToTarget, armnn::CpuRef, armnn::DirectCompatibility, Layer::GetInputSlot(), Layer::GetOutputSlot(), armnn::GpuAcc, Layer::SetBackendId(), and OutputSlot::SetEdgeStrategy().
bool CheckOrder | ( | const armnn::Graph & | graph, |
const armnn::Layer * | first, | ||
const armnn::Layer * | second | ||
) |
Checks that first comes before second in the order.
Definition at line 22 of file GraphTests.cpp.
References BOOST_AUTO_TEST_SUITE(), Graph::Print(), and Graph::TopologicalSort().
Referenced by BOOST_AUTO_TEST_CASE().