commit 6331f91a4a1cb1ad16c569d98bb9ddf704788464
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Jun 03 17:10:02 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Wed Jun 05 15:06:39 2019 +0000
tree 338cce081966bfb42f635b6febd68642d492b9f8
parent 18f2d1ccf9e743e61ed3733ae5a38f796a759db8

IVGCVSW-2971 Support QSymm16 for DetectionPostProcess workloads

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

src/backends/reference/test/RefLayerTests.cpp

diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp
index b2f71a8..f54a8d0 100644
--- a/src/backends/reference/test/RefLayerTests.cpp
+++ b/src/backends/reference/test/RefLayerTests.cpp
@@ -624,11 +624,23 @@
 }
 BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsUint8)
 {
-    DetectionPostProcessRegularNmsUint8Test<armnn::RefWorkloadFactory>();
+    DetectionPostProcessRegularNmsQuantizedTest<
+        armnn::RefWorkloadFactory, armnn::DataType::QuantisedAsymm8>();
 }
 BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsUint8)
 {
-    DetectionPostProcessFastNmsUint8Test<armnn::RefWorkloadFactory>();
+    DetectionPostProcessRegularNmsQuantizedTest<
+        armnn::RefWorkloadFactory, armnn::DataType::QuantisedAsymm8>();
+}
+BOOST_AUTO_TEST_CASE(DetectionPostProcessRegularNmsInt16)
+{
+    DetectionPostProcessRegularNmsQuantizedTest<
+        armnn::RefWorkloadFactory, armnn::DataType::QuantisedSymm16>();
+}
+BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsInt16)
+{
+    DetectionPostProcessFastNmsQuantizedTest<
+        armnn::RefWorkloadFactory, armnn::DataType::QuantisedSymm16>();
 }
 
 // Dequantize