6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit QuantizeFixture(
const std::string & inputShape,
18 const std::string & outputShape,
19 const std::string & dataType)
24 "operator_codes": [ { "builtin_code": "QUANTIZE" } ], 28 "shape": )" + inputShape + R"(, 31 "name": "inputTensor", 40 "shape": )" + outputShape + R"( , 41 "type": )" + dataType + R"(, 43 "name": "outputTensor", 59 "builtin_options_type": "QuantizeOptions", 62 "custom_options_format": "FLEXBUFFERS" 76 struct SimpleQuantizeFixtureQAsymm8 : QuantizeFixture
78 SimpleQuantizeFixtureQAsymm8() : QuantizeFixture(
"[ 1, 6 ]",
85 RunTest<2, armnn::DataType::Float32, armnn::DataType::QuantisedAsymm8>(
87 {{
"inputTensor", { 0.0f, 1.5f, 7.5f, 150.0f, 300.0f, 382.5f }}},
88 {{
"outputTensor", { 0u, 1u, 5u, 100u, 200u, 255u }}});
91 struct SimpleQuantizeFixtureQSymm16 : QuantizeFixture
93 SimpleQuantizeFixtureQSymm16() : QuantizeFixture(
"[ 1, 6 ]",
100 RunTest<2, armnn::DataType::Float32, armnn::DataType::QuantisedSymm16>(
102 {{
"inputTensor", { 0.0f, 1.5f, 7.5f, 49150.5f, -1.5f,-49152.0f }}},
103 {{
"outputTensor", { 0, 1, 5, 32767, -1, -32768 }}});
106 struct SimpleQuantizeFixtureQSymmS8 : QuantizeFixture
108 SimpleQuantizeFixtureQSymmS8() : QuantizeFixture(
"[ 1, 6 ]",
115 RunTest<2, armnn::DataType::Float32, armnn::DataType::QSymmS8>(
117 {{
"inputTensor", { 0.0f, 1.5f, 7.5f, 190.5f, -192.0f, -1.5f }}},
118 {{
"outputTensor", { 0, 1, 5, 127, -128, -1 }}});
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(SimpleQuantizeQAsymm8, SimpleQuantizeFixtureQAsymm8)
BOOST_AUTO_TEST_SUITE_END()
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)