22.08
|
#include <armnnTestUtils/LayerTestResult.hpp>
#include <ResolveType.hpp>
#include <armnn/backends/IBackendInternal.hpp>
Go to the source code of this file.
Functions | |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>, std::size_t NumDims> | |
LayerTestResult< T, NumDims > | BatchMatMulTestImpl (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, armnn::BatchMatMulDescriptor descriptor, const std::vector< T > &inputX, const std::vector< T > &inputY, const std::vector< T > &outputExpected, const armnn::TensorInfo &inputXInfo, const armnn::TensorInfo &inputYInfo, const armnn::TensorInfo &outputInfo) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 2 > | BatchMatMul2DSimpleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | BatchMatMul3DSimpleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 4 > | BatchMatMulNCHWSimpleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 4 > | BatchMatMulNHWCSimpleTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | BatchMatMul3DBatchTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | BatchMatMul3DBroadcastTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | BatchMatMul3D2DBroadcastTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 5 > | BatchMatMulNDHWCNHWCTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 2 > | BatchMatMul2DTinyTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> | |
LayerTestResult< T, 3 > | BatchMatMul3DNonSquareTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
LayerTestResult<T, 2> BatchMatMul2DSimpleTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 67 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 2> BatchMatMul2DTinyTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 815 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 3> BatchMatMul3D2DBroadcastTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 620 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 3> BatchMatMul3DBatchTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 427 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 3> BatchMatMul3DBroadcastTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 525 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 3> BatchMatMul3DNonSquareTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 901 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 3> BatchMatMul3DSimpleTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 156 of file BatchMatMulTestImpl.cpp.
References armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 4> BatchMatMulNCHWSimpleTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 245 of file BatchMatMulTestImpl.cpp.
References armnn::NCHW, armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 5> BatchMatMulNDHWCNHWCTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 715 of file BatchMatMulTestImpl.cpp.
References armnn::NDHWC, armnn::NHWC, armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, 4> BatchMatMulNHWCSimpleTest | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 336 of file BatchMatMulTestImpl.cpp.
References armnn::NHWC, armnn::QAsymmS8, armnn::QAsymmU8, and armnn::QSymmS16.
LayerTestResult<T, NumDims> BatchMatMulTestImpl | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory, | ||
armnn::BatchMatMulDescriptor | descriptor, | ||
const std::vector< T > & | inputX, | ||
const std::vector< T > & | inputY, | ||
const std::vector< T > & | outputExpected, | ||
const armnn::TensorInfo & | inputXInfo, | ||
const armnn::TensorInfo & | inputYInfo, | ||
const armnn::TensorInfo & | outputInfo | ||
) |
Definition at line 20 of file BatchMatMulTestImpl.cpp.
References armnn::BatchMatMul, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), TensorInfo::GetNumElements(), TensorInfo::GetShape(), and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.