IVGCVSW-2619 Support static quantization of StridedSlice

Change-Id: Iff0902ea2751b079563ce82fbbec6db5157c3392
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>

1 files changed, 30 insertions, 0 deletions

diff --git a/src/armnn/test/QuantizerTest.cpp b/src/armnn/test/QuantizerTest.cpp
index 21ac3e66ec..3edea9cc6e 100644
--- a/src/armnn/test/QuantizerTest.cpp
+++ b/src/armnn/test/QuantizerTest.cpp
@@ -1153,5 +1153,35 @@ BOOST_AUTO_TEST_CASE(QuantizeResizeBilinear)
     VisitLayersTopologically(quantizedNetwork.get(), validator);
 }
 
+class TestStridedSliceQuantization : public TestLeakyReLuActivationQuantization
+{
+public:
+    virtual void VisitStridedSliceLayer(const IConnectableLayer* layer,
+                                        const StridedSliceDescriptor& desc,
+                                        const char* name = nullptr)
+    {
+        CheckForwardedQuantizationSettings(layer);
+    }
+};
+
+BOOST_AUTO_TEST_CASE(QuantizeStridedSlice)
+{
+    auto network = INetwork::Create();
+
+    TensorShape shape{3U};
+    TensorInfo info(shape, DataType::Float32);
+
+    IConnectableLayer* activation = CreateStartOfLeakyReluNetwork(network.get(), info);
+
+    // Add the layer under test
+    StridedSliceDescriptor stridedSliceDesc;
+    IConnectableLayer* stridedSlice = network->AddStridedSliceLayer(stridedSliceDesc);
+    CompleteLeakyReluNetwork(network.get(), activation, stridedSlice, info);
+
+    auto quantizedNetwork = INetworkQuantizer::Create(network.get())->ExportNetwork();
+    TestStridedSliceQuantization validator;
+    VisitLayersTopologically(quantizedNetwork.get(), validator);
+}
+
 BOOST_AUTO_TEST_SUITE_END()
 } // namespace armnn