15 explicit StridedSliceFixture(
const std::string& inputShape,
16 const std::string& begin,
17 const std::string& end,
18 const std::string& stride,
19 const std::string& beginMask,
20 const std::string& endMask,
21 const std::string& shrinkAxisMask,
22 const std::string& ellipsisMask,
23 const std::string& newAxisMask,
24 const std::string& dataLayout,
25 const std::string& outputShape,
26 const std::string& dataType)
34 layer_type: "InputLayer", 40 layerName: "InputLayer", 44 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 49 dimensions: )" + inputShape + R"(, 50 dataType: )" + dataType + R"( 58 layer_type: "StridedSliceLayer", 62 layerName: "StridedSliceLayer", 63 layerType: "StridedSlice", 66 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 71 dimensions: )" + outputShape + R"(, 72 dataType: )" + dataType + R"( 77 begin: )" + begin + R"(, 79 stride: )" + stride + R"(, 80 beginMask: )" + beginMask + R"(, 81 endMask: )" + endMask + R"(, 82 shrinkAxisMask: )" + shrinkAxisMask + R"(, 83 ellipsisMask: )" + ellipsisMask + R"(, 84 newAxisMask: )" + newAxisMask + R"(, 85 dataLayout: )" + dataLayout + R"(, 90 layer_type: "OutputLayer", 96 layerName: "OutputLayer", 100 connection: {sourceLayerIndex:1, outputSlotIndex:0 }, 105 dimensions: )" + outputShape + R"(, 106 dataType: )" + dataType + R"( 120 struct SimpleStridedSliceFixture : StridedSliceFixture
122 SimpleStridedSliceFixture() : StridedSliceFixture(
"[ 3, 2, 3, 1 ]",
138 RunTest<4, armnn::DataType::Float32>(0,
140 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
141 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
142 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
145 1.0f, 1.0f, 5.0f, 5.0f
149 struct StridedSliceMaskFixture : StridedSliceFixture
151 StridedSliceMaskFixture() : StridedSliceFixture(
"[ 3, 2, 3, 1 ]",
167 RunTest<4, armnn::DataType::Float32>(0,
169 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
170 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
171 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
174 1.0f, 1.0f, 1.0f, 2.0f, 2.0f, 2.0f,
175 3.0f, 3.0f, 3.0f, 4.0f, 4.0f, 4.0f,
176 5.0f, 5.0f, 5.0f, 6.0f, 6.0f, 6.0f
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")