20.02
|
#include <Graph.hpp>
#include <Network.hpp>
#include <reference/RefWorkloadFactory.hpp>
#include <boost/test/unit_test.hpp>
#include <test/GraphUtils.hpp>
Go to the source code of this file.
Functions | |
BOOST_AUTO_TEST_CASE (OptimizeValidateCpuRefWorkloads) | |
BOOST_AUTO_TEST_CASE (OptimizeValidateWorkloadsCpuRefPermuteLayer) | |
BOOST_AUTO_TEST_CASE (OptimizeValidateWorkloadsCpuRefMeanLayer) | |
BOOST_AUTO_TEST_CASE (DebugTestOnCpuRef) | |
BOOST_AUTO_TEST_CASE | ( | OptimizeValidateCpuRefWorkloads | ) |
Definition at line 16 of file RefOptimizedNetworkTests.cpp.
References BOOST_CHECK(), IOutputSlot::Connect(), armnn::CpuRef, IRuntime::Create(), armnn::Float32, IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), options, and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | OptimizeValidateWorkloadsCpuRefPermuteLayer | ) |
Definition at line 84 of file RefOptimizedNetworkTests.cpp.
References BOOST_CHECK(), IOutputSlot::Connect(), armnn::CpuRef, INetwork::Create(), IRuntime::Create(), armnn::Float32, IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), options, and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | OptimizeValidateWorkloadsCpuRefMeanLayer | ) |
Definition at line 117 of file RefOptimizedNetworkTests.cpp.
References BOOST_CHECK(), IOutputSlot::Connect(), armnn::CpuRef, INetwork::Create(), IRuntime::Create(), armnn::Float32, IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), options, and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | DebugTestOnCpuRef | ) |
Definition at line 150 of file RefOptimizedNetworkTests.cpp.
References Network::AddActivationLayer(), Network::AddInputLayer(), Network::AddOutputLayer(), BOOST_AUTO_TEST_SUITE_END(), armnn::BoundedReLu, IOutputSlot::Connect(), armnn::CpuRef, IRuntime::Create(), armnn::Float32, IConnectableLayer::GetOutputSlot(), GraphHasNamedLayer(), ActivationDescriptor::m_A, ActivationDescriptor::m_B, OptimizerOptions::m_Debug, ActivationDescriptor::m_Function, armnn::Optimize(), and options.