15 explicit GatherNdFixture(
const std::string& paramsShape,
16 const std::string& indicesShape,
17 const std::string& outputShape,
18 const std::string& indicesData,
19 const std::string dataType,
20 const std::string constDataType)
28 layer_type: "InputLayer", 34 layerName: "InputLayer", 38 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 43 dimensions: )" + paramsShape + R"(, 44 dataType: )" + dataType + R"( 49 layer_type: "ConstantLayer", 53 layerName: "ConstantLayer", 54 layerType: "Constant", 58 dimensions: )" + indicesShape + R"(, 65 dimensions: )" + indicesShape + R"(, 66 dataType: )" + dataType + R"( 68 data_type: )" + constDataType + R"(, 70 data: )" + indicesData + R"(, 74 layer_type: "GatherNdLayer", 78 layerName: "GatherNdLayer", 79 layerType: "GatherNd", 83 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 87 connection: {sourceLayerIndex:1, outputSlotIndex:0 } 92 dimensions: )" + outputShape + R"(, 93 dataType: )" + dataType + R"( 98 layer_type: "OutputLayer", 104 layerName: "OutputLayer", 108 connection: {sourceLayerIndex:2, outputSlotIndex:0 }, 113 dimensions: )" + outputShape + R"(, 114 dataType: )" + dataType + R"( 120 weightsLayoutScheme: 1, 128 struct SimpleGatherNdFixtureFloat32 : GatherNdFixture
130 SimpleGatherNdFixtureFloat32() : GatherNdFixture(
"[ 6, 3 ]",
"[ 3, 1 ]",
"[ 3, 3 ]",
131 "[ 5, 1, 0 ]",
"Float32",
"IntData") {}
136 RunTest<4, armnn::DataType::Float32>(0,
137 {{
"InputLayer", { 1, 2, 3,
143 {{
"OutputLayer", { 16, 17, 18,
TEST_SUITE("Deserializer_GatherNd")
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")