20.02
|
Visitor object for quantizing layers in a network. More...
#include <QuantizerVisitor.hpp>
Public Member Functions | |
QuantizerVisitor (const RangeTracker &rangeTracker, const IQuantizationScheme *quantizationScheme, bool preserveType=false) | |
~QuantizerVisitor ()=default | |
void | VisitAbsLayer (const IConnectableLayer *layer, const char *name=nullptr) override |
Functions to quantize the individual layers, overridden from ILayerVisitor. More... | |
void | VisitActivationLayer (const IConnectableLayer *layer, const ActivationDescriptor &activationDescriptor, 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 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 IConnectableLayer *layer, const 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 | VisitBatchNormalizationLayer (const IConnectableLayer *layer, const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const 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 | VisitBatchToSpaceNdLayer (const IConnectableLayer *layer, const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, 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 | VisitComparisonLayer (const IConnectableLayer *layer, const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr) override |
Function a Comparison layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitConcatLayer (const IConnectableLayer *layer, const OriginsDescriptor &originsDescriptor, 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 IConnectableLayer *layer, const ConstTensor &input, const char *name=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 IConnectableLayer *layer, const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Function that a 2D convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDepthToSpaceLayer (const IConnectableLayer *layer, const DepthToSpaceDescriptor &depthToSpaceDescriptor, 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 IConnectableLayer *layer, const DepthwiseConvolution2dDescriptor &desc, const ConstTensor &weights, const Optional< 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 | VisitElementwiseUnaryLayer (const IConnectableLayer *layer, const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr) override |
Function a ElementwiseUnary layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitFullyConnectedLayer (const IConnectableLayer *layer, const FullyConnectedDescriptor &desc, const ConstTensor &weights, const Optional< 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 | VisitInputLayer (const IConnectableLayer *layer, 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 IConnectableLayer *layer, const 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 | VisitLogSoftmaxLayer (const IConnectableLayer *layer, const 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 | VisitMeanLayer (const IConnectableLayer *layer, const MeanDescriptor &meanDescriptor, const char *name=nullptr) override |
Function a Mean layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMultiplicationLayer (const 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 | VisitNormalizationLayer (const IConnectableLayer *layer, const 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 | VisitOutputLayer (const IConnectableLayer *layer, 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 IConnectableLayer *, const 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 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 | VisitReshapeLayer (const IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const ResizeBilinearDescriptor &resizeDesc, 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 IConnectableLayer *, 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 | VisitSplitterLayer (const IConnectableLayer *layer, const SplitterDescriptor &splitterDescriptor, const char *name=nullptr) override |
Function that a splitter layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitStackLayer (const IConnectableLayer *layer, const 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 IConnectableLayer *layer, const 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 IConnectableLayer *layer, const char *name=nullptr) override |
Function a subtraction layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeConvolution2dLayer (const IConnectableLayer *layer, const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Function that a 2D transpose convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeLayer (const IConnectableLayer *layer, const TransposeDescriptor &descriptor, const char *name=nullptr) override |
Function that a transpose layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
INetworkPtr | RetrieveFinalNetwork () |
Extract the quantized network. More... | |
Public Member Functions inherited from LayerVisitorBase< VisitorThrowingPolicy > | |
void | VisitAbsLayer (const IConnectableLayer *, const char *) override |
Function an absolute layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitActivationLayer (const IConnectableLayer *, const ActivationDescriptor &, const char *) override |
Function that an activation layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitAdditionLayer (const IConnectableLayer *, const char *) override |
Function that an addition layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitArgMinMaxLayer (const IConnectableLayer *, const ArgMinMaxDescriptor &, const char *) override |
Function that an arg min max layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitBatchNormalizationLayer (const IConnectableLayer *, const BatchNormalizationDescriptor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const char *) override |
Function that a batch normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitBatchToSpaceNdLayer (const IConnectableLayer *, const BatchToSpaceNdDescriptor &, const char *) override |
Function that a batch to space ND layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitComparisonLayer (const IConnectableLayer *, const ComparisonDescriptor &, const char *) override |
Function a Comparison layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitConcatLayer (const IConnectableLayer *, const ConcatDescriptor &, const char *) override |
Function that a concat layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitConstantLayer (const IConnectableLayer *, const ConstTensor &, const char *) 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 IConnectableLayer *, const Convolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override |
Function that a 2D convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDepthToSpaceLayer (const IConnectableLayer *, const DepthToSpaceDescriptor &, const char *) override |
Function a depth to space layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDepthwiseConvolution2dLayer (const IConnectableLayer *, const DepthwiseConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) 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 IConnectableLayer *, const char *) override |
Function that a Dequantize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDetectionPostProcessLayer (const IConnectableLayer *, const DetectionPostProcessDescriptor &, const ConstTensor &, const char *) override |
Function that a Detection PostProcess layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitDivisionLayer (const IConnectableLayer *, const char *) override |
Function a division layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitElementwiseUnaryLayer (const IConnectableLayer *, const ElementwiseUnaryDescriptor &, const char *) override |
Function a ElementwiseUnary layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitEqualLayer (const IConnectableLayer *, const char *) override |
Function an Equal layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitFloorLayer (const IConnectableLayer *, const char *) override |
Function a floor layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitFullyConnectedLayer (const IConnectableLayer *, const FullyConnectedDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override |
Function that a fully connected layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitGatherLayer (const IConnectableLayer *, const char *) override |
Function a Gather layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitGreaterLayer (const IConnectableLayer *, const char *) override |
Function a Greater layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitInputLayer (const IConnectableLayer *, LayerBindingId, const char *) override |
Function that an InputLayer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitInstanceNormalizationLayer (const IConnectableLayer *, const InstanceNormalizationDescriptor &, const char *) override |
Function that an instance normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitL2NormalizationLayer (const IConnectableLayer *, const L2NormalizationDescriptor &, const char *) override |
Function that an L2 normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitLogSoftmaxLayer (const IConnectableLayer *, const LogSoftmaxDescriptor &, const char *) override |
Function that a log softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitLstmLayer (const IConnectableLayer *, const LstmDescriptor &, const LstmInputParams &, const char *) override |
Function an Lstm layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMaximumLayer (const IConnectableLayer *, const char *) override |
Function a Maximum layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMeanLayer (const IConnectableLayer *, const MeanDescriptor &, const char *) override |
Function a Mean layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMergeLayer (const IConnectableLayer *, const char *) override |
Function that a merge layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMergerLayer (const IConnectableLayer *, const MergerDescriptor &, const char *) override |
Function that a merger layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMinimumLayer (const IConnectableLayer *, const char *) override |
Function a Minimum layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitMultiplicationLayer (const IConnectableLayer *, const char *) override |
Function that a multiplication layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitNormalizationLayer (const IConnectableLayer *, const NormalizationDescriptor &, const char *) override |
Function that a normalization layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitOutputLayer (const IConnectableLayer *, LayerBindingId, const char *) override |
Function an output layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPadLayer (const IConnectableLayer *, const PadDescriptor &, const char *) override |
Function a pad layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPermuteLayer (const IConnectableLayer *, const PermuteDescriptor &, const char *) override |
Function that a permute layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPooling2dLayer (const IConnectableLayer *, const Pooling2dDescriptor &, const char *) override |
Function that a pooling layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitPreluLayer (const IConnectableLayer *, const char *) override |
Function that a PReLU activation layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitQuantizeLayer (const IConnectableLayer *, const char *) override |
Function a quantize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitQuantizedLstmLayer (const IConnectableLayer *, const QuantizedLstmInputParams &, const char *) override |
Function a QuantizedLstm layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitReshapeLayer (const IConnectableLayer *, const ReshapeDescriptor &, const char *) override |
Function a reshape layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitResizeBilinearLayer (const IConnectableLayer *, const ResizeBilinearDescriptor &, const char *) override |
Function that a resize bilinear layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitResizeLayer (const IConnectableLayer *, const ResizeDescriptor &, const char *) override |
Function that a resize layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitRsqrtLayer (const IConnectableLayer *, const char *) override |
Function a Reciprocal of square root layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSliceLayer (const IConnectableLayer *, const SliceDescriptor &, const char *) override |
Function that a slice layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSoftmaxLayer (const IConnectableLayer *, const SoftmaxDescriptor &, const char *) override |
Function that a softmax layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSpaceToBatchNdLayer (const IConnectableLayer *, const SpaceToBatchNdDescriptor &, const char *) override |
Function a space to batch layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSpaceToDepthLayer (const IConnectableLayer *, const SpaceToDepthDescriptor &, const char *) override |
Function a space to depth layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSplitterLayer (const IConnectableLayer *, const ViewsDescriptor &, const char *) override |
Function that a splitter layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitStackLayer (const IConnectableLayer *, const StackDescriptor &, const char *) override |
Function a stack layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitStandInLayer (const IConnectableLayer *, const StandInDescriptor &, const char *) override |
Function a StandInLayer should call back to when its Accept(ILaterVisitor&) function is invoked. More... | |
void | VisitStridedSliceLayer (const IConnectableLayer *, const StridedSliceDescriptor &, const char *) override |
Function a strided slice layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSubtractionLayer (const IConnectableLayer *, const char *) override |
Function a subtraction layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitSwitchLayer (const IConnectableLayer *, const char *) override |
Function a switch layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeConvolution2dLayer (const IConnectableLayer *, const TransposeConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override |
Function that a 2D transpose convolution layer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
void | VisitTransposeLayer (const IConnectableLayer *, const TransposeDescriptor &, const char *) 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 LayerVisitorBase< VisitorThrowingPolicy > | |
LayerVisitorBase () | |
virtual | ~LayerVisitorBase () |
Protected Member Functions inherited from ILayerVisitor | |
ILayerVisitor () | |
virtual | ~ILayerVisitor () |
Visitor object for quantizing layers in a network.
Definition at line 25 of file QuantizerVisitor.hpp.
QuantizerVisitor | ( | const RangeTracker & | rangeTracker, |
const IQuantizationScheme * | quantizationScheme, | ||
bool | preserveType = false |
||
) |
Definition at line 14 of file QuantizerVisitor.cpp.
References OutputSlot::CalculateIndexOnOwner(), IQuantizationScheme::ComputeScheme(), IOutputSlot::Connect(), InputSlot::GetConnectedOutputSlot(), IQuantizationScheme::GetDataType(), IConnectableLayer::GetGuid(), Layer::GetGuid(), BaseTensor< MemoryType >::GetInfo(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetNumInputSlots(), IConnectableLayer::GetOutputSlot(), OutputSlot::GetOwningLayer(), TensorInfo::GetQuantizationScale(), RangeTracker::GetRange(), OutputSlot::GetTensorInfo(), armnn::info, armnn::numeric_cast(), IOutputSlot::SetTensorInfo(), armnn::Signed32, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
|
default |
|
inline |
Extract the quantized network.
Definition at line 195 of file QuantizerVisitor.hpp.
|
overridevirtual |
Functions to quantize the individual layers, overridden from ILayerVisitor.
Implements ILayerVisitor.
Definition at line 116 of file QuantizerVisitor.cpp.
References armnn::Abs, and QuantizerVisitor::VisitElementwiseUnaryLayer().
|
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 121 of file QuantizerVisitor.cpp.
|
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 130 of file QuantizerVisitor.cpp.
|
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 137 of file QuantizerVisitor.cpp.
|
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 146 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst().
|
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 177 of file QuantizerVisitor.cpp.
|
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 186 of file QuantizerVisitor.cpp.
|
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 195 of file QuantizerVisitor.cpp.
|
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 204 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst().
|
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 215 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst(), and OptionalBase::has_value().
|
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 241 of file QuantizerVisitor.cpp.
|
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 250 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst(), and OptionalBase::has_value().
|
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 276 of file QuantizerVisitor.cpp.
Referenced by QuantizerVisitor::VisitAbsLayer(), and QuantizerVisitor::VisitRsqrtLayer().
|
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 285 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst(), and OptionalBase::has_value().
|
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 311 of file QuantizerVisitor.cpp.
References IOutputSlot::Connect(), armnn::Float16, armnn::Float32, TensorInfo::GetDataType(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), IOutputSlot::GetTensorInfo(), and IOutputSlot::SetTensorInfo().
|
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 329 of file QuantizerVisitor.cpp.
|
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 338 of file QuantizerVisitor.cpp.
|
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 347 of file QuantizerVisitor.cpp.
|
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 356 of file QuantizerVisitor.cpp.
|
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 364 of file QuantizerVisitor.cpp.
|
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 373 of file QuantizerVisitor.cpp.
References IOutputSlot::Connect(), armnn::Float16, armnn::Float32, IInputSlot::GetConnection(), TensorInfo::GetDataType(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetOutputSlot(), IOutputSlot::GetTensorInfo(), armnn::info, and IOutputSlot::SetTensorInfo().
|
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 394 of file QuantizerVisitor.cpp.
|
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 403 of file QuantizerVisitor.cpp.
|
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 412 of file QuantizerVisitor.cpp.
|
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 421 of file QuantizerVisitor.cpp.
|
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 429 of file QuantizerVisitor.cpp.
|
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 438 of file QuantizerVisitor.cpp.
References armnn::Bilinear, ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, ResizeDescriptor::m_Method, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, ResizeDescriptor::m_TargetWidth, and QuantizerVisitor::VisitResizeLayer().
|
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 451 of file QuantizerVisitor.cpp.
Referenced by QuantizerVisitor::VisitResizeBilinearLayer().
|
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 460 of file QuantizerVisitor.cpp.
References armnn::Rsqrt, and QuantizerVisitor::VisitElementwiseUnaryLayer().
|
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 465 of file QuantizerVisitor.cpp.
|
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 474 of file QuantizerVisitor.cpp.
|
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 483 of file QuantizerVisitor.cpp.
|
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 492 of file QuantizerVisitor.cpp.
|
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 501 of file QuantizerVisitor.cpp.
|
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 510 of file QuantizerVisitor.cpp.
|
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 519 of file QuantizerVisitor.cpp.
|
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 528 of file QuantizerVisitor.cpp.
|
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 536 of file QuantizerVisitor.cpp.
References armnn::CreateQuantizedConst(), and OptionalBase::has_value().
|
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 564 of file QuantizerVisitor.cpp.