diff options
author | Finn Williams <finn.williams@arm.com> | 2019-01-22 14:18:11 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-24 09:13:42 +0000 |
commit | c42c38413c72cf7c31bb0353a2b836b7a2754f37 (patch) | |
tree | 5000f60a900a45b97f771ff3813c477c73309d72 /src/armnnTfLiteParser/test | |
parent | db2b160bf9e7759d0157dfa57ee940290f5170e3 (diff) | |
download | armnn-c42c38413c72cf7c31bb0353a2b836b7a2754f37.tar.gz |
IVGCVSW-2430 Add logistic parser to tf-lite
* Added implementation and unit tests for
sigmoid function for tf-lite parser
* Refactored relu, relu6 and logisitc parser
to reduce code duplication
Change-Id: I00a2bd90bbc9144a2f84981f63b2cd1756b68a16
Diffstat (limited to 'src/armnnTfLiteParser/test')
-rw-r--r-- | src/armnnTfLiteParser/test/Activations.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/Activations.cpp b/src/armnnTfLiteParser/test/Activations.cpp index 534ae4cb73..dac16ce2c6 100644 --- a/src/armnnTfLiteParser/test/Activations.cpp +++ b/src/armnnTfLiteParser/test/Activations.cpp @@ -84,4 +84,13 @@ BOOST_FIXTURE_TEST_CASE(ParseReLu6, ReLu6Fixture) { 0.0f, 0.0f, 6.0f, 0.0f, 0.0f, 0.5f, 0.0f }); } +struct SigmoidFixture : ActivationFixture +{ + SigmoidFixture() : ActivationFixture("LOGISTIC", "FLOAT32") {} +}; +BOOST_FIXTURE_TEST_CASE(ParseLogistic, SigmoidFixture) +{ + RunTest<2, armnn::DataType::Float32>(0, { -1.0f, -0.5f, 4.0f, -4.0f, 0.0f, 0.5f, -0.75f }, + {0.268941f, 0.377541f, 0.982013f, 0.0179862f, 0.5f, 0.622459f, 0.320821f }); +} BOOST_AUTO_TEST_SUITE_END() |