20.02
|
#include <ResolveType.hpp>
#include <armnn/Types.hpp>
#include <backendsCommon/CpuTensorHandle.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/WorkloadFactory.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadFactoryHelper.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>
Go to the source code of this file.
Functions | |
template<typename FactoryType , armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
void | DetectionPostProcessImpl (const armnn::TensorInfo &boxEncodingsInfo, const armnn::TensorInfo &scoresInfo, const armnn::TensorInfo &anchorsInfo, const std::vector< T > &boxEncodingsData, const std::vector< T > &scoresData, const std::vector< T > &anchorsData, const std::vector< float > &expectedDetectionBoxes, const std::vector< float > &expectedDetectionClasses, const std::vector< float > &expectedDetectionScores, const std::vector< float > &expectedNumDetections, bool useRegularNms) |
template<armnn::DataType QuantizedType, typename RawType = armnn::ResolveType<QuantizedType>> | |
void | QuantizeData (RawType *quant, const float *dequant, const armnn::TensorInfo &info) |
template<typename FactoryType > | |
void | DetectionPostProcessRegularNmsFloatTest () |
template<typename FactoryType , armnn::DataType QuantizedType, typename RawType = armnn::ResolveType<QuantizedType>> | |
void | DetectionPostProcessRegularNmsQuantizedTest () |
template<typename FactoryType > | |
void | DetectionPostProcessFastNmsFloatTest () |
template<typename FactoryType , armnn::DataType QuantizedType, typename RawType = armnn::ResolveType<QuantizedType>> | |
void | DetectionPostProcessFastNmsQuantizedTest () |
void DetectionPostProcessFastNmsFloatTest | ( | ) |
Definition at line 308 of file DetectionPostProcessTestImpl.hpp.
References armnn::Float32.
void DetectionPostProcessFastNmsQuantizedTest | ( | ) |
Definition at line 327 of file DetectionPostProcessTestImpl.hpp.
References anchorsInfo, scoresInfo, TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
Referenced by BOOST_AUTO_TEST_CASE().
void DetectionPostProcessImpl | ( | const armnn::TensorInfo & | boxEncodingsInfo, |
const armnn::TensorInfo & | scoresInfo, | ||
const armnn::TensorInfo & | anchorsInfo, | ||
const std::vector< T > & | boxEncodingsData, | ||
const std::vector< T > & | scoresData, | ||
const std::vector< T > & | anchorsData, | ||
const std::vector< float > & | expectedDetectionBoxes, | ||
const std::vector< float > & | expectedDetectionClasses, | ||
const std::vector< float > & | expectedDetectionScores, | ||
const std::vector< float > & | expectedNumDetections, | ||
bool | useRegularNms | ||
) |
Definition at line 139 of file DetectionPostProcessTestImpl.hpp.
References AllocateAndCopyDataToITensorHandle(), anchors(), anchorsInfo, boxEncodings(), CompareTensors(), CopyDataFromITensorHandle(), CopyDataToITensorHandle(), armnn::Float32, ProfilerManager::GetInstance(), DetectionPostProcessQueueDescriptor::m_Anchors, DetectionPostProcessDescriptor::m_DetectionsPerClass, DetectionPostProcessDescriptor::m_MaxClassesPerDetection, DetectionPostProcessDescriptor::m_MaxDetections, DetectionPostProcessDescriptor::m_NmsIouThreshold, DetectionPostProcessDescriptor::m_NmsScoreThreshold, DetectionPostProcessDescriptor::m_NumClasses, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, DetectionPostProcessDescriptor::m_ScaleH, DetectionPostProcessDescriptor::m_ScaleW, DetectionPostProcessDescriptor::m_ScaleX, DetectionPostProcessDescriptor::m_ScaleY, DetectionPostProcessDescriptor::m_UseRegularNms, ProfilerManager::RegisterProfiler(), scores(), and scoresInfo.
void DetectionPostProcessRegularNmsFloatTest | ( | ) |
Definition at line 244 of file DetectionPostProcessTestImpl.hpp.
References armnn::Float32.
void DetectionPostProcessRegularNmsQuantizedTest | ( | ) |
Definition at line 263 of file DetectionPostProcessTestImpl.hpp.
References anchorsInfo, scoresInfo, TensorInfo::SetQuantizationOffset(), and TensorInfo::SetQuantizationScale().
Referenced by BOOST_AUTO_TEST_CASE().
void QuantizeData | ( | RawType * | quant, |
const float * | dequant, | ||
const armnn::TensorInfo & | info | ||
) |
Definition at line 234 of file DetectionPostProcessTestImpl.hpp.
References TensorInfo::GetNumElements(), TensorInfo::GetQuantizationOffset(), and TensorInfo::GetQuantizationScale().