diff options
Diffstat (limited to 'src/backends/reference/test/RefLayerTests.cpp')
-rw-r--r-- | src/backends/reference/test/RefLayerTests.cpp | 16 |
1 files changed, 14 insertions, 2 deletions
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 |