13 explicit PackFixture(
const std::string & inputShape,
14 const unsigned int numInputs,
15 const std::string & outputShape,
16 const std::string & axis)
21 "operator_codes": [ { "builtin_code": "PACK" } ], 25 for (
unsigned int i = 0; i < numInputs; ++i)
29 "shape": )" + inputShape + R"(, 31 "buffer": )" + std::to_string(i) + R"(, 32 "name": "inputTensor)" + std::to_string(i + 1) + R"(", 42 std::string inputIndexes; 43 for (
unsigned int i = 0; i < numInputs-1; ++i)
45 inputIndexes += std::to_string(i) + R
"(, )"; 47 inputIndexes += std::to_string(numInputs-1); 51 "shape": )" + outputShape + R"( , 53 "buffer": )" + std::to_string(numInputs) + R"(, 54 "name": "outputTensor", 63 "inputs": [ )" + inputIndexes + R"( ], 68 "inputs": [ )" + inputIndexes + R"( ], 70 "builtin_options_type": "PackOptions", 72 "axis": )" + axis + R"(, 73 "values_count": )" + std::to_string(numInputs) + R"( 75 "custom_options_format": "FLEXBUFFERS" 81 for (
unsigned int i = 0; i < numInputs-1; ++i)
94 struct SimplePackFixture : PackFixture
96 SimplePackFixture() : PackFixture(
"[ 3, 2, 3 ]",
104 RunTest<4, armnn::DataType::Float32>(
106 { {
"inputTensor1", { 1, 2, 3,
114 {
"inputTensor2", { 19, 20, 21,
122 { {
"outputTensor", { 1, 19,
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("TensorflowLiteParser_Pack")
void Setup(bool testDynamic=true)