20.02
|
#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 | |
Definition at line 15 of file MoveTransposeUp.hpp.
|
protecteddefault |
Referenced by MoveTransposeUpImpl::Run().
|
protecteddefault |
Referenced by MoveTransposeUpImpl::Run().
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 22 of file MoveTransposeUp.hpp.
References armnn::Activation, armnn::Addition, Layer::BeginInputSlots(), Layer::EndInputSlots(), armnn::FakeQuantization, armnn::Floor, InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), OutputSlot::GetNumConnections(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), Layer::GetType(), Graph::InsertNewLayer(), armnn::MemCopy, MoveTransposeUpImpl::MoveTransposeUpImpl(), armnn::Multiplication, OutputHandler::SetTensorInfo(), armnnUtils::TransposeTensorShape(), and MoveTransposeUpImpl::~MoveTransposeUpImpl().