#include <MoveTransposeUp.hpp>

Public Member Functions
void	Run (Graph &graph, InputSlot &connection) const
	Run for every connection between a base Layer (any) and a child TransposeLayer. More...

Protected Member Functions
	MoveTransposeUpImpl ()=default

	~MoveTransposeUpImpl ()=default

Detailed Description

Definition at line 16 of file MoveTransposeUp.hpp.

Constructor & Destructor Documentation

◆ MoveTransposeUpImpl()

MoveTransposeUpImpl ( )

protecteddefault

◆ ~MoveTransposeUpImpl()

~MoveTransposeUpImpl ( )

protecteddefault

Member Function Documentation

◆ Run()

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

inline

Run for every connection between a base Layer (any) and a child TransposeLayer.

If the type of the base layer allows it, it moves the permutation to the inputs of the base layer. I.e., adds equivalent permutations before the inputs of the base layer and moves the connections in the output of the child transpose layer to the output of the base layer.

Definition at line 23 of file MoveTransposeUp.hpp.

     {
         OutputSlot& baseOutput = *connection.GetConnectedOutputSlot();
  
         if (baseOutput.GetNumConnections() == 1U)
         {
             Layer& base = baseOutput.GetOwningLayer();
  
             if (CanMoveTransposeToInputs(base))
             {
                 auto transpose = PolymorphicDowncast<TransposeLayer*>(&connection.GetOwningLayer());
                 const PermutationVector& perm = transpose->GetPermutation();
  
                 // Inserts an equivalent transpose before every input of the base layer.
                 for (auto baseInput = base.BeginInputSlots(); baseInput != base.EndInputSlots(); ++baseInput)
                 {
                     // Inserts a new transpose layer.
                     const std::string name = std::string("moved_up-") + transpose->GetName();
                     TransposeLayer& permLayer = *graph.InsertNewLayer<TransposeLayer>(*baseInput, perm, name.c_str());
  
                     // Sets output tensor info for the new layer.
                     OutputSlot& parentOutput = *permLayer.GetInputSlot(0).GetConnectedOutputSlot();
                     const TensorInfo permOutInfo = armnnUtils::TransposeTensorShape(parentOutput.GetTensorInfo(), perm);
                     permLayer.GetOutputHandler().SetTensorInfo(permOutInfo);
                 }
  
                 // Bypasses transpose. It will be removed as it's left unconnected.
                 transpose->GetOutputSlot().MoveAllConnections(base.GetOutputSlot());
             }
         }
     }

References Layer::BeginInputSlots(), Layer::EndInputSlots(), InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), OutputSlot::GetNumConnections(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), Graph::InsertNewLayer(), OutputHandler::SetTensorInfo(), and armnnUtils::TransposeTensorShape().

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

src/armnn/optimizations/MoveTransposeUp.hpp