1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "../TestUtils.hpp"
7
8 #include <Optimizer.hpp>
9
10 #include <boost/test/unit_test.hpp>
11
12 using namespace armnn;
13
14 BOOST_AUTO_TEST_SUITE(Optimizer)
15 using namespace armnn::optimizations;
16
BOOST_AUTO_TEST_CASE(OptimizeInversePermutesTest)17 BOOST_AUTO_TEST_CASE(OptimizeInversePermutesTest)
18 {
19 armnn::Graph graph;
20
21 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
22
23 graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
24
25 // Inserts two permutes, one the inverse of the other.
26 graph.InsertNewLayer<armnn::PermuteLayer>(output->GetInputSlot(0), armnn::PermuteDescriptor({ 0, 2, 3, 1 }),
27 "perm0231");
28 graph.InsertNewLayer<armnn::PermuteLayer>(output->GetInputSlot(0), armnn::PermuteDescriptor({ 0, 3, 1, 2 }),
29 "perm0312");
30
31 BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
32 &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::PermuteLayer>,
33 &IsLayerOfType<armnn::OutputLayer>));
34
35 armnn::Optimizer::Pass(graph, armnn::MakeOptimizations(OptimizeInversePermutes()));
36
37 // The permutes are removed.
38 BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
39 &IsLayerOfType<armnn::OutputLayer>));
40 }
41
BOOST_AUTO_TEST_CASE(OptimizeInverseTransposesTest)42 BOOST_AUTO_TEST_CASE(OptimizeInverseTransposesTest)
43 {
44 armnn::Graph graph;
45
46 auto output = graph.AddLayer<armnn::OutputLayer>(0, "output");
47
48 graph.InsertNewLayer<armnn::InputLayer>(output->GetInputSlot(0), 0, "input");
49
50 // Inserts two permutes, one the inverse of the other.
51 graph.InsertNewLayer<armnn::TransposeLayer>(output->GetInputSlot(0),
52 armnn::TransposeDescriptor({ 0, 3, 1, 2 }),
53 "transpose0312");
54 graph.InsertNewLayer<armnn::TransposeLayer>(output->GetInputSlot(0),
55 armnn::TransposeDescriptor({ 0, 2, 3, 1 }),
56 "transpose0231");
57
58 BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
59 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::TransposeLayer>,
60 &IsLayerOfType<armnn::OutputLayer>));
61
62 armnn::Optimizer::Pass(graph, armnn::MakeOptimizations(OptimizeInverseTransposes()));
63
64 // The permutes are removed.
65 BOOST_TEST(CheckSequence(graph.cbegin(), graph.cend(), &IsLayerOfType<armnn::InputLayer>,
66 &IsLayerOfType<armnn::OutputLayer>));
67 }
68
69 BOOST_AUTO_TEST_SUITE_END()