Go to the source code of this file.
◆ TEST_SUITE()
TEST_SUITE |
( |
"Deserializer_Gather" |
| ) |
|
Definition at line 11 of file DeserializeGather.cpp.
References ParserFlatbuffersSerializeFixture::Setup(), and TEST_CASE_FIXTURE().
15 explicit GatherFixture(
const std::string& inputShape,
16 const std::string& indicesShape,
17 const std::string& input1Content,
18 const std::string& outputShape,
19 const std::string& axis,
20 const std::string dataType,
21 const std::string constDataType)
29 layer_type: "InputLayer", 35 layerName: "InputLayer", 39 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 44 dimensions: )" + inputShape + R"(, 45 dataType: )" + dataType + R"( 50 layer_type: "ConstantLayer", 54 layerName: "ConstantLayer", 55 layerType: "Constant", 59 dimensions: )" + indicesShape + R"(, 66 dimensions: )" + indicesShape + R"(, 67 dataType: )" + dataType + R"( 69 data_type: )" + constDataType + R"(, 71 data: )" + input1Content + R"(, 75 layer_type: "GatherLayer", 79 layerName: "GatherLayer", 84 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 88 connection: {sourceLayerIndex:1, outputSlotIndex:0 } 93 dimensions: )" + outputShape + R"(, 94 dataType: )" + dataType + R"( 102 layer_type: "OutputLayer", 108 layerName: "OutputLayer", 112 connection: {sourceLayerIndex:2, outputSlotIndex:0 }, 117 dimensions: )" + outputShape + R"(, 118 dataType: )" + dataType + R"( 124 weightsLayoutScheme: 1, 132 struct SimpleGatherFixtureFloat32 : GatherFixture
134 SimpleGatherFixtureFloat32() : GatherFixture(
"[ 3, 2, 3 ]",
"[ 2, 3 ]",
"[1, 2, 1, 2, 1, 0]",
135 "[ 2, 3, 2, 3 ]",
"0",
"Float32",
"IntData") {}
140 RunTest<4, armnn::DataType::Float32>(0,
141 {{
"InputLayer", { 1, 2, 3,
147 {{
"OutputLayer", { 7, 8, 9,
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")