ArmNN
NotReleased
|
#include <LayerWithParameters.hpp>
Public Types | |
using | DescriptorType = Parameters |
Public Member Functions | |
const Parameters & | GetParameters () const |
void | SerializeLayerParameters (ParameterStringifyFunction &fn) const override |
Public Member Functions inherited from Layer | |
Layer (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char *name) | |
Layer (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, DataLayout layout, const char *name) | |
const std::string & | GetNameStr () const |
const OutputHandler & | GetOutputHandler (unsigned int i=0) const |
OutputHandler & | GetOutputHandler (unsigned int i=0) |
const std::vector< InputSlot > & | GetInputSlots () const |
const std::vector< OutputSlot > & | GetOutputSlots () const |
std::vector< InputSlot >::iterator | BeginInputSlots () |
std::vector< InputSlot >::iterator | EndInputSlots () |
std::vector< OutputSlot >::iterator | BeginOutputSlots () |
std::vector< OutputSlot >::iterator | EndOutputSlots () |
bool | IsOutputUnconnected () |
void | ResetPriority () const |
LayerPriority | GetPriority () const |
LayerType | GetType () const |
DataType | GetDataType () const |
const BackendId & | GetBackendId () const |
void | SetBackendId (const BackendId &id) |
virtual std::unique_ptr< IWorkload > | CreateWorkload (const IWorkloadFactory &factory) const =0 |
virtual void | CreateTensorHandles (const TensorHandleFactoryRegistry ®istry, const IWorkloadFactory &factory, const bool IsMemoryManaged=true) |
virtual Layer * | Clone (Graph &graph) const =0 |
void | VerifyLayerConnections (unsigned int expectedConnections, const CheckLocation &location) const |
virtual void | ValidateTensorShapesFromInputs ()=0 |
std::vector< TensorShape > | InferOutputShapes (const std::vector< TensorShape > &inputShapes) const override |
virtual void | ReleaseConstantData () |
template<typename Op > | |
void | OperateOnConstantTensors (Op op) |
const char * | GetName () const override |
unsigned int | GetNumInputSlots () const override |
unsigned int | GetNumOutputSlots () const override |
const InputSlot & | GetInputSlot (unsigned int index) const override |
InputSlot & | GetInputSlot (unsigned int index) override |
const OutputSlot & | GetOutputSlot (unsigned int index=0) const override |
OutputSlot & | GetOutputSlot (unsigned int index=0) override |
void | SetGuid (LayerGuid guid) |
LayerGuid | GetGuid () const final |
void | AddRelatedLayerName (const std::string layerName) |
const std::list< std::string > & | GetRelatedLayerNames () |
virtual void | Reparent (Graph &dest, std::list< Layer *>::const_iterator iterator)=0 |
Public Member Functions inherited from IConnectableLayer | |
virtual void | Accept (ILayerVisitor &visitor) const =0 |
Protected Member Functions | |
LayerWithParameters (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const Parameters ¶m, const char *name) | |
~LayerWithParameters ()=default | |
template<typename QueueDescriptor > | |
WorkloadInfo | PrepInfoAndDesc (QueueDescriptor &descriptor) const |
Helper function to reduce duplication in *LayerCreateWorkload. More... | |
Protected Member Functions inherited from Layer | |
virtual | ~Layer ()=default |
template<typename QueueDescriptor > | |
void | CollectQueueDescriptorInputs (QueueDescriptor &descriptor, WorkloadInfo &info) const |
template<typename QueueDescriptor > | |
void | CollectQueueDescriptorOutputs (QueueDescriptor &descriptor, WorkloadInfo &info) const |
template<typename QueueDescriptor > | |
WorkloadInfo | PrepInfoAndDesc (QueueDescriptor &descriptor) const |
Helper function to reduce duplication in *LayerCreateWorkload. More... | |
template<typename LayerType , typename ... Params> | |
LayerType * | CloneBase (Graph &graph, Params &&... params) const |
virtual ConstantTensors | GetConstantTensorsByRef () |
Protected Member Functions inherited from IConnectableLayer | |
~IConnectableLayer () | |
Objects are not deletable via the handle. More... | |
Protected Attributes | |
Parameters | m_Param |
The parameters for the layer (not including tensor-valued weights etc.). More... | |
Protected Attributes inherited from Layer | |
std::vector< OutputHandler > | m_OutputHandlers |
Additional Inherited Members | |
Protected Types inherited from Layer | |
using | ConstantTensors = std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >> |
Definition at line 13 of file LayerWithParameters.hpp.
using DescriptorType = Parameters |
Definition at line 16 of file LayerWithParameters.hpp.
|
inlineprotected |
Definition at line 29 of file LayerWithParameters.hpp.
|
protecteddefault |
|
inline |
Definition at line 18 of file LayerWithParameters.hpp.
Referenced by BOOST_AUTO_TEST_CASE().
|
inlineprotected |
Helper function to reduce duplication in *LayerCreateWorkload.
Definition at line 43 of file LayerWithParameters.hpp.
|
inlineoverridevirtual |
Helper to serialize the layer parameters to string (currently used in DotSerializer and company).
Reimplemented from Layer.
Definition at line 22 of file LayerWithParameters.hpp.
Referenced by DepthwiseConvolution2dLayer::SerializeLayerParameters(), and Convolution2dLayer::SerializeLayerParameters().
|
protected |
The parameters for the layer (not including tensor-valued weights etc.).
Definition at line 50 of file LayerWithParameters.hpp.
Referenced by MeanLayer::Clone(), PadLayer::Clone(), ConcatLayer::CreateWorkload(), LayerWithParameters< ResizeDescriptor >::GetParameters(), LayerWithParameters< ResizeDescriptor >::PrepInfoAndDesc(), and LayerWithParameters< ResizeDescriptor >::SerializeLayerParameters().