ArmNN
NotReleased
|
#include <LayerVisitorBase.hpp>
Protected Member Functions | |
LayerVisitorBase () | |
virtual | ~LayerVisitorBase () |
Protected Member Functions inherited from ILayerVisitor | |
ILayerVisitor () | |
virtual | ~ILayerVisitor () |
Definition at line 25 of file LayerVisitorBase.hpp.
|
inlineprotected |
Definition at line 28 of file LayerVisitorBase.hpp.
|
inlineprotectedvirtual |
Definition at line 29 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function an absolute layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 32 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an activation layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
activationDescriptor | - ActivationDescriptor to configure the activation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 35 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an addition layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 39 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an arg min max layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
argMinMaxDescriptor | - ArgMinMaxDescriptor to configure the activation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 42 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a batch normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
mean | - Pre-calculated mean for each channel. |
variance | - Pre-calculated variance for each channel. |
beta | - Per-channel additive factor. |
gamma | - Per-channel multiplicative factor. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 46 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a batch to space ND layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
batchToSpaceNdDescriptor | - Description of the layer. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 54 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Comparison layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
comparisonDescriptor | - Description of the layer. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 58 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a concat layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
concatDescriptor | - ConcatDescriptor (synonym for OriginsDescriptor) to configure the concatenation process. Number of Views must be equal to the number of inputs, and their order must match - e.g. first view corresponds to the first input, second view to the second input, etc.... |
name | - Optional name for the layer. |
Reimplemented from ILayerVisitor.
Definition at line 62 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a layer with no inputs and a single output, which always corresponds to the passed in constant tensor should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
input | - Tensor to be provided as the only output of the layer. The layer will maintain its own copy of the tensor data, meaning the memory referenced by input can be freed or reused after this function is called. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 66 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a 2D convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
convolution2dDescriptor | - Description of the 2D convolution layer. |
weights | - Tensor for the weights data. |
biases | - Optional tensor for the bias data. If specified, must match the output tensor shape. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 70 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a depth to space layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
depthToSpaceDescriptor | - Parameters for the depth to space operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 76 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a 2D depthwise convolution layer with biases should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
convolution2dDescriptor | - Description of the 2D depthwise convolution layer. |
weights | - Tensor for the weights. Expected format: [channelMultiplier, inputChannels, height, width]. |
biases | - Optional tensor for the bias data. If specified, must match the output tensor shape. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 80 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a Dequantize layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 86 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a Detection PostProcess layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
descriptor | - Description of the Detection PostProcess layer. |
anchors | - Tensor for the anchors. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 89 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a division layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 94 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a ElementwiseUnary layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
elementwiseUnaryDescriptor | - Description of the layer. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 97 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function an Equal layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 101 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a floor layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 104 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a fully connected layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
fullyConnectedDescriptor | - Description of the fully connected layer. |
weights | - Tensor for the weights data. |
biases | - Optional tensor for the bias data. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 107 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Gather layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 113 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Greater layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 116 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an InputLayer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
id | - User generated id to uniquely identify a particular input. The same id needs to be specified when passing the inputs to the IRuntime::EnqueueWorkload() function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 119 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an instance normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
desc | - Parameters for the instance normalization operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 123 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that an L2 normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. Normalization is performed along dimension 1, but requires a 4d input.
layer | - pointer to the layer which is calling back to this visit function. |
desc | - Parameters for the L2 normalization operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 127 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a log softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
logSoftmaxDescriptor | - LogSoftmaxDescriptor to configure the log softmax. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 131 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function an Lstm layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
descriptor | - Parameters controlling the operation of the Lstm operation. |
params | - The weights and biases for the LSTM cell. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 135 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Maximum layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 140 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Mean layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
meanDescriptor | - Parameters for the mean operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 143 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a merge layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 147 of file LayerVisitorBase.hpp.
Referenced by BOOST_AUTO_TEST_CASE().
|
inlineoverridevirtual |
Function that a merger layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
mergerDescriptor | - MergerDescriptor (synonym for OriginsDescriptor) to configure the concatenation process. Number of Views must be equal to the number of inputs, and their order must match - e.g. first view corresponds to the first input, second view to the second input, etc.... |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 150 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Minimum layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 154 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a multiplication layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 157 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
normalizationDescriptor | - NormalizationDescriptor to configure the normalization. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 160 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function an output layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
id | - User generated id to uniquely identify a particular output. The same id needs to be specified when passing the outputs to the IRuntime::EnqueueWorkload() function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 164 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a pad layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
paddings | - n by 2 tensor, where n is the rank of the input tensor, such that paddings[i,0] indicates the amount of padding to add in front of dimension i, and paddings[i,1] indicates the amount of padding to add after the end of dimension i |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 168 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a permute layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
permuteDescriptor | - PermuteDescriptor to configure the permute. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 172 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a pooling layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
pooling2dDescriptor | - Pooling2dDescriptor to configure the pooling. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 176 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a PReLU activation layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 180 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a QuantizedLstm layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
params | - The weights and biases for the Quantized LSTM cell |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 186 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a quantize layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 183 of file LayerVisitorBase.hpp.
Referenced by BOOST_AUTO_TEST_CASE().
|
inlineoverridevirtual |
Function a reshape layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
reshapeDescriptor | - Parameters for the reshape operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 190 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a resize bilinear layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
resizeDesc | - Parameters for the resize operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 194 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a resize layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
resizeDescriptor | - Parameters for the resize operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 198 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a Reciprocal of square root layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 202 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a slice layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
sliceDescriptor | - SliceDescriptor to configure the slice operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 205 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
softmaxDescriptor | - SoftmaxDescriptor to configure the softmax. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 209 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a space to batch layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
spaceToBatchNdDescriptor | - Parameters for the space to batch operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 213 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a space to depth layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
spaceToDepthDescriptor | - Parameters for the space to depth operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 217 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a splitter layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
splitterDescriptor | - ViewsDescriptor to configure the splitting process. Number of Views must be equal to the number of outputs, and their order must match - e.g. first view corresponds to the first output, second view to the second output, etc.... |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 221 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a stack layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
stackDescriptor | - Parameters for the stack operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 225 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a StandInLayer should call back to when its Accept(ILaterVisitor&) function is invoked
layer | - pointer to the layer which is calling back to this visit function. |
standInDescriptor | - Parameters for the stand-in layer. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 229 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a strided slice layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
stridedSliceDescriptor | - Parameters for the strided slice operation. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 233 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a subtraction layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 237 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function a switch layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 240 of file LayerVisitorBase.hpp.
|
inlineoverridevirtual |
Function that a 2D transpose convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked.
layer | - pointer to the layer which is calling back to this visit function. |
descriptor | - Description of the 2D transpose convolution layer. |
weights | - Tensor for the weights data. |
biases | - Optional tensor for the bias data. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 243 of file LayerVisitorBase.hpp.