diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-02-18 17:14:10 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-02-18 17:14:10 +0000 |
commit | a6b504a8925174739f5a064cf77d1563cca38708 (patch) | |
tree | 518cb4dadd9ad9c6b8b351d961865a89daa8d127 /src/armnn/test/QuantizerTest.cpp | |
parent | fc413c0c977e6c9680a2aa6546e977be0a2efdb9 (diff) | |
download | armnn-a6b504a8925174739f5a064cf77d1563cca38708.tar.gz |
IVGCVSW-2619 Support static quantization of StridedSlice
Change-Id: Iff0902ea2751b079563ce82fbbec6db5157c3392
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/armnn/test/QuantizerTest.cpp')
-rw-r--r-- | src/armnn/test/QuantizerTest.cpp | 30 |
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 |