6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 14 explicit TransposeConvFixture(
const std::string& inputShape,
15 const std::string& outputShape,
16 const std::string& filterShape,
17 const std::string& filterData,
18 const std::string& strideX,
19 const std::string& strideY,
20 const std::string& dataType)
25 "operator_codes": [ { "builtin_code": "TRANSPOSE_CONV" } ], 32 "name": "outputShapeTensor", 41 "shape": )" + filterShape + R"(, 42 "type": ")" + dataType + R"(", 44 "name": "filterTensor", 53 "shape": )" + inputShape + R"(, 54 "type": ")" + dataType + R"(", 56 "name": "inputTensor", 65 "shape": )" + outputShape + R"(, 66 "type": ")" + dataType + R"(", 68 "name": "outputTensor", 82 "inputs": [ 0, 1, 2 ], 84 "builtin_options_type": "TransposeConvOptions", 87 "stride_w": )" + strideX + R"(, 88 "stride_h": )" + strideY + R"( 90 "custom_options_format": "FLEXBUFFERS" 95 { "data": )" + outputShape + R"( }, 96 { "data": )" + filterData + R"( }, 106 struct SimpleTransposeConvFixture : TransposeConvFixture
108 SimpleTransposeConvFixture()
109 : TransposeConvFixture(
"[ 1, 2, 2, 1 ]",
121 RunTest<4, armnn::DataType::QAsymmU8>(
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ParseSimpleTransposeConv, SimpleTransposeConvFixture)
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)