20.02
|
#include <Serializer.hpp>
Public Member Functions | |
SerializerVisitor () | |
~SerializerVisitor () | |
flatbuffers::FlatBufferBuilder & | GetFlatBufferBuilder () |
std::vector< int > & | GetInputIds () |
std::vector< int > & | GetOutputIds () |
std::vector< flatbuffers::Offset< armnnSerializer::AnyLayer > > & | GetSerializedLayers () |
flatbuffers::Offset< armnnSerializer::FeatureCompatibilityVersions > | GetVersionTable () |
void | VisitAbsLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function an absolute layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitActivationLayer (const armnn::IConnectableLayer *layer, const armnn::ActivationDescriptor &descriptor, const char *name=nullptr) override |
Function that an activation layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitAdditionLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function that an addition layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitArgMinMaxLayer (const armnn::IConnectableLayer *layer, const armnn::ArgMinMaxDescriptor &argMinMaxDescriptor, const char *name=nullptr) override |
Function that an arg min max layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitBatchToSpaceNdLayer (const armnn::IConnectableLayer *layer, const armnn::BatchToSpaceNdDescriptor &descriptor, const char *name=nullptr) override |
Function that a batch to space ND layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitBatchNormalizationLayer (const armnn::IConnectableLayer *layer, const armnn::BatchNormalizationDescriptor &BatchNormalizationDescriptor, const armnn::ConstTensor &mean, const armnn::ConstTensor &variance, const armnn::ConstTensor &beta, const armnn::ConstTensor &gamma, const char *name=nullptr) override |
Function that a batch normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitComparisonLayer (const armnn::IConnectableLayer *layer, const armnn::ComparisonDescriptor &descriptor, const char *name=nullptr) override |
Function a Comparison layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitConcatLayer (const armnn::IConnectableLayer *layer, const armnn::ConcatDescriptor &concatDescriptor, const char *name=nullptr) override |
Function that a concat layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitConstantLayer (const armnn::IConnectableLayer *layer, const armnn::ConstTensor &input, const char *=nullptr) override |
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. More... | |
void | VisitConvolution2dLayer (const armnn::IConnectableLayer *layer, const armnn::Convolution2dDescriptor &descriptor, const armnn::ConstTensor &weights, const armnn::Optional< armnn::ConstTensor > &biases, const char *=nullptr) override |
Function that a 2D convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDepthToSpaceLayer (const armnn::IConnectableLayer *layer, const armnn::DepthToSpaceDescriptor &descriptor, const char *name=nullptr) override |
Function a depth to space layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDepthwiseConvolution2dLayer (const armnn::IConnectableLayer *layer, const armnn::DepthwiseConvolution2dDescriptor &descriptor, const armnn::ConstTensor &weights, const armnn::Optional< armnn::ConstTensor > &biases, const char *name=nullptr) override |
Function that a 2D depthwise convolution layer with biases should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDequantizeLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function that a Dequantize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDetectionPostProcessLayer (const armnn::IConnectableLayer *layer, const armnn::DetectionPostProcessDescriptor &descriptor, const armnn::ConstTensor &anchors, const char *name=nullptr) override |
Function that a Detection PostProcess layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDivisionLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a division layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitElementwiseUnaryLayer (const armnn::IConnectableLayer *layer, const armnn::ElementwiseUnaryDescriptor &descriptor, const char *name=nullptr) override |
Function a ElementwiseUnary layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitEqualLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function an Equal layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitFloorLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a floor layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitFullyConnectedLayer (const armnn::IConnectableLayer *layer, const armnn::FullyConnectedDescriptor &fullyConnectedDescriptor, const armnn::ConstTensor &weights, const armnn::Optional< armnn::ConstTensor > &biases, const char *name=nullptr) override |
Function that a fully connected layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitGatherLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a Gather layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitGreaterLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a Greater layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitInputLayer (const armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const char *name=nullptr) override |
Function that an InputLayer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitInstanceNormalizationLayer (const armnn::IConnectableLayer *layer, const armnn::InstanceNormalizationDescriptor &instanceNormalizationDescriptor, const char *name=nullptr) override |
Function that an instance normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitL2NormalizationLayer (const armnn::IConnectableLayer *layer, const armnn::L2NormalizationDescriptor &l2NormalizationDescriptor, const char *name=nullptr) override |
Function that an L2 normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitLogSoftmaxLayer (const armnn::IConnectableLayer *layer, const armnn::LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr) override |
Function that a log softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitLstmLayer (const armnn::IConnectableLayer *layer, const armnn::LstmDescriptor &descriptor, const armnn::LstmInputParams ¶ms, const char *name=nullptr) override |
Function an Lstm layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMeanLayer (const armnn::IConnectableLayer *layer, const armnn::MeanDescriptor &descriptor, const char *name) override |
Function a Mean layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMinimumLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a Minimum layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMaximumLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a Maximum layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMergeLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function that a merge layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMergerLayer (const armnn::IConnectableLayer *layer, const armnn::MergerDescriptor &mergerDescriptor, const char *name=nullptr) override |
Function that a merger layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMultiplicationLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function that a multiplication layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitOutputLayer (const armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const char *name=nullptr) override |
Function an output layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPadLayer (const armnn::IConnectableLayer *layer, const armnn::PadDescriptor &PadDescriptor, const char *name=nullptr) override |
Function a pad layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPermuteLayer (const armnn::IConnectableLayer *layer, const armnn::PermuteDescriptor &PermuteDescriptor, const char *name=nullptr) override |
Function that a permute layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPooling2dLayer (const armnn::IConnectableLayer *layer, const armnn::Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr) override |
Function that a pooling layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPreluLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function that a PReLU activation layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitQuantizeLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a quantize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitQuantizedLstmLayer (const armnn::IConnectableLayer *layer, const armnn::QuantizedLstmInputParams ¶ms, const char *name=nullptr) override |
Function a QuantizedLstm layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitReshapeLayer (const armnn::IConnectableLayer *layer, const armnn::ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr) override |
Function a reshape layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitResizeLayer (const armnn::IConnectableLayer *layer, const armnn::ResizeDescriptor &resizeDescriptor, const char *name=nullptr) override |
Function that a resize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitResizeBilinearLayer (const armnn::IConnectableLayer *layer, const armnn::ResizeBilinearDescriptor &resizeDescriptor, const char *name=nullptr) override |
Function that a resize bilinear layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitRsqrtLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a Reciprocal of square root layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSliceLayer (const armnn::IConnectableLayer *layer, const armnn::SliceDescriptor &sliceDescriptor, const char *name=nullptr) override |
Function that a slice layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSoftmaxLayer (const armnn::IConnectableLayer *layer, const armnn::SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr) override |
Function that a softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSpaceToBatchNdLayer (const armnn::IConnectableLayer *layer, const armnn::SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr) override |
Function a space to batch layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSpaceToDepthLayer (const armnn::IConnectableLayer *layer, const armnn::SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr) override |
Function a space to depth layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitNormalizationLayer (const armnn::IConnectableLayer *layer, const armnn::NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr) override |
Function that a normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSplitterLayer (const armnn::IConnectableLayer *layer, const armnn::ViewsDescriptor &viewsDescriptor, const char *name=nullptr) override |
Function that a splitter layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitStandInLayer (const armnn::IConnectableLayer *layer, const armnn::StandInDescriptor &standInDescriptor, const char *name=nullptr) override |
Function a StandInLayer should call back to when its Accept(ILaterVisitor&) function is invoked. More... | |
void | VisitStackLayer (const armnn::IConnectableLayer *layer, const armnn::StackDescriptor &stackDescriptor, const char *name=nullptr) override |
Function a stack layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitStridedSliceLayer (const armnn::IConnectableLayer *layer, const armnn::StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr) override |
Function a strided slice layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSubtractionLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a subtraction layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSwitchLayer (const armnn::IConnectableLayer *layer, const char *name=nullptr) override |
Function a switch layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeConvolution2dLayer (const armnn::IConnectableLayer *layer, const armnn::TransposeConvolution2dDescriptor &descriptor, const armnn::ConstTensor &weights, const armnn::Optional< armnn::ConstTensor > &biases, const char *=nullptr) override |
Function that a 2D transpose convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeLayer (const armnn::IConnectableLayer *layer, const armnn::TransposeDescriptor &descriptor, const char *name=nullptr) override |
Function that a transpose layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
Public Member Functions inherited from ILayerVisitor | |
virtual void | StartVisit () |
virtual void | FinishVisit () |
Additional Inherited Members | |
Protected Member Functions inherited from ILayerVisitor | |
ILayerVisitor () | |
virtual | ~ILayerVisitor () |
Definition at line 21 of file Serializer.hpp.
|
inline |
Definition at line 24 of file Serializer.hpp.
|
inline |
Definition at line 25 of file Serializer.hpp.
|
inline |
Definition at line 27 of file Serializer.hpp.
|
inline |
Definition at line 32 of file Serializer.hpp.
|
inline |
Definition at line 37 of file Serializer.hpp.
|
inline |
Definition at line 42 of file Serializer.hpp.
References anchors(), ARMNN_DEPRECATED_MSG, SerializerVisitor::GetVersionTable(), SerializerVisitor::VisitAbsLayer(), SerializerVisitor::VisitActivationLayer(), SerializerVisitor::VisitAdditionLayer(), SerializerVisitor::VisitArgMinMaxLayer(), SerializerVisitor::VisitBatchNormalizationLayer(), SerializerVisitor::VisitBatchToSpaceNdLayer(), SerializerVisitor::VisitComparisonLayer(), SerializerVisitor::VisitConcatLayer(), SerializerVisitor::VisitConstantLayer(), SerializerVisitor::VisitConvolution2dLayer(), SerializerVisitor::VisitDepthToSpaceLayer(), SerializerVisitor::VisitDepthwiseConvolution2dLayer(), SerializerVisitor::VisitDequantizeLayer(), SerializerVisitor::VisitDetectionPostProcessLayer(), SerializerVisitor::VisitDivisionLayer(), SerializerVisitor::VisitElementwiseUnaryLayer(), SerializerVisitor::VisitEqualLayer(), SerializerVisitor::VisitFloorLayer(), SerializerVisitor::VisitFullyConnectedLayer(), SerializerVisitor::VisitGatherLayer(), SerializerVisitor::VisitGreaterLayer(), SerializerVisitor::VisitInputLayer(), SerializerVisitor::VisitInstanceNormalizationLayer(), SerializerVisitor::VisitL2NormalizationLayer(), SerializerVisitor::VisitLogSoftmaxLayer(), SerializerVisitor::VisitLstmLayer(), SerializerVisitor::VisitMaximumLayer(), SerializerVisitor::VisitMeanLayer(), SerializerVisitor::VisitMergeLayer(), SerializerVisitor::VisitMergerLayer(), SerializerVisitor::VisitMinimumLayer(), SerializerVisitor::VisitMultiplicationLayer(), SerializerVisitor::VisitNormalizationLayer(), SerializerVisitor::VisitOutputLayer(), SerializerVisitor::VisitPadLayer(), SerializerVisitor::VisitPermuteLayer(), SerializerVisitor::VisitPooling2dLayer(), SerializerVisitor::VisitPreluLayer(), SerializerVisitor::VisitQuantizedLstmLayer(), SerializerVisitor::VisitQuantizeLayer(), SerializerVisitor::VisitReshapeLayer(), SerializerVisitor::VisitResizeBilinearLayer(), SerializerVisitor::VisitResizeLayer(), SerializerVisitor::VisitRsqrtLayer(), SerializerVisitor::VisitSliceLayer(), SerializerVisitor::VisitSoftmaxLayer(), SerializerVisitor::VisitSpaceToBatchNdLayer(), SerializerVisitor::VisitSpaceToDepthLayer(), SerializerVisitor::VisitSplitterLayer(), SerializerVisitor::VisitStackLayer(), SerializerVisitor::VisitStandInLayer(), SerializerVisitor::VisitStridedSliceLayer(), SerializerVisitor::VisitSubtractionLayer(), SerializerVisitor::VisitSwitchLayer(), SerializerVisitor::VisitTransposeConvolution2dLayer(), and SerializerVisitor::VisitTransposeLayer().
flatbuffers::Offset< armnnSerializer::FeatureCompatibilityVersions > GetVersionTable | ( | ) |
Definition at line 1483 of file Serializer.cpp.
References IOutputSlot::CalculateIndexOnOwner(), IInputSlot::GetConnection(), TensorInfo::GetDataType(), armnnSerializer::GetFlatBufferDataType(), IConnectableLayer::GetInputSlot(), TensorShape::GetNumDimensions(), IConnectableLayer::GetNumInputSlots(), IConnectableLayer::GetNumOutputSlots(), IConnectableLayer::GetOutputSlot(), IOutputSlot::GetOwningLayerGuid(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), and IOutputSlot::GetTensorInfo().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 129 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 139 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferActivationFunction(), armnn::IgnoreUnused(), ActivationDescriptor::m_A, ActivationDescriptor::m_B, and ActivationDescriptor::m_Function.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 164 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 179 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferArgMinMaxFunction(), armnn::IgnoreUnused(), ArgMinMaxDescriptor::m_Axis, and ArgMinMaxDescriptor::m_Function.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 232 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), BatchNormalizationDescriptor::m_DataLayout, and BatchNormalizationDescriptor::m_Eps.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 202 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), BatchToSpaceNdDescriptor::m_BlockShape, BatchToSpaceNdDescriptor::m_Crops, and BatchToSpaceNdDescriptor::m_DataLayout.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 263 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferComparisonOperation(), armnn::IgnoreUnused(), and ComparisonDescriptor::m_Operation.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 729 of file Serializer.cpp.
References OriginsDescriptor::GetConcatAxis(), OriginsDescriptor::GetNumDimensions(), OriginsDescriptor::GetNumViews(), OriginsDescriptor::GetViewOrigin(), and armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 279 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 300 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), OptionalBase::has_value(), armnn::IgnoreUnused(), Convolution2dDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_DataLayout, Convolution2dDescriptor::m_DilationX, Convolution2dDescriptor::m_DilationY, Convolution2dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadTop, Convolution2dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideY, and OptionalReferenceSwitch< IsReference, T >::value().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 341 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), SpaceToDepthDescriptor::m_BlockSize, and SpaceToDepthDescriptor::m_DataLayout.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 357 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), OptionalBase::has_value(), armnn::IgnoreUnused(), DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_DilationX, DepthwiseConvolution2dDescriptor::m_DilationY, DepthwiseConvolution2dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadTop, DepthwiseConvolution2dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideY, and OptionalReferenceSwitch< IsReference, T >::value().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 394 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 405 of file Serializer.cpp.
References armnn::IgnoreUnused(), DetectionPostProcessDescriptor::m_DetectionsPerClass, DetectionPostProcessDescriptor::m_MaxClassesPerDetection, DetectionPostProcessDescriptor::m_MaxDetections, DetectionPostProcessDescriptor::m_NmsIouThreshold, DetectionPostProcessDescriptor::m_NmsScoreThreshold, DetectionPostProcessDescriptor::m_NumClasses, DetectionPostProcessDescriptor::m_ScaleH, DetectionPostProcessDescriptor::m_ScaleW, DetectionPostProcessDescriptor::m_ScaleX, DetectionPostProcessDescriptor::m_ScaleY, and DetectionPostProcessDescriptor::m_UseRegularNms.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 436 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 446 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferUnaryOperation(), armnn::IgnoreUnused(), and ElementwiseUnaryDescriptor::m_Operation.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 461 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 471 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1002 of file Serializer.cpp.
References armnn::IgnoreUnused(), FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_TransposeWeightMatrix, and OptionalReferenceSwitch< IsReference, T >::value().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 481 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 491 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 87 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 501 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), InstanceNormalizationDescriptor::m_Beta, InstanceNormalizationDescriptor::m_DataLayout, InstanceNormalizationDescriptor::m_Eps, and InstanceNormalizationDescriptor::m_Gamma.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 521 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), L2NormalizationDescriptor::m_DataLayout, and L2NormalizationDescriptor::m_Eps.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 542 of file Serializer.cpp.
References armnn::IgnoreUnused(), SoftmaxDescriptor::m_Axis, and SoftmaxDescriptor::m_Beta.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 566 of file Serializer.cpp.
References armnn::IgnoreUnused(), LstmDescriptor::m_ActivationFunc, LstmInputParams::m_CellBias, LstmInputParams::m_CellLayerNormWeights, LstmInputParams::m_CellToForgetWeights, LstmInputParams::m_CellToInputWeights, LstmInputParams::m_CellToOutputWeights, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmInputParams::m_ForgetGateBias, LstmInputParams::m_ForgetLayerNormWeights, LstmInputParams::m_InputGateBias, LstmInputParams::m_InputLayerNormWeights, LstmInputParams::m_InputToCellWeights, LstmInputParams::m_InputToForgetWeights, LstmInputParams::m_InputToInputWeights, LstmInputParams::m_InputToOutputWeights, LstmDescriptor::m_LayerNormEnabled, LstmInputParams::m_OutputGateBias, LstmInputParams::m_OutputLayerNormWeights, LstmDescriptor::m_PeepholeEnabled, LstmInputParams::m_ProjectionBias, LstmDescriptor::m_ProjectionEnabled, LstmInputParams::m_ProjectionWeights, LstmInputParams::m_RecurrentToCellWeights, LstmInputParams::m_RecurrentToForgetWeights, LstmInputParams::m_RecurrentToInputWeights, and LstmInputParams::m_RecurrentToOutputWeights.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 674 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 684 of file Serializer.cpp.
References armnn::IgnoreUnused(), MeanDescriptor::m_Axis, and MeanDescriptor::m_KeepDims.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 712 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 722 of file Serializer.cpp.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 702 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 764 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1158 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnnSerializer::GetFlatBufferNormalizationAlgorithmChannel(), armnnSerializer::GetFlatBufferNormalizationAlgorithmMethod(), armnn::IgnoreUnused(), NormalizationDescriptor::m_Alpha, NormalizationDescriptor::m_Beta, NormalizationDescriptor::m_DataLayout, NormalizationDescriptor::m_K, NormalizationDescriptor::m_NormChannelType, NormalizationDescriptor::m_NormMethodType, and NormalizationDescriptor::m_NormSize.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 109 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 775 of file Serializer.cpp.
References armnn::IgnoreUnused(), PadDescriptor::m_PadList, and PadDescriptor::m_PadValue.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 801 of file Serializer.cpp.
References PermutationVector::GetSize(), armnn::IgnoreUnused(), and PermuteDescriptor::m_DimMappings.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 947 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnnSerializer::GetFlatBufferOutputShapeRounding(), armnnSerializer::GetFlatBufferPaddingMethod(), armnnSerializer::GetFlatBufferPoolingAlgorithm(), armnn::IgnoreUnused(), Pooling2dDescriptor::m_DataLayout, Pooling2dDescriptor::m_OutputShapeRounding, Pooling2dDescriptor::m_PadBottom, Pooling2dDescriptor::m_PaddingMethod, Pooling2dDescriptor::m_PadLeft, Pooling2dDescriptor::m_PadRight, Pooling2dDescriptor::m_PadTop, Pooling2dDescriptor::m_PoolHeight, Pooling2dDescriptor::m_PoolType, Pooling2dDescriptor::m_PoolWidth, Pooling2dDescriptor::m_StrideX, and Pooling2dDescriptor::m_StrideY.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 976 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1333 of file Serializer.cpp.
References armnn::Boolean, armnn::Float16, armnn::Float32, QuantizedLstmInputParams::GetCellBias(), TensorInfo::GetDataType(), armnnSerializer::GetFlatBufferConstTensorData(), armnnSerializer::GetFlatBufferDataType(), QuantizedLstmInputParams::GetForgetGateBias(), IConnectableLayer::GetGuid(), BaseTensor< MemoryType >::GetInfo(), QuantizedLstmInputParams::GetInputGateBias(), QuantizedLstmInputParams::GetInputToCellWeights(), QuantizedLstmInputParams::GetInputToForgetWeights(), QuantizedLstmInputParams::GetInputToInputWeights(), QuantizedLstmInputParams::GetInputToOutputWeights(), BaseTensor< MemoryType >::GetMemoryArea(), IConnectableLayer::GetName(), BaseTensor< MemoryType >::GetNumBytes(), TensorShape::GetNumDimensions(), QuantizedLstmInputParams::GetOutputGateBias(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), QuantizedLstmInputParams::GetRecurrentToCellWeights(), QuantizedLstmInputParams::GetRecurrentToForgetWeights(), QuantizedLstmInputParams::GetRecurrentToInputWeights(), QuantizedLstmInputParams::GetRecurrentToOutputWeights(), TensorInfo::GetShape(), armnn::IgnoreUnused(), armnn::QAsymmU8, armnn::QSymmS16, and armnn::Signed32.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 991 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 829 of file Serializer.cpp.
References TensorShape::GetNumDimensions(), armnn::IgnoreUnused(), and ReshapeDescriptor::m_TargetShape.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 855 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), ResizeBilinearDescriptor::m_DataLayout, ResizeBilinearDescriptor::m_TargetHeight, and ResizeBilinearDescriptor::m_TargetWidth.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 876 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnnSerializer::GetFlatBufferResizeMethod(), armnn::IgnoreUnused(), ResizeDescriptor::m_DataLayout, ResizeDescriptor::m_Method, ResizeDescriptor::m_TargetHeight, and ResizeDescriptor::m_TargetWidth.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 898 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 908 of file Serializer.cpp.
References armnn::IgnoreUnused(), SliceDescriptor::m_Begin, and SliceDescriptor::m_Size.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 925 of file Serializer.cpp.
References armnn::IgnoreUnused(), and SoftmaxDescriptor::m_Beta.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1041 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), SpaceToBatchNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_DataLayout, and SpaceToBatchNdDescriptor::m_PadList.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1072 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), SpaceToDepthDescriptor::m_BlockSize, and SpaceToDepthDescriptor::m_DataLayout.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1092 of file Serializer.cpp.
References OriginsDescriptor::GetConcatAxis(), OriginsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumDimensions(), OriginsDescriptor::GetNumViews(), ViewsDescriptor::GetNumViews(), ViewsDescriptor::GetOrigins(), ViewsDescriptor::GetViewOrigin(), ViewsDescriptor::GetViewSizes(), and armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1183 of file Serializer.cpp.
References TensorShape::GetNumDimensions(), armnn::IgnoreUnused(), StackDescriptor::m_Axis, StackDescriptor::m_InputShape, and StackDescriptor::m_NumInputs.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1206 of file Serializer.cpp.
References armnn::IgnoreUnused(), StandInDescriptor::m_NumInputs, and StandInDescriptor::m_NumOutputs.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1222 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), armnn::IgnoreUnused(), StridedSliceDescriptor::m_Begin, StridedSliceDescriptor::m_BeginMask, StridedSliceDescriptor::m_DataLayout, StridedSliceDescriptor::m_EllipsisMask, StridedSliceDescriptor::m_End, StridedSliceDescriptor::m_EndMask, StridedSliceDescriptor::m_NewAxisMask, StridedSliceDescriptor::m_ShrinkAxisMask, and StridedSliceDescriptor::m_Stride.
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1249 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1259 of file Serializer.cpp.
References armnn::IgnoreUnused().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
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 1269 of file Serializer.cpp.
References armnnSerializer::GetFlatBufferDataLayout(), OptionalBase::has_value(), armnn::IgnoreUnused(), TransposeConvolution2dDescriptor::m_BiasEnabled, TransposeConvolution2dDescriptor::m_DataLayout, TransposeConvolution2dDescriptor::m_PadBottom, TransposeConvolution2dDescriptor::m_PadLeft, TransposeConvolution2dDescriptor::m_PadRight, TransposeConvolution2dDescriptor::m_PadTop, TransposeConvolution2dDescriptor::m_StrideX, TransposeConvolution2dDescriptor::m_StrideY, and OptionalReferenceSwitch< IsReference, T >::value().
Referenced by SerializerVisitor::GetSerializedLayers().
|
overridevirtual |
Function that a transpose 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. |
transposeDescriptor | - TransposeDescriptor to configure the transpose. |
name | - Optional name for the layer. |
Implements ILayerVisitor.
Definition at line 1306 of file Serializer.cpp.
References PermutationVector::GetSize(), armnn::IgnoreUnused(), and TransposeDescriptor::m_DimMappings.
Referenced by SerializerVisitor::GetSerializedLayers().