From 8675b68e1ed73c56cd00c1eb3a9c9921a1898309 Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Thu, 14 Feb 2019 13:41:49 +0000 Subject: IVGCVSW-2616: static quantization of reshape Change-Id: Iaefa145b8a218b3708c8a9208fdabd4984376ae9 Signed-off-by: Nina Drozd --- src/armnn/test/QuantizerTest.cpp | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) (limited to 'src/armnn/test/QuantizerTest.cpp') diff --git a/src/armnn/test/QuantizerTest.cpp b/src/armnn/test/QuantizerTest.cpp index a46b443fa7..b73ac20457 100644 --- a/src/armnn/test/QuantizerTest.cpp +++ b/src/armnn/test/QuantizerTest.cpp @@ -1052,5 +1052,36 @@ BOOST_AUTO_TEST_CASE(QuantizeMerger) VisitLayersTopologically(quantizedNetwork.get(), validator); } +BOOST_AUTO_TEST_CASE(QuantizeReshape) +{ + class TestReshapeQuantization : public TestLeakyReLuActivationQuantization + { + public: + virtual void VisitReshapeLayer(const IConnectableLayer* layer, + const ReshapeDescriptor& reshapeDescriptor, + const char* name = nullptr) override + { + CheckForwardedQuantizationSettings(layer); + } + }; + + INetworkPtr network = INetwork::Create(); + + TensorShape shape{1U}; + TensorInfo info(shape, DataType::Float32); + + IConnectableLayer* activation = CreateStartOfLeakyReluNetwork(network.get(), info); + + // Add the layer under test + ReshapeDescriptor descriptor({1, 2, 3, 4}); + IConnectableLayer* reshape = network->AddReshapeLayer(descriptor); + + CompleteLeakyReluNetwork(network.get(), activation, reshape, info); + + auto quantizedNetwork = INetworkQuantizer::Create(network.get())->ExportNetwork(); + TestReshapeQuantization validator; + VisitLayersTopologically(quantizedNetwork.get(), validator); +} + BOOST_AUTO_TEST_SUITE_END() } // namespace armnn -- cgit v1.2.1