20.02
|
#include "FullyConnectedTestImpl.hpp"
#include <QuantizeHelper.hpp>
#include <backendsCommon/CpuTensorHandle.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.
LayerTestResult<float, 2> FullyConnectedFloat32Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | biasEnabled, | ||
bool | transposeWeights | ||
) |
Definition at line 247 of file FullyConnectedTestImpl.cpp.
References armnn::Float32, LayerTestResult< T, n >::outputExpected, and armnn::swap().
LayerTestResult<float, 2> FullyConnectedLargeTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | transposeWeights | ||
) |
Definition at line 344 of file FullyConnectedTestImpl.cpp.
LayerTestResult<T, 2> FullyConnectedLargeTestCommon | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | transposeWeights, | ||
float | qScale = 0.0f , |
||
int32_t | qOffset = 0 |
||
) |
Definition at line 148 of file FullyConnectedTestImpl.cpp.
References LayerTestResult< T, n >::outputExpected, TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and armnn::swap().
LayerTestResult<T, 2> FullyConnectedTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | biasEnabled | ||
) |
Definition at line 71 of file FullyConnectedTestImpl.cpp.
References armnn::GetBiasTypeFromWeightsType(), LayerTestResult< T, n >::outputExpected, TensorInfo::SetQuantizationScale(), and true.
template LayerTestResult<armnn::ResolveType<armnn::DataType::QAsymmU8>, 2> FullyConnectedTest< armnn::DataType::QAsymmU8 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | biasEnabled | ||
) |
template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 2> FullyConnectedTest< armnn::DataType::QSymmS16 > | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
bool | biasEnabled | ||
) |
LayerTestResult<T, 2> SimpleFullyConnectedTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::TensorInfo | inputTensorInfo, | ||
armnn::TensorInfo | outputTensorInfo, | ||
armnn::TensorInfo | weightsDesc, | ||
armnn::TensorInfo | biasesDesc, | ||
boost::multi_array< T, 2 > & | weights, | ||
boost::multi_array< B, 1 > & | bias, | ||
boost::multi_array< T, 4 > & | input, | ||
bool | biasEnabled, | ||
bool | transposeWeights | ||
) |
Definition at line 24 of file FullyConnectedTestImpl.cpp.
References AllocateAndCopyDataToITensorHandle(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateFullyConnected(), IWorkloadFactory::CreateTensorHandle(), armnn::IgnoreUnused(), FullyConnectedQueueDescriptor::m_Bias, FullyConnectedDescriptor::m_BiasEnabled, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, FullyConnectedDescriptor::m_TransposeWeightMatrix, FullyConnectedQueueDescriptor::m_Weight, and LayerTestResult< T, n >::output.