commit 7599aa47f29531ebcbb3badea73ec181616bb29b
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Aug 20 15:30:04 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Aug 21 10:22:10 2019 +0000
tree 4d8b91f118e7bc12cdfd9d1501ebaab29b4bc451
parent 675fa4f1aac534dbcf73b9f2e9405d832c2e9d8c

IVGCVSW-3661 Add multi-channel TransposeConvolution2d unit tests to NEON backend

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I9d19ae4559a98b9e740c1d4d05add0c5788093d6

src/backends/neon/test/NeonLayerTests.cpp

diff --git a/src/backends/neon/test/NeonLayerTests.cpp b/src/backends/neon/test/NeonLayerTests.cpp
index a701362..e795174 100644
--- a/src/backends/neon/test/NeonLayerTests.cpp
+++ b/src/backends/neon/test/NeonLayerTests.cpp
@@ -835,6 +835,19 @@
                      true,
                      DataLayout::NHWC)
 
+ARMNN_AUTO_TEST_CASE(MultiChannelTransposeConvolution2dFloatNchw,
+                     MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
+                     DataLayout::NCHW)
+ARMNN_AUTO_TEST_CASE(MultiChannelTransposeConvolution2dFloatNhwc,
+                     MultiChannelTransposeConvolution2dTest<DataType::Float32, DataType::Float32>,
+                     DataLayout::NHWC)
+ARMNN_AUTO_TEST_CASE(MultiChannelTransposeConvolution2dUint8Nchw,
+                     MultiChannelTransposeConvolution2dTest<DataType::QuantisedAsymm8, DataType::Signed32>,
+                     DataLayout::NCHW)
+ARMNN_AUTO_TEST_CASE(MultiChannelTransposeConvolution2dUint8Nhwc,
+                     MultiChannelTransposeConvolution2dTest<DataType::QuantisedAsymm8, DataType::Signed32>,
+                     DataLayout::NHWC)
+
 // ============================================================================
 // COMPARE tests