6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit ArgMinMaxFixture(
const std::string& operatorCode,
18 const std::string& inputShape,
19 const std::string& outputShape,
20 const std::string& axisData)
25 "operator_codes": [ { "builtin_code": )" + operatorCode + R"( } ], 29 "shape": )" + inputShape + R"(, 32 "name": "inputTensor", 41 "shape": )" + outputShape + R"( , 44 "name": "outputTensor", 72 "custom_options_format": "FLEXBUFFERS" 79 { "data": )" + axisData + R"(, }, 88 struct SimpleArgMaxFixture :
public ArgMinMaxFixture
90 SimpleArgMaxFixture() : ArgMinMaxFixture(
"ARG_MAX",
98 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
100 {{
"inputTensor", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f } } },
101 {{
"outputTensor", { 3l } } });
104 struct ArgMaxFixture :
public ArgMinMaxFixture
106 ArgMaxFixture() : ArgMinMaxFixture(
"ARG_MAX",
114 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
116 {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
117 8.0f, 7.0f, 6.0f, 6.0f,
118 100.0f, 20.0f, 300.0f, 40.0f,
119 500.0f, 476.0f, 450.0f, 426.0f,
120 50.0f, 60.0f, 70.0f, 80.0f,
121 10.0f, 200.0f, 30.0f, 400.0f } } },
122 {{
"outputTensor", { 1, 2, 1, 2,
126 struct SimpleArgMinFixture :
public ArgMinMaxFixture
128 SimpleArgMinFixture() : ArgMinMaxFixture(
"ARG_MIN",
136 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
138 {{
"inputTensor", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f } } },
139 {{
"outputTensor", { 1l } } });
142 struct ArgMinFixture :
public ArgMinMaxFixture
144 ArgMinFixture() : ArgMinMaxFixture(
"ARG_MIN",
152 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
154 {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
155 8.0f, 7.0f, 6.0f, 6.0f,
156 100.0f, 20.0f, 300.0f, 40.0f,
157 500.0f, 476.0f, 450.0f, 426.0f,
158 50.0f, 60.0f, 70.0f, 80.0f,
159 10.0f, 200.0f, 30.0f, 400.0f } } },
160 {{
"outputTensor", { 0, 0, 0, 0,
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ParseSimpleArgMax, SimpleArgMaxFixture)
BOOST_AUTO_TEST_SUITE_END()
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)