13 explicit SumFixture(
const std::string& inputShape,
14 const std::string& outputShape,
15 const std::string& axisShape,
16 const std::string& axisData)
21 "operator_codes": [ { "builtin_code": "SUM" } ], 25 "shape": )" + inputShape + R"(, 28 "name": "inputTensor", 37 "shape": )" + outputShape + R"( , 40 "name": "outputTensor", 49 "shape": )" + axisShape + R"( , 68 "builtin_options_type": "ReducerOptions", 72 "custom_options_format": "FLEXBUFFERS" 79 { "data": )" + axisData + R"(, }, 87 struct SimpleSumFixture :
public SumFixture
89 SimpleSumFixture() : SumFixture(
"[ 1, 3, 2, 4 ]",
"[ 1, 1, 1, 4 ]",
"[ 2 ]",
"[ 1, 0, 0, 0, 2, 0, 0, 0 ]") {}
94 RunTest<4, armnn::DataType::Float32, armnn::DataType::Float32>
95 (0, {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
96 5.0f, 6.0f, 7.0f, 8.0f,
98 10.0f, 20.0f, 30.0f, 40.0f,
99 50.0f, 60.0f, 70.0f, 80.0f,
101 100.0f, 200.0f, 300.0f, 400.0f,
102 500.0f, 600.0f, 700.0f, 800.0f } } },
103 {{
"outputTensor", { 666.0f, 888.0f, 1110.0f, 1332.0f } } });
TEST_SUITE("TensorflowLiteParser_Sum")
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)