13 explicit ReduceMaxFixture(
const std::string& inputShape,
14 const std::string& outputShape,
15 const std::string& axisShape,
16 const std::string& axisData)
21 "operator_codes": [ { "builtin_code": "REDUCE_MAX" } ], 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 SimpleReduceMaxFixture :
public ReduceMaxFixture
89 SimpleReduceMaxFixture() : ReduceMaxFixture(
"[ 1, 1, 2, 3 ]",
"[ 1, 1, 1, 3 ]",
"[ 1 ]",
"[ 2,0,0,0 ]") {}
94 RunTest<4, armnn::DataType::Float32, armnn::DataType::Float32>
95 (0, {{
"inputTensor", { 1001.0f, 11.0f, 1003.0f,
96 10.0f, 1002.0f, 12.0f } } },
97 {{
"outputTensor", { 1001.0f, 1002.0f, 1003.0f } } });
102 explicit ReduceMinFixture(
const std::string& inputShape,
103 const std::string& outputShape,
104 const std::string& axisShape,
105 const std::string& axisData)
110 "operator_codes": [ { "builtin_code": "REDUCE_MIN" } ], 114 "shape": )" + inputShape + R"(, 117 "name": "inputTensor", 126 "shape": )" + outputShape + R"( , 129 "name": "outputTensor", 138 "shape": )" + axisShape + R"( , 157 "builtin_options_type": "ReducerOptions", 161 "custom_options_format": "FLEXBUFFERS" 168 { "data": )" + axisData + R"(, }, 172 SetupSingleInputSingleOutput("inputTensor",
"outputTensor");
176 struct SimpleReduceMinFixture :
public ReduceMinFixture
178 SimpleReduceMinFixture() : ReduceMinFixture(
"[ 1, 1, 2, 3 ]",
"[ 1, 1, 1, 3 ]",
"[ 1 ]",
"[ 2, 0, 0, 0 ]") {}
183 RunTest<4, armnn::DataType::Float32, armnn::DataType::Float32>
184 (0, {{
"inputTensor", { 1001.0f, 11.0f, 1003.0f,
185 10.0f, 1002.0f, 12.0f } } },
186 {{
"outputTensor", { 10.0f, 11.0f, 12.0f } } });
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("TensorflowLiteParser_Reduce")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)