#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>
Go to the source code of this file.
◆ TEST_SUITE()
TEST_SUITE |
( |
"TensorflowLiteParser_ExpandDims" |
| ) |
|
Definition at line 11 of file ExpandDims.cpp.
References ParserFlatbuffersFixture::SetupSingleInputSingleOutput(), and TEST_CASE_FIXTURE().
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_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)