6 #include "../TestUtils.hpp" 10 #include <doctest/doctest.h> 12 using namespace armnn;
18 TEST_CASE(
"OptimizeInversePermutesTest")
33 &IsLayerOfType<armnn::PermuteLayer>, &IsLayerOfType<armnn::PermuteLayer>,
34 &IsLayerOfType<armnn::OutputLayer>));
40 &IsLayerOfType<armnn::OutputLayer>));
43 TEST_CASE(
"OptimizeInverseTransposesTest")
60 &IsLayerOfType<armnn::TransposeLayer>, &IsLayerOfType<armnn::TransposeLayer>,
61 &IsLayerOfType<armnn::OutputLayer>));
67 &IsLayerOfType<armnn::OutputLayer>));
TEST_SUITE("TestConstTensorLayerVisitor")
OptimizeForConnection< PermuteLayer, PermuteLayer, OptimizeInversePermutesImpl< PermuteLayer > > OptimizeInversePermutes
Optimizer::Optimizations MakeOptimizations(Args &&... args)
OptimizeForConnection< TransposeLayer, TransposeLayer, OptimizeInversePermutesImpl< TransposeLayer > > OptimizeInverseTransposes
LayerT * AddLayer(Args &&... args)
Adds a new layer, of type LayerType, to the graph constructed with the arguments passed.
ConstIterator cbegin() const
Returns const iterator pointing to the beginning of the list. Lowercase for range-based for loops...
static void Pass(Graph &graph, const Optimizations &optimizations)
Copyright (c) 2021 ARM Limited and Contributors.
This layer represents a permutation operation.
A layer user-provided data can be bound to (e.g. inputs, outputs).
This layer represents a transpose operation.
bool CheckSequence(const armnn::Graph::ConstIterator first, const armnn::Graph::ConstIterator last)
A TransposeDescriptor for the TransposeLayer.
ConstIterator cend() const
Returns const iterator pointing to the end of the list. Lowercase for range-based for loops...
LayerT * InsertNewLayer(InputSlot &insertBefore, Args &&... args)
Inserts a new layer between the output slot currently connected to insertBefore and insertBefore itse...
A PermuteDescriptor for the PermuteLayer.