From 6331f91a4a1cb1ad16c569d98bb9ddf704788464 Mon Sep 17 00:00:00 2001
From: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Date: Mon, 3 Jun 2019 17:10:02 +0100
Subject: 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 | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(limited to 'src/backends/reference/test/RefLayerTests.cpp')

diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp
index b2f71a8920..f54a8d067c 100644
--- a/src/backends/reference/test/RefLayerTests.cpp
+++ b/src/backends/reference/test/RefLayerTests.cpp
@@ -624,11 +624,23 @@ BOOST_AUTO_TEST_CASE(DetectionPostProcessFastNmsFloat)
 }
 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
-- 
cgit v1.2.1