21.08
|
#include "FullyConnectedTestImpl.hpp"
#include <QuantizeHelper.hpp>
#include <backendsCommon/TensorHandle.hpp>
#include <backendsCommon/test/DataTypeUtils.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>
Go to the source code of this file.
Functions | |
template<typename T , typename B > | |
LayerTestResult< T, 2 > | SimpleFullyConnectedTestImpl (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, armnn::TensorInfo inputTensorInfo, armnn::TensorInfo outputTensorInfo, armnn::TensorInfo weightsTensorInfo, armnn::TensorInfo biasesTensorInfo, std::vector< T > &weights, std::vector< B > &bias, std::vector< T > &input, bool biasEnabled, bool transposeWeights, bool constantWeights) |
template<armnn::DataType ArmnnType, typename T > | |
LayerTestResult< T, 2 > | FullyConnectedTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool biasEnabled, bool constantWeights) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 2 > | FullyConnectedLargeTestCommon (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool transposeWeights, float qScale=0.0f, int32_t qOffset=0) |
template LayerTestResult< armnn::ResolveType< armnn::DataType::QAsymmU8 >, 2 > | FullyConnectedTest< armnn::DataType::QAsymmU8 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool biasEnabled, bool constWeights) |
template LayerTestResult< armnn::ResolveType< armnn::DataType::QSymmS16 >, 2 > | FullyConnectedTest< armnn::DataType::QSymmS16 > (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool biasEnabled, bool constWeights) |
LayerTestResult< float, 2 > | FullyConnectedFloat32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool biasEnabled, bool transposeWeights) |
LayerTestResult< float, 2 > | FullyConnectedLargeTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, bool transposeWeights) |
LayerTestResult<float, 2> FullyConnectedFloat32Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
bool | transposeWeights | ||
) |
Definition at line 278 of file FullyConnectedTestImpl.cpp.
References armnn::Float32, LayerTestResult< T, n >::m_ExpectedData, and armnn::swap().
Referenced by TEST_SUITE().
LayerTestResult<float, 2> FullyConnectedLargeTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | transposeWeights | ||
) |
Definition at line 373 of file FullyConnectedTestImpl.cpp.
Referenced by TEST_SUITE().
LayerTestResult<T, 2> FullyConnectedLargeTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | transposeWeights, | ||
float | qScale = 0.0f , |
||
int32_t | qOffset = 0 |
||
) |
Definition at line 177 of file FullyConnectedTestImpl.cpp.
References LayerTestResult< T, n >::m_ExpectedData, TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and armnn::swap().
LayerTestResult<T, 2> FullyConnectedTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
bool | constantWeights | ||
) |
Definition at line 95 of file FullyConnectedTestImpl.cpp.
References armnn::GetBiasTypeFromWeightsType(), LayerTestResult< T, n >::m_ExpectedData, and TensorInfo::SetQuantizationScale().
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 2> FullyConnectedTest< armnn::DataType::QAsymmU8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
bool | constWeights | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 2> FullyConnectedTest< armnn::DataType::QSymmS16 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
bool | biasEnabled, | ||
bool | constWeights | ||
) |
LayerTestResult<T, 2> SimpleFullyConnectedTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
armnn::TensorInfo | inputTensorInfo, | ||
armnn::TensorInfo | outputTensorInfo, | ||
armnn::TensorInfo | weightsTensorInfo, | ||
armnn::TensorInfo | biasesTensorInfo, | ||
std::vector< T > & | weights, | ||
std::vector< B > & | bias, | ||
std::vector< T > & | input, | ||
bool | biasEnabled, | ||
bool | transposeWeights, | ||
bool | constantWeights | ||
) |
Definition at line 24 of file FullyConnectedTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateFullyConnected(), ITensorHandleFactory::CreateTensorHandle(), TensorInfo::GetNumElements(), and LayerTestResult< T, n >::m_ActualData.