14 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>, std::
size_t NumDims>
20 const std::vector<T>& inputX,
21 const std::vector<T>& inputY,
22 const std::vector<T>& outputExpected,
27 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
33 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
39 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
45 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
51 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
57 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
63 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
69 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
75 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
81 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
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)
LayerTestResult< T, 3 > BatchMatMul3DBroadcastTest(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)
LayerTestResult< T, 3 > BatchMatMul3D2DBroadcastTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
LayerTestResult< T, 4 > BatchMatMulNCHWSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 5 > BatchMatMulNDHWCNHWCTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 3 > BatchMatMul3DSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
A BatchMatMulDescriptor for the BatchMatMul operator.
LayerTestResult< T, 4 > BatchMatMulNHWCSimpleTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 3 > BatchMatMul3DNonSquareTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 3 > BatchMatMul3DBatchTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
LayerTestResult< T, 2 > BatchMatMul2DTinyTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)