#include <TransposeAsReshape.hpp>

Public Member Functions
void	Run (Graph &graph, TransposeLayer &transpose) const
	Run for every TransposeLayer. Replaces it with a ReshapeLayer if they are equivalent. More...

Protected Member Functions
	TransposeAsReshapeImpl ()=default

	~TransposeAsReshapeImpl ()=default

Detailed Description

Definition at line 14 of file TransposeAsReshape.hpp.

Constructor & Destructor Documentation

◆ TransposeAsReshapeImpl()

TransposeAsReshapeImpl ( )

protecteddefault

Referenced by TransposeAsReshapeImpl::Run().

◆ ~TransposeAsReshapeImpl()

~TransposeAsReshapeImpl ( )

protecteddefault

Referenced by TransposeAsReshapeImpl::Run().

Member Function Documentation

◆ Run()

void Run	(	Graph &	graph,
		TransposeLayer &	transpose
	)		const

inline

Run for every TransposeLayer. Replaces it with a ReshapeLayer if they are equivalent.

Definition at line 18 of file TransposeAsReshape.hpp.

References Layer::GetInputSlot(), Layer::GetName(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), TransposeLayer::GetPermutation(), TensorInfo::GetShape(), PermutationVector::GetSize(), OutputHandler::GetTensorInfo(), Graph::InsertNewLayer(), OutputSlot::MoveAllConnections(), TransposeAsReshapeImpl::TransposeAsReshapeImpl(), and TransposeAsReshapeImpl::~TransposeAsReshapeImpl().

     {
         if (IsReshape(transpose))
         {
             const TensorInfo& outInfo = transpose.GetOutputHandler().GetTensorInfo();
 
             const std::string name = std::string("as_reshape-") + transpose.GetName();
             const ReshapeDescriptor descriptor{outInfo.GetShape()};
             // Inserts NewLayer so layers don't need to be re-sorted.
             auto reshape = graph.InsertNewLayer<ReshapeLayer>(transpose.GetInputSlot(0), descriptor, name.c_str());
 
             // Bypass transpose. It will be deleted since it's left unconnected.
             transpose.GetOutputSlot().MoveAllConnections(reshape->GetOutputSlot());
         }
     }

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

src/armnn/optimizations/TransposeAsReshape.hpp