13 explicit GatherNdFixture(
const std::string& paramsShape,
14 const std::string& indicesShape,
15 const std::string& outputShape,
16 const std::string& dataType =
"FLOAT32",
17 const std::string& scale =
"1.0",
18 const std::string& offset =
"0")
23 "operator_codes": [ { "builtin_code": "GATHER_ND" } ], 27 "shape": )" + paramsShape + R"(, 28 "type": )" + dataType + R"(, 34 "scale": [ )" + scale + R"( ], 35 "zero_point": [ )" + offset + R"( ], 36 "details_type": "NONE", 37 "quantized_dimension": 0 40 "shape_signature": )" + paramsShape + R"( 43 "shape": )" + indicesShape + R"( , 48 "details_type": "NONE", 49 "quantized_dimension": 0 54 "shape": )" + outputShape + R"(, 55 "type": )" + dataType + R"(, 61 "scale": [ )" + scale + R"( ], 62 "zero_point": [ )" + offset + R"( ], 63 "details_type": "NONE", 64 "quantized_dimension": 0 67 "shape_signature": )" + outputShape + R"( 77 "builtin_options_type": "NONE", 78 "custom_options_format": "FLEXBUFFERS" 93 struct SimpleGatherNdFixture :
public GatherNdFixture
95 SimpleGatherNdFixture() : GatherNdFixture(
"[ 5, 2 ]",
"[ 3, 1 ]",
"[ 3, 2 ]" ) {}
100 RunTest<2, armnn::DataType::Float32, armnn::DataType::Signed32, armnn::DataType::Float32>
102 {{
"params", { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 }}},
103 {{
"indices", { 1, 3, 4 }}},
104 {{
"output", { 3, 4, 7, 8, 9, 10 }}});
107 struct GatherNdUint8Fixture :
public GatherNdFixture
109 GatherNdUint8Fixture() : GatherNdFixture(
"[ 8 ]",
"[ 3, 1 ]",
"[ 3 ]",
"UINT8") {}
114 RunTest<1, armnn::DataType::QAsymmU8, armnn::DataType::Signed32, armnn::DataType::QAsymmU8>
116 {{
"params", { 1, 2, 3, 4, 5, 6, 7, 8 }}},
117 {{
"indices", { 7, 6, 5 }}},
118 {{
"output", { 8, 7, 6 }}});
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
void Setup(bool testDynamic=true)