diff options
Diffstat (limited to 'src/armnnTfLiteParser/test/Activations.cpp')
-rw-r--r-- | src/armnnTfLiteParser/test/Activations.cpp | 20 |
1 files changed, 11 insertions, 9 deletions
diff --git a/src/armnnTfLiteParser/test/Activations.cpp b/src/armnnTfLiteParser/test/Activations.cpp index f74c22d107..980edc4c5d 100644 --- a/src/armnnTfLiteParser/test/Activations.cpp +++ b/src/armnnTfLiteParser/test/Activations.cpp @@ -3,12 +3,13 @@ // SPDX-License-Identifier: MIT // -#include <boost/test/unit_test.hpp> #include "ParserFlatbuffersFixture.hpp" #include "../TfLiteParser.hpp" -BOOST_AUTO_TEST_SUITE(TensorflowLiteParser) +#include <doctest/doctest.h> +TEST_SUITE("TensorflowLiteParser_Activations") +{ struct ActivationFixture : ParserFlatbuffersFixture { @@ -68,7 +69,7 @@ struct ReLuFixture : ActivationFixture { ReLuFixture() : ActivationFixture("RELU", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseReLu, ReLuFixture) +TEST_CASE_FIXTURE(ReLuFixture, "ParseReLu") { RunTest<2, armnn::DataType::Float32>(0, { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f }, { 0.0f, 0.0f, 1.25f, 0.0f, 0.0f, 0.5f, 0.0f }); @@ -78,7 +79,7 @@ struct ReLu6Fixture : ActivationFixture { ReLu6Fixture() : ActivationFixture("RELU6", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseReLu6, ReLu6Fixture) +TEST_CASE_FIXTURE(ReLu6Fixture, "ParseReLu6") { RunTest<2, armnn::DataType::Float32>(0, { -1.0f, -0.5f, 7.25f, -3.0f, 0.0f, 0.5f, -0.75f }, { 0.0f, 0.0f, 6.0f, 0.0f, 0.0f, 0.5f, 0.0f }); @@ -88,7 +89,7 @@ struct SigmoidFixture : ActivationFixture { SigmoidFixture() : ActivationFixture("LOGISTIC", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseLogistic, SigmoidFixture) +TEST_CASE_FIXTURE(SigmoidFixture, "ParseLogistic") { 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 }); @@ -99,7 +100,7 @@ struct TanHFixture : ActivationFixture TanHFixture() : ActivationFixture("TANH", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseTanH, TanHFixture) +TEST_CASE_FIXTURE(TanHFixture, "ParseTanH") { RunTest<2, armnn::DataType::Float32>(0, { -0.1f, -0.2f, -0.3f, -0.4f, 0.1f, 0.2f, 0.3f }, @@ -111,7 +112,7 @@ struct EluFixture : ActivationFixture EluFixture() : ActivationFixture("ELU", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseElu, EluFixture) +TEST_CASE_FIXTURE(EluFixture, "ParseElu") { RunTest<2, armnn::DataType::Float32>(0, { -2.0f, -1.0f, -0.0f, 0.0f, 1.0f, 2.0f, 3.0f }, @@ -123,10 +124,11 @@ struct HardSwishFixture : ActivationFixture HardSwishFixture() : ActivationFixture("HARD_SWISH", "FLOAT32") {} }; -BOOST_FIXTURE_TEST_CASE(ParseHardSwish, HardSwishFixture) +TEST_CASE_FIXTURE(HardSwishFixture, "ParseHardSwish") { 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() + +} |