22.05
|
#include <armnnTestUtils/LayerTestResult.hpp>
#include <Half.hpp>
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>
Go to the source code of this file.
Functions | |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 2 > | SimpleGatherNd2dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | SimpleGatherNd3dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 4 > | SimpleGatherNd4dTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
LayerTestResult<T, 2> SimpleGatherNd2dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 68 of file GatherNdTestImpl.cpp.
References TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and armnn::Signed32.
LayerTestResult<T, 3> SimpleGatherNd3dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 111 of file GatherNdTestImpl.cpp.
References TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and armnn::Signed32.
LayerTestResult<T, 4> SimpleGatherNd4dTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 172 of file GatherNdTestImpl.cpp.
References TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and armnn::Signed32.