ArmNN
NotReleased
|
#include "CommonTestUtils.hpp"
#include <Graph.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
#include <backendsCommon/WorkloadData.hpp>
#include <boost/cast.hpp>
#include <boost/test/unit_test.hpp>
#include <utility>
Go to the source code of this file.
Functions | |
BOOST_AUTO_TEST_CASE (ReleaseBatchNormalizationLayerConstantDataTest) | |
BOOST_AUTO_TEST_CASE (ReleaseConvolution2dLayerConstantDataTest) | |
BOOST_AUTO_TEST_CASE (ReleaseDepthwiseConvolution2dLayerConstantDataTest) | |
BOOST_AUTO_TEST_CASE (ReleaseFullyConnectedLayerConstantDataTest) | |
BOOST_AUTO_TEST_CASE | ( | ReleaseBatchNormalizationLayerConstantDataTest | ) |
Definition at line 29 of file LayerReleaseConstantDataTest.cpp.
References Graph::AddLayer(), BOOST_CHECK(), Connect(), armnn::Float32, BatchNormalizationLayer::m_Beta, BatchNormalizationDescriptor::m_Eps, BatchNormalizationLayer::m_Gamma, BatchNormalizationLayer::m_Mean, BatchNormalizationLayer::m_Variance, and Layer::ReleaseConstantData().
BOOST_AUTO_TEST_CASE | ( | ReleaseConvolution2dLayerConstantDataTest | ) |
Definition at line 75 of file LayerReleaseConstantDataTest.cpp.
References Graph::AddLayer(), BOOST_CHECK(), Connect(), armnn::Float32, armnn::GetBiasDataType(), Convolution2dLayer::m_Bias, Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, Convolution2dLayer::m_Weight, and Layer::ReleaseConstantData().
BOOST_AUTO_TEST_CASE | ( | ReleaseDepthwiseConvolution2dLayerConstantDataTest | ) |
Definition at line 119 of file LayerReleaseConstantDataTest.cpp.
References Graph::AddLayer(), BOOST_CHECK(), Connect(), armnn::Float32, DepthwiseConvolution2dLayer::m_Bias, DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, DepthwiseConvolution2dLayer::m_Weight, and Layer::ReleaseConstantData().
BOOST_AUTO_TEST_CASE | ( | ReleaseFullyConnectedLayerConstantDataTest | ) |
Definition at line 160 of file LayerReleaseConstantDataTest.cpp.
References Graph::AddLayer(), BOOST_AUTO_TEST_SUITE_END(), BOOST_CHECK(), Connect(), armnn::GetBiasDataType(), FullyConnectedLayer::m_Bias, FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_TransposeWeightMatrix, FullyConnectedLayer::m_Weight, armnn::QAsymmU8, and Layer::ReleaseConstantData().