diff options
| author | Jan Eilers <jan.eilers@arm.com> | 2020-07-28 14:00:06 +0100 |
|---|---|---|
| committer | Jan Eilers <jan.eilers@arm.com> | 2020-07-28 16:43:36 +0100 |
| commit | 2f746b3f346d3efa9071fc53592652425869d6b3 (patch) | |
| tree | 9985280d6b510733b58fc76c729ad0f8679a287e /src/armnnTfLiteParser/test | |
| parent | 8d23636d188838ec13c193949a1a5c7af1c3d8d8 (diff) | |
| download | armnn-2f746b3f346d3efa9071fc53592652425869d6b3.tar.gz | |
Github#433 Add HardSwish support to TfLiteParser
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: Ic476f8d80bba080ab459db9e6a59cbafd307d129
Diffstat (limited to 'src/armnnTfLiteParser/test')
| -rw-r--r-- | src/armnnTfLiteParser/test/Activations.cpp | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/Activations.cpp b/src/armnnTfLiteParser/test/Activations.cpp index e8153a28cb..e57477e620 100644 --- a/src/armnnTfLiteParser/test/Activations.cpp +++ b/src/armnnTfLiteParser/test/Activations.cpp @@ -105,4 +105,16 @@ BOOST_FIXTURE_TEST_CASE(ParseTanH, TanHFixture) { -0.1f, -0.2f, -0.3f, -0.4f, 0.1f, 0.2f, 0.3f }, { -0.09966799f, -0.19737528f, -0.29131261f, -0.379949f, 0.09966799f, 0.19737528f, 0.29131261f }); } + +struct HardSwishFixture : ActivationFixture +{ + HardSwishFixture() : ActivationFixture("HARD_SWISH", "FLOAT32") {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseHardSwish, HardSwishFixture) +{ + RunTest<2, armnn::DataType::Float32>(0, + { -4.0f, -3.0f, -2.9f, 1.2f, 2.2f, 3.0f, 4.0f }, + { -0.0f, -0.0f, -0.04833334f, 0.84f, 1.90666667f, 3.0f, 4.0f }); +} BOOST_AUTO_TEST_SUITE_END() |