diff options
author | Rob Hughes <robert.hughes@arm.com> | 2019-09-24 09:34:53 +0100 |
---|---|---|
committer | Jim Flynn Arm <jim.flynn@arm.com> | 2019-09-25 02:44:48 +0000 |
commit | 95e73d77b9a79f7d350a39d85f07d09cd58422cc (patch) | |
tree | 8b713e8d29d433a28b29b8a9ce34052e8cccd08c /src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp | |
parent | 4833cea9036df428634cf64d8f1c4b54fc5da41f (diff) | |
download | armnn-95e73d77b9a79f7d350a39d85f07d09cd58422cc.tar.gz |
NNXSW-1826 Move tests for Optimization classes to separate files
This splits up the >1000 line OptimizerTests.cpp file.
Each Optimization class now has its own test file, all of which are in a
subfolder of tests called "optimizations".
The original OptimizerTests.cpp now contains mostly (completely?) tests
for validating output shapes, which perhaps should be moved to
test files specific to the layer types they are testing.
Change-Id: Icd1196cad8b720abcb156921aab1adbd4026756b
Signed-off-by: Rob Hughes <robert.hughes@arm.com>
Diffstat (limited to 'src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp')
-rw-r--r-- | src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp b/src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp new file mode 100644 index 0000000000..d16b8f7f77 --- /dev/null +++ b/src/armnn/test/optimizations/OptimizeConsecutiveReshapesTests.cpp @@ -0,0 +1,80 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include "../TestUtils.hpp" + +#include <Optimizer.hpp> + +#include <boost/test/unit_test.hpp> + +BOOST_AUTO_TEST_SUITE(Optimizer) +using namespace armnn::optimizations; + +BOOST_AUTO_TEST_CASE(OptimizeConsecutiveReshapesTest) +{ + armnn::Graph graph; + + const armnn::TensorInfo info0({ 1, 2, 3, 5 }, armnn::DataType::Float32); + + auto output = graph.AddLayer<armnn::OutputLayer>(0, "output"); + auto input = graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input"); + + input->GetOutputHandler().SetTensorInfo(info0); + + { + // Inserts two reshapes. + const armnn::TensorInfo info1({ 1, 30, 1, 1 }, armnn::DataType::Float32); + const armnn::TensorInfo info2({ 1, 2, 1, 15 }, armnn::DataType::Float32); + + std::string reshape1Name = "reshape1"; + std::string reshape2Name = "reshape2"; + + auto reshape1 = graph.InsertNewLayer<armnn::ReshapeLayer>( + output->GetInputSlot(0), armnn::ReshapeDescriptor{ info1.GetShape() }, reshape1Name.c_str()); + auto reshape2 = graph.InsertNewLayer<armnn::ReshapeLayer>( + output->GetInputSlot(0), armnn::ReshapeDescriptor{ info2.GetShape() }, reshape2Name.c_str()); + + reshape1->GetOutputHandler().SetTensorInfo(info1); + reshape2->GetOutputHandler().SetTensorInfo(info2); + + BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>, + &IsLayerOfType<armnn::ReshapeLayer>, &IsLayerOfType<armnn::ReshapeLayer>, + &IsLayerOfType<armnn::OutputLayer>)); + + armnn::Optimizer::Pass(graph, armnn::MakeOptimizations(OptimizeConsecutiveReshapes())); + + auto checkReshape = [&info2](const armnn::Layer* const layer) -> bool { + const auto reshapeLayer = static_cast<const armnn::ReshapeLayer*>(layer); + return IsLayerOfType<armnn::ReshapeLayer>(layer) && + (reshapeLayer->GetParameters().m_TargetShape == info2.GetShape()) && + (reshapeLayer->GetOutputHandler().GetTensorInfo().GetShape() == info2.GetShape()); + }; + + // The two reshapes are replaced by a single equivalent reshape. + BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>, checkReshape, + &IsLayerOfType<armnn::OutputLayer>)); + + // Check the new reshape layer has the other two reshapes as related layers + std::list<std::string> testRelatedLayers = { reshape2Name, reshape1Name }; + + BOOST_TEST(CheckRelatedLayers<armnn::ReshapeLayer>(graph, testRelatedLayers)); + } + + { + // Inserts a reshape to the input shape. + auto reshapeToIn = graph.InsertNewLayer<armnn::ReshapeLayer>( + output->GetInputSlot(0), armnn::ReshapeDescriptor{ info0.GetShape() }, "reshapeToIn"); + + reshapeToIn->GetOutputHandler().SetTensorInfo(info0); + + armnn::Optimizer::Pass(graph, armnn::MakeOptimizations(OptimizeConsecutiveReshapes())); + + // The two reshapes are removed. + BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>, + &IsLayerOfType<armnn::OutputLayer>)); + } +} + +BOOST_AUTO_TEST_SUITE_END()
\ No newline at end of file |