6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit PackFixture(
const std::string & inputShape,
18 const unsigned int numInputs,
19 const std::string & outputShape,
20 const std::string & axis)
25 "operator_codes": [ { "builtin_code": "PACK" } ], 29 for (
unsigned int i = 0; i < numInputs; ++i)
33 "shape": )" + inputShape + R"(, 35 "buffer": )" + std::to_string(i) + R"(, 36 "name": "inputTensor)" + std::to_string(i + 1) + R"(", 46 std::string inputIndexes; 47 for (
unsigned int i = 0; i < numInputs-1; ++i)
49 inputIndexes += std::to_string(i) + R
"(, )"; 51 inputIndexes += std::to_string(numInputs-1); 55 "shape": )" + outputShape + R"( , 57 "buffer": )" + std::to_string(numInputs) + R"(, 58 "name": "outputTensor", 67 "inputs": [ )" + inputIndexes + R"( ], 72 "inputs": [ )" + inputIndexes + R"( ], 74 "builtin_options_type": "PackOptions", 76 "axis": )" + axis + R"(, 77 "values_count": )" + std::to_string(numInputs) + R"( 79 "custom_options_format": "FLEXBUFFERS" 85 for (
unsigned int i = 0; i < numInputs-1; ++i)
98 struct SimplePackFixture : PackFixture
100 SimplePackFixture() : PackFixture(
"[ 3, 2, 3 ]",
108 RunTest<4, armnn::DataType::Float32>(
110 { {
"inputTensor1", { 1, 2, 3,
118 {
"inputTensor2", { 19, 20, 21,
126 { {
"outputTensor", { 1, 19,
BOOST_FIXTURE_TEST_CASE(ParsePack, SimplePackFixture)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)