diff options
author | Samuel Yap <samuel.yap@arm.com> | 2022-08-08 14:07:42 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-08-30 17:03:33 +0100 |
commit | dc8ed9d75e54e914a970e137900930fa64a0782b (patch) | |
tree | 8bcaedaae81a6afbdbe3c9a4e69e45840f18cdb4 /src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp | |
parent | 9c9d5b9d796d243d88bd7a7aebb2e7e6c467e3a4 (diff) | |
download | armnn-dc8ed9d75e54e914a970e137900930fa64a0782b.tar.gz |
IVGCVSW-7105: BatchMatMul Optional Parameter Support
* Added transpose parameters to pre-transpose each input tensor's slices
* Added adjoint parameters to pre-adjoint each input tensor's slices
* Small refactoring (BatchMatMulDescriptor static helpers and BatchMatMulImpl constructor)
* Updated input validation and output shape inference for parameters
* Additional layer unit tests for parameters added
* Versionings incremented
Signed-off-by: Samuel Yap <samuel.yap@arm.com>
Change-Id: Ibe5242a8a5bf604c13de0dc65844fd6c421cc667
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp index 9e2139667b..0b261fba37 100644 --- a/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.hpp @@ -82,4 +82,22 @@ 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); + +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> +LayerTestResult<T, 2> BatchMatMul2DTranspSimpleTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> +LayerTestResult<T, 2> BatchMatMul2DAdjointSimpleTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); + +template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> +LayerTestResult<T, 4> BatchMatMulNHWCParamsTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, const armnn::ITensorHandleFactory& tensorHandleFactory);
\ No newline at end of file |