21.05
|
#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 Graph::AllocateDynamicBuffers(), IOutputSlot::Connect(), armnn::CpuRef, IRuntime::Create(), INetwork::Create(), armnn::Float32, armnn::GetGraphForTesting(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | OptimizeValidateWorkloadsCpuRefPermuteLayer | ) |
Definition at line 86 of file RefOptimizedNetworkTests.cpp.
References Graph::AllocateDynamicBuffers(), IOutputSlot::Connect(), armnn::CpuRef, IRuntime::Create(), INetwork::Create(), armnn::Float32, armnn::GetGraphForTesting(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | OptimizeValidateWorkloadsCpuRefMeanLayer | ) |
Definition at line 122 of file RefOptimizedNetworkTests.cpp.
References Graph::AllocateDynamicBuffers(), IOutputSlot::Connect(), armnn::CpuRef, IRuntime::Create(), INetwork::Create(), armnn::Float32, armnn::GetGraphForTesting(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), armnn::Optimize(), and IOutputSlot::SetTensorInfo().
BOOST_AUTO_TEST_CASE | ( | DebugTestOnCpuRef | ) |
Definition at line 156 of file RefOptimizedNetworkTests.cpp.
References Graph::AllocateDynamicBuffers(), BOOST_AUTO_TEST_SUITE_END(), armnn::BoundedReLu, armnn::CpuRef, IRuntime::Create(), INetwork::Create(), armnn::Float32, armnn::GetGraphForTesting(), GraphHasNamedLayer(), ActivationDescriptor::m_A, ActivationDescriptor::m_B, OptimizerOptions::m_Debug, ActivationDescriptor::m_Function, and armnn::Optimize().