6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit SubFixture(
const std::string & inputShape1,
18 const std::string & inputShape2,
19 const std::string & outputShape,
20 const std::string & activation=
"NONE")
25 "operator_codes": [ { "builtin_code": "SUB" } ], 29 "shape": )" + inputShape1 + R"(, 32 "name": "inputTensor1", 41 "shape": )" + inputShape2 + R"(, 44 "name": "inputTensor2", 53 "shape": )" + outputShape + R"( , 56 "name": "outputTensor", 72 "builtin_options_type": "SubOptions", 74 "fused_activation_function": )" + activation + R"( 76 "custom_options_format": "FLEXBUFFERS" 91 struct SimpleSubFixture : SubFixture
93 SimpleSubFixture() : SubFixture(
"[ 1, 4 ]",
100 RunTest<2, armnn::DataType::QAsymmU8>(
102 {{
"inputTensor1", { 4, 5, 6, 7 }},
103 {
"inputTensor2", { 3, 2, 1, 0 }}},
104 {{
"outputTensor", { 1, 3, 5, 7 }}});
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(SimpleSub, SimpleSubFixture)
BOOST_AUTO_TEST_SUITE_END()