From c42c38413c72cf7c31bb0353a2b836b7a2754f37 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Tue, 22 Jan 2019 14:18:11 +0000 Subject: 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 --- src/armnnTfLiteParser/test/Activations.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/armnnTfLiteParser/test/Activations.cpp') 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() -- cgit v1.2.1