From 7515d0730acc926316886a85401bcd36f640627e Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Wed, 16 Dec 2020 12:50:01 +0000 Subject: IVGCVSW-5614 Enable Hard Swish and Elu activations * Enabled Hard Swish and Elu in TfLiteDelegate * Added support for Elu in TfLiteParser Signed-off-by: Matthew Sloyan Change-Id: If09321b58568a98e14cabce610a1586556da041e --- src/armnnTfLiteParser/test/Activations.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/armnnTfLiteParser/test') diff --git a/src/armnnTfLiteParser/test/Activations.cpp b/src/armnnTfLiteParser/test/Activations.cpp index e57477e620..f74c22d107 100644 --- a/src/armnnTfLiteParser/test/Activations.cpp +++ b/src/armnnTfLiteParser/test/Activations.cpp @@ -106,6 +106,18 @@ BOOST_FIXTURE_TEST_CASE(ParseTanH, TanHFixture) { -0.09966799f, -0.19737528f, -0.29131261f, -0.379949f, 0.09966799f, 0.19737528f, 0.29131261f }); } +struct EluFixture : ActivationFixture +{ + EluFixture() : ActivationFixture("ELU", "FLOAT32") {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseElu, EluFixture) +{ + RunTest<2, armnn::DataType::Float32>(0, + { -2.0f, -1.0f, -0.0f, 0.0f, 1.0f, 2.0f, 3.0f }, + { -0.86466471676f, -0.63212055882f, -0.0f, 0.0f, 1.0f, 2.0f, 3.0f }); +} + struct HardSwishFixture : ActivationFixture { HardSwishFixture() : ActivationFixture("HARD_SWISH", "FLOAT32") {} -- cgit v1.2.1