diff options
Diffstat (limited to 'src/backends/reference/test')
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 83f947ce09..9c10a68018 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -837,6 +837,37 @@ BOOST_AUTO_TEST_CASE(RefDetectionPostProcessFastNmsUint8Test) 1.0f, 1, 0.01f, 0, 0.5f, 0); } +// HardSwish +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestFloat32) +{ + HardSwishEndToEndTest<armnn::DataType::Float32>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestFloat16) +{ + HardSwishEndToEndTest<armnn::DataType::Float16>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestBFloat16) +{ +HardSwishEndToEndTest<armnn::DataType::BFloat16>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestQAsymmS8) +{ + HardSwishEndToEndTest<armnn::DataType::QAsymmS8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestQAsymmU8) +{ + HardSwishEndToEndTest<armnn::DataType::QAsymmU8>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefHardSwishEndToEndTestQSymmS16) +{ + HardSwishEndToEndTest<armnn::DataType::QSymmS16>(defaultBackends); +} + // LogSoftmax BOOST_AUTO_TEST_CASE(RefLogSoftmaxEndToEndTest) { |