21.08
|
#include <RedirectMembersToConstantInputs.hpp>
Public Member Functions | |
void | Run (Graph &graph, Layer &layer) const |
Search for layers with ConstantLayers as inputs. More... | |
Protected Member Functions | |
RedirectMembersToConstantInputsImpl ()=default | |
~RedirectMembersToConstantInputsImpl ()=default | |
Definition at line 18 of file RedirectMembersToConstantInputs.hpp.
|
protecteddefault |
Referenced by RedirectMembersToConstantInputsImpl::Run().
|
protecteddefault |
Referenced by RedirectMembersToConstantInputsImpl::Run().
Search for layers with ConstantLayers as inputs.
If the inputs are constant redirect the layers member variable for ConstTensors (e.g. m_weights) to the data stored in the ConstantLayer it is connected to.
Definition at line 23 of file RedirectMembersToConstantInputs.hpp.
References armnn::BatchNormalization, armnn::Constant, armnn::Convolution2d, armnn::DepthwiseConvolution2d, armnn::DetectionPostProcess, armnn::FullyConnected, OutputSlot::GetOwningLayer(), Layer::GetType(), armnn::IgnoreUnused(), armnn::Lstm, ConstantLayer::m_LayerOutput, RedirectMembersToConstantInputsImpl::RedirectMembersToConstantInputsImpl(), armnn::TransposeConvolution2d, and RedirectMembersToConstantInputsImpl::~RedirectMembersToConstantInputsImpl().