diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-10-31 22:09:23 +0000 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2022-11-01 12:43:16 +0000 |
commit | a38da5989dc51fe2665480d18a9dd9dff6bb1115 (patch) | |
tree | ae30172be63f7a14b81b5bc02dfe01411d7272db /src/backends/neon/test/NeonEndToEndTests.cpp | |
parent | a4035b8970963fae22760df8f4234e9976c8cf85 (diff) | |
download | armnn-a38da5989dc51fe2665480d18a9dd9dff6bb1115.tar.gz |
IVGCVSW-6496 Add EndToEnd Layer test for Batch MatMul Workload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I6a541db9a602609282cc6f33af930ca141b83c41
Diffstat (limited to 'src/backends/neon/test/NeonEndToEndTests.cpp')
-rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index d680e6deed..4c7608ff93 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -7,6 +7,7 @@ #include <backendsCommon/test/ActivationEndToEndTestImpl.hpp> #include <backendsCommon/test/ArgMinMaxEndToEndTestImpl.hpp> +#include <backendsCommon/test/BatchMatMulEndToEndTestImpl.hpp> #include <backendsCommon/test/ComparisonEndToEndTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> #include <backendsCommon/test/DepthToSpaceEndToEndTestImpl.hpp> @@ -128,6 +129,11 @@ TEST_CASE("NeonGreaterBroadcastEndToEndUint8Test") expectedOutput); } +TEST_CASE("NeonBatchMatMulEndToEndFloat32Test") +{ + BatchMatMulEndToEnd<armnn::DataType::Float32>(neonDefaultBackends); +} + TEST_CASE("NeonConcatEndToEndDim0Test") { ConcatDim0EndToEnd<armnn::DataType::Float32>(neonDefaultBackends); |