#include <OptimizeInversePermutes.hpp>

Public Member Functions
void	Run (Graph &graph, InputSlot &connection) const

Protected Member Functions
	OptimizeInversePermutesImpl ()=default

	~OptimizeInversePermutesImpl ()=default

Detailed Description

Definition at line 16 of file OptimizeInversePermutes.hpp.

Constructor & Destructor Documentation

◆ OptimizeInversePermutesImpl()

OptimizeInversePermutesImpl ( )

protecteddefault

Referenced by OptimizeInversePermutesImpl::Run().

◆ ~OptimizeInversePermutesImpl()

~OptimizeInversePermutesImpl ( )

protecteddefault

Referenced by OptimizeInversePermutesImpl::Run().

Member Function Documentation

◆ Run()

void Run	(	Graph &	graph,
		InputSlot &	connection
	)		const

inline

Run for every connection between a base PermuteLayer and a child PermuteLayer. Bypasses both layers for that connection if one is the inverse of the other.

Definition at line 21 of file OptimizeInversePermutes.hpp.

References InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), OutputSlot::MoveAllConnections(), OptimizeInversePermutesImpl::OptimizeInversePermutesImpl(), and OptimizeInversePermutesImpl::~OptimizeInversePermutesImpl().

     {
         boost::ignore_unused(graph);
         Layer& base = connection.GetConnectedOutputSlot()->GetOwningLayer();
         auto child = boost::polymorphic_downcast<PermuteLayer*>(&connection.GetOwningLayer());
 
         if (child->IsInverse(*boost::polymorphic_downcast<PermuteLayer*>(&base)))
         {
             // Bypass both layers. Child will be removed as it's left unconnected.
             // Base layer will be removed if left unconnected.
             child->GetOutputSlot().MoveAllConnections(*base.GetInputSlot(0).GetConnectedOutputSlot());
         }
     }

The documentation for this class was generated from the following file:

src/armnn/optimizations/OptimizeInversePermutes.hpp