diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/backendsCommon/test/EndToEndTestImpl.hpp | 36 | ||||
-rw-r--r-- | src/backends/cl/test/ClEndToEndTests.cpp | 5 | ||||
-rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 5 | ||||
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 5 |
4 files changed, 51 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp index 358f4e3fc2..4221f626da 100644 --- a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp @@ -766,4 +766,40 @@ inline void ExportOutputWithSeveralOutputSlotConnectionsTest(std::vector<Backend BOOST_TEST(found != std::string::npos); } +inline void StridedSliceInvalidSliceEndToEndTest(std::vector<BackendId> backends) +{ + using namespace armnn; + + // Create runtime in which test will run + IRuntime::CreationOptions options; + IRuntimePtr runtime(armnn::IRuntime::Create(options)); + + // build up the structure of the network + INetworkPtr net(INetwork::Create()); + + IConnectableLayer* input = net->AddInputLayer(0); + + // Configure a strided slice with a stride the same size as the input but with a ShrinkAxisMask on the first + // dim of the output to make it too small to hold the specified slice. + StridedSliceDescriptor descriptor; + descriptor.m_Begin = {0, 0}; + descriptor.m_End = {2, 3}; + descriptor.m_Stride = {1, 1}; + descriptor.m_BeginMask = 0; + descriptor.m_EndMask = 0; + descriptor.m_ShrinkAxisMask = 1; + IConnectableLayer* stridedSlice = net->AddStridedSliceLayer(descriptor); + + IConnectableLayer* output0 = net->AddOutputLayer(0); + + input->GetOutputSlot(0).Connect(stridedSlice->GetInputSlot(0)); + stridedSlice->GetOutputSlot(0).Connect(output0->GetInputSlot(0)); + + input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 2, 3 }, DataType::Float32)); + stridedSlice->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 3 }, DataType::Float32)); + + // Attempt to optimize the network and check that the correct exception is thrown + BOOST_CHECK_THROW(Optimize(*net, backends, runtime->GetDeviceSpec()), armnn::LayerValidationException); +} + } // anonymous namespace diff --git a/src/backends/cl/test/ClEndToEndTests.cpp b/src/backends/cl/test/ClEndToEndTests.cpp index 10abcc9fe6..05f9538d6f 100644 --- a/src/backends/cl/test/ClEndToEndTests.cpp +++ b/src/backends/cl/test/ClEndToEndTests.cpp @@ -126,6 +126,11 @@ BOOST_AUTO_TEST_CASE(DequantizeEndToEndOffsetTest) DequantizeEndToEndOffset<armnn::DataType::QAsymmU8>(defaultBackends); } +BOOST_AUTO_TEST_CASE(ClStridedSliceInvalidSliceEndToEndTest) +{ + StridedSliceInvalidSliceEndToEndTest(defaultBackends); +} + BOOST_AUTO_TEST_CASE(ClGreaterSimpleEndToEndTest) { const std::vector<uint8_t> expectedOutput({ 0, 0, 0, 0, 1, 1, 1, 1, diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index abded64915..081b8af60a 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -513,6 +513,11 @@ BOOST_AUTO_TEST_CASE(NeonArgMinAxis3TestQuantisedAsymm8) ArgMinAxis3EndToEnd<armnn::DataType::QAsymmU8>(defaultBackends); } +BOOST_AUTO_TEST_CASE(NeonStridedSliceInvalidSliceEndToEndTest) +{ + StridedSliceInvalidSliceEndToEndTest(defaultBackends); +} + BOOST_AUTO_TEST_CASE(NeonDetectionPostProcessRegularNmsTest, * boost::unit_test::disabled()) { std::vector<float> boxEncodings({ diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 54a68810f6..bdda12f392 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -1210,6 +1210,11 @@ BOOST_AUTO_TEST_CASE(RefExportOutputWithSeveralOutputSlotConnectionsTest) ExportOutputWithSeveralOutputSlotConnectionsTest(defaultBackends); } +BOOST_AUTO_TEST_CASE(RefStridedSliceInvalidSliceEndToEndTest) +{ + StridedSliceInvalidSliceEndToEndTest(defaultBackends); +} + #endif BOOST_AUTO_TEST_SUITE_END() |