6 #include "../TfLiteParser.hpp" 15 explicit ExpandDimsFixture(
const std::string& inputShape,
16 const std::string& outputShape,
17 const std::string& axis)
22 "operator_codes": [ { "builtin_code": "EXPAND_DIMS" } ], 26 "shape": )" + inputShape + R"(, 29 "name": "inputTensor", 38 "shape": )" + outputShape + R"( , 41 "name": "outputTensor", 53 "name": "expand_dims", 69 "custom_options_format": "FLEXBUFFERS" 76 { "data": )" + axis + R"(, }, 84 struct ExpandDimsFixture3dto4Daxis0 : ExpandDimsFixture
86 ExpandDimsFixture3dto4Daxis0() : ExpandDimsFixture(
"[ 2, 2, 1 ]",
"[ 1, 2, 2, 1 ]",
"[ 0, 0, 0, 0 ]") {}
91 RunTest<4, armnn::DataType::QAsymmU8>(0, {{
"inputTensor", { 1, 2, 3, 4 } } },
92 {{
"outputTensor", { 1, 2, 3, 4 } } });
95 struct ExpandDimsFixture3dto4Daxis3 : ExpandDimsFixture
97 ExpandDimsFixture3dto4Daxis3() : ExpandDimsFixture(
"[ 1, 2, 2 ]",
"[ 1, 2, 2, 1 ]",
"[ 3, 0, 0, 0 ]") {}
102 RunTest<4, armnn::DataType::QAsymmU8>(0, {{
"inputTensor", { 1, 2, 3, 4 } } },
103 {{
"outputTensor", { 1, 2, 3, 4 } } });
TEST_SUITE("TensorflowLiteParser_ExpandDims")
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)