ArmNN
NotReleased
|
#include <boost/test/unit_test.hpp>
#include <Graph.hpp>
#include <SubgraphView.hpp>
#include <SubgraphViewSelector.hpp>
#include <armnn/backends/OptimizationViews.hpp>
#include <Network.hpp>
#include "CommonTestUtils.hpp"
#include "MockBackend.hpp"
Go to the source code of this file.
Functions | |
void | CheckLayers (Graph &graph) |
BOOST_AUTO_TEST_CASE (OptimizedViewsSubgraphLayerCount) | |
BOOST_AUTO_TEST_CASE (OptimizedViewsSubgraphLayerCountFailValidate) | |
BOOST_AUTO_TEST_CASE (OptimizeViewsValidateDeviceMockBackend) | |
BOOST_AUTO_TEST_CASE | ( | OptimizedViewsSubgraphLayerCount | ) |
Definition at line 51 of file OptimizationViewsTests.cpp.
References OptimizationViews::AddFailedSubgraph(), Graph::AddLayer(), OptimizationViews::AddSubstitution(), OptimizationViews::AddUntouchedSubgraph(), BOOST_CHECK(), OutputSlot::Connect(), CreateInputsFrom(), CreateOutputsFrom(), CreateSubgraphViewFrom(), OptimizationViews::GetGraph(), Layer::GetInputSlot(), Layer::GetOutputSlot(), Graph::SubstituteSubgraph(), and OptimizationViews::Validate().
BOOST_AUTO_TEST_CASE | ( | OptimizedViewsSubgraphLayerCountFailValidate | ) |
Definition at line 119 of file OptimizationViewsTests.cpp.
References Graph::AddLayer(), OptimizationViews::AddSubstitution(), OptimizationViews::AddUntouchedSubgraph(), BOOST_CHECK(), OutputSlot::Connect(), CreateInputsFrom(), CreateOutputsFrom(), CreateSubgraphViewFrom(), OptimizationViews::GetGraph(), Layer::GetInputSlot(), Layer::GetOutputSlot(), Graph::SubstituteSubgraph(), and OptimizationViews::Validate().
BOOST_AUTO_TEST_CASE | ( | OptimizeViewsValidateDeviceMockBackend | ) |
Definition at line 182 of file OptimizationViewsTests.cpp.
References BOOST_AUTO_TEST_SUITE_END(), BOOST_CHECK(), CheckLayers(), IOutputSlot::Connect(), INetwork::Create(), IRuntime::Create(), armnn::Float32, OptimizedNetwork::GetGraph(), MockBackend::GetIdStatic(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), options, and IOutputSlot::SetTensorInfo().
void CheckLayers | ( | Graph & | graph | ) |
Definition at line 18 of file OptimizationViewsTests.cpp.
References BOOST_AUTO_TEST_SUITE(), armnn::Input, armnn::Output, and armnn::PreCompiled.
Referenced by BOOST_AUTO_TEST_CASE().