20.11
|
#include "LayerTestResult.hpp"
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/WorkloadFactory.hpp>
Go to the source code of this file.
Functions | |
template<typename T , std::size_t n> | |
LayerTestResult< int32_t, 1 > | RankTest (armnn::TensorInfo inputTensorInfo, boost::multi_array< T, n > input, armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< int32_t, 1 > | RankDimSize1Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< int32_t, 1 > | RankDimSize2Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< int32_t, 1 > | RankDimSize3Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< int32_t, 1 > | RankDimSize4Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
LayerTestResult<int32_t, 1> RankDimSize1Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 53 of file RankTestImpl.cpp.
LayerTestResult<int32_t, 1> RankDimSize2Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 67 of file RankTestImpl.cpp.
LayerTestResult<int32_t, 1> RankDimSize3Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 81 of file RankTestImpl.cpp.
LayerTestResult<int32_t, 1> RankDimSize4Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 95 of file RankTestImpl.cpp.
LayerTestResult<int32_t, 1> RankTest | ( | armnn::TensorInfo | inputTensorInfo, |
boost::multi_array< T, n > | input, | ||
armnn::IWorkloadFactory & | workloadFactory, | ||
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 15 of file RankTestImpl.cpp.
References CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateRank(), ITensorHandleFactory::CreateTensorHandle(), armnn::IgnoreUnused(), LayerTestResult< T, n >::output, LayerTestResult< T, n >::outputExpected, armnn::Scalar, and armnn::Signed32.