ArmNN
NotReleased
|
#include "LayerTestResult.hpp"
#include <armnn/ArmNN.hpp>
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/Workload.hpp>
#include <backendsCommon/WorkloadData.hpp>
#include <backendsCommon/WorkloadFactory.hpp>
#include <backendsCommon/test/DataTypeUtils.hpp>
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>
#include <memory>
Go to the source code of this file.
Functions | |
std::unique_ptr< armnn::IWorkload > | CreateWorkload (const armnn::IWorkloadFactory &workloadFactory, const armnn::WorkloadInfo &info, const armnn::ElementwiseUnaryQueueDescriptor &descriptor) |
template<std::size_t NumDims, armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, NumDims > | ElementwiseUnaryTestHelper (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, armnn::UnaryOperation op, const unsigned int shape[NumDims], std::vector< float > values, float quantScale, int quantOffset, const unsigned int outShape[NumDims], std::vector< float > outValues, float outQuantScale, int outQuantOffset) |
template<std::size_t NumDims, armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, NumDims > | ElementwiseUnaryTestHelper (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, armnn::UnaryOperation op, const unsigned int shape[NumDims], std::vector< float > values, const unsigned int outShape[NumDims], std::vector< float > outValues, float quantScale=1.0f, int quantOffset=0) |
std::unique_ptr<armnn::IWorkload> CreateWorkload | ( | const armnn::IWorkloadFactory & | workloadFactory, |
const armnn::WorkloadInfo & | info, | ||
const armnn::ElementwiseUnaryQueueDescriptor & | descriptor | ||
) |
Definition at line 8 of file ElementwiseUnaryTestImpl.cpp.
References IWorkloadFactory::CreateElementwiseUnary().
Referenced by ElementwiseUnaryTestHelper().
LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::UnaryOperation | op, | ||
const unsigned int | shape[NumDims], | ||
std::vector< float > | values, | ||
float | quantScale, | ||
int | quantOffset, | ||
const unsigned int | outShape[NumDims], | ||
std::vector< float > | outValues, | ||
float | outQuantScale, | ||
int | outQuantOffset | ||
) |
Definition at line 35 of file ElementwiseUnaryTestImpl.hpp.
References CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateTensorHandle(), CreateWorkload(), QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, LayerTestResult< T, n >::output, LayerTestResult< T, n >::outputExpected, and TensorInfo::SetQuantizationScale().
LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
armnn::UnaryOperation | op, | ||
const unsigned int | shape[NumDims], | ||
std::vector< float > | values, | ||
const unsigned int | outShape[NumDims], | ||
std::vector< float > | outValues, | ||
float | quantScale = 1.0f , |
||
int | quantOffset = 0 |
||
) |
Definition at line 90 of file ElementwiseUnaryTestImpl.hpp.