13 explicit ArgMinMaxFixture(
const std::string& operatorCode,
14 const std::string& inputShape,
15 const std::string& outputShape,
16 const std::string& axisData)
21 "operator_codes": [ { "builtin_code": )" + operatorCode + R"( } ], 25 "shape": )" + inputShape + R"(, 28 "name": "inputTensor", 37 "shape": )" + outputShape + R"( , 40 "name": "outputTensor", 68 "custom_options_format": "FLEXBUFFERS" 75 { "data": )" + axisData + R"(, }, 84 struct SimpleArgMaxFixture :
public ArgMinMaxFixture
86 SimpleArgMaxFixture() : ArgMinMaxFixture(
"ARG_MAX",
94 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
96 {{
"inputTensor", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f } } },
97 {{
"outputTensor", { 3l } } });
100 struct ArgMaxFixture :
public ArgMinMaxFixture
102 ArgMaxFixture() : ArgMinMaxFixture(
"ARG_MAX",
110 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
112 {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
113 8.0f, 7.0f, 6.0f, 6.0f,
114 100.0f, 20.0f, 300.0f, 40.0f,
115 500.0f, 476.0f, 450.0f, 426.0f,
116 50.0f, 60.0f, 70.0f, 80.0f,
117 10.0f, 200.0f, 30.0f, 400.0f } } },
118 {{
"outputTensor", { 1, 2, 1, 2,
122 struct SimpleArgMinFixture :
public ArgMinMaxFixture
124 SimpleArgMinFixture() : ArgMinMaxFixture(
"ARG_MIN",
132 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
134 {{
"inputTensor", { 6.0f, 2.0f, 8.0f, 10.0f, 9.0f } } },
135 {{
"outputTensor", { 1l } } });
138 struct ArgMinFixture :
public ArgMinMaxFixture
140 ArgMinFixture() : ArgMinMaxFixture(
"ARG_MIN",
148 RunTest<3, armnn::DataType::Float32, armnn::DataType::Signed32>(
150 {{
"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f,
151 8.0f, 7.0f, 6.0f, 6.0f,
152 100.0f, 20.0f, 300.0f, 40.0f,
153 500.0f, 476.0f, 450.0f, 426.0f,
154 50.0f, 60.0f, 70.0f, 80.0f,
155 10.0f, 200.0f, 30.0f, 400.0f } } },
156 {{
"outputTensor", { 0, 0, 0, 0,
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)