diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-02-14 13:41:49 +0000 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-02-15 16:10:20 +0000 |
commit | 8675b68e1ed73c56cd00c1eb3a9c9921a1898309 (patch) | |
tree | cc2d57afec4aedb1c6d9f5eac6b77a2e5ad9cd55 /src/armnn/test/QuantizerTest.cpp | |
parent | e870bef8c93a3c711362b3530aa49c353f048d01 (diff) | |
download | armnn-8675b68e1ed73c56cd00c1eb3a9c9921a1898309.tar.gz |
IVGCVSW-2616: static quantization of reshape
Change-Id: Iaefa145b8a218b3708c8a9208fdabd4984376ae9
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'src/armnn/test/QuantizerTest.cpp')
-rw-r--r-- | src/armnn/test/QuantizerTest.cpp | 31 |
1 files changed, 31 insertions, 0 deletions
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 |