20.02
|
Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine. More...
#include <QuantizationDataSet.hpp>
Public Member Functions | |
void | VisitInputLayer (const armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const char *name) |
Function that an InputLayer should call back to when its Accept(ILayerVisitor&) function is invoked. More... | |
armnn::TensorInfo | GetTensorInfo (armnn::LayerBindingId) |
Public Member Functions inherited from LayerVisitorBase< armnn::VisitorNoThrowPolicy > | |
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< armnn::VisitorNoThrowPolicy > | |
LayerVisitorBase () | |
virtual | ~LayerVisitorBase () |
Protected Member Functions inherited from ILayerVisitor | |
ILayerVisitor () | |
virtual | ~ILayerVisitor () |
Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine.
Definition at line 46 of file QuantizationDataSet.hpp.
armnn::TensorInfo GetTensorInfo | ( | armnn::LayerBindingId | layerBindingId | ) |
Definition at line 60 of file QuantizationDataSet.cpp.
Referenced by main().
|
virtual |
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 52 of file QuantizationDataSet.cpp.
References IConnectableLayer::GetOutputSlot(), IOutputSlot::GetTensorInfo(), and armnn::IgnoreUnused().