ArmNN
NotReleased
|
#include <MovePermuteUp.hpp>
Public Member Functions | |
void | Run (Graph &graph, InputSlot &connection) const |
Protected Member Functions | |
MovePermuteUpImpl ()=default | |
~MovePermuteUpImpl ()=default | |
Definition at line 15 of file MovePermuteUp.hpp.
|
protecteddefault |
Referenced by MovePermuteUpImpl::Run().
|
protecteddefault |
Referenced by MovePermuteUpImpl::Run().
Run for every connection between a base Layer (any) and a child PermuteLayer. 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 permute layer to the output of the base layer.
Definition at line 22 of file MovePermuteUp.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, MovePermuteUpImpl::MovePermuteUpImpl(), armnn::Multiplication, armnnUtils::Permuted(), OutputHandler::SetTensorInfo(), and MovePermuteUpImpl::~MovePermuteUpImpl().