ArmNN  NotReleased
Public Member Functions | List of all members
InputLayerVisitor Class Reference

Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine. More...

#include <QuantizationDataSet.hpp>

Inheritance diagram for InputLayerVisitor:
LayerVisitorBase< armnn::VisitorNoThrowPolicy > ILayerVisitor

Public Member Functions

void VisitInputLayer (const armnn::IConnectableLayer *layer, armnn::LayerBindingId id, const char *name)
 
armnn::TensorInfo GetTensorInfo (armnn::LayerBindingId)
 
- Public Member Functions inherited from LayerVisitorBase< armnn::VisitorNoThrowPolicy >
void VisitAbsLayer (const IConnectableLayer *, const char *) override
 
void VisitActivationLayer (const IConnectableLayer *, const ActivationDescriptor &, const char *) override
 
void VisitAdditionLayer (const IConnectableLayer *, const char *) override
 
void VisitArgMinMaxLayer (const IConnectableLayer *, const ArgMinMaxDescriptor &, const char *) override
 
void VisitBatchNormalizationLayer (const IConnectableLayer *, const BatchNormalizationDescriptor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const ConstTensor &, const char *) override
 
void VisitBatchToSpaceNdLayer (const IConnectableLayer *, const BatchToSpaceNdDescriptor &, const char *) override
 
void VisitComparisonLayer (const IConnectableLayer *, const ComparisonDescriptor &, const char *) override
 
void VisitConcatLayer (const IConnectableLayer *, const ConcatDescriptor &, const char *) override
 
void VisitConstantLayer (const IConnectableLayer *, const ConstTensor &, const char *) override
 
void VisitConvolution2dLayer (const IConnectableLayer *, const Convolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitDepthToSpaceLayer (const IConnectableLayer *, const DepthToSpaceDescriptor &, const char *) override
 
void VisitDepthwiseConvolution2dLayer (const IConnectableLayer *, const DepthwiseConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitDequantizeLayer (const IConnectableLayer *, const char *) override
 
void VisitDetectionPostProcessLayer (const IConnectableLayer *, const DetectionPostProcessDescriptor &, const ConstTensor &, const char *) override
 
void VisitDivisionLayer (const IConnectableLayer *, const char *) override
 
void VisitElementwiseUnaryLayer (const IConnectableLayer *, const ElementwiseUnaryDescriptor &, const char *) override
 
void VisitEqualLayer (const IConnectableLayer *, const char *) override
 
void VisitFloorLayer (const IConnectableLayer *, const char *) override
 
void VisitFullyConnectedLayer (const IConnectableLayer *, const FullyConnectedDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
void VisitGatherLayer (const IConnectableLayer *, const char *) override
 
void VisitGreaterLayer (const IConnectableLayer *, const char *) override
 
void VisitInputLayer (const IConnectableLayer *, LayerBindingId, const char *) override
 
void VisitInstanceNormalizationLayer (const IConnectableLayer *, const InstanceNormalizationDescriptor &, const char *) override
 
void VisitL2NormalizationLayer (const IConnectableLayer *, const L2NormalizationDescriptor &, const char *) override
 
void VisitLogSoftmaxLayer (const IConnectableLayer *, const LogSoftmaxDescriptor &, const char *) override
 
void VisitLstmLayer (const IConnectableLayer *, const LstmDescriptor &, const LstmInputParams &, const char *) override
 
void VisitMaximumLayer (const IConnectableLayer *, const char *) override
 
void VisitMeanLayer (const IConnectableLayer *, const MeanDescriptor &, const char *) override
 
void VisitMergeLayer (const IConnectableLayer *, const char *) override
 
void VisitMergerLayer (const IConnectableLayer *, const MergerDescriptor &, const char *) override
 
void VisitMinimumLayer (const IConnectableLayer *, const char *) override
 
void VisitMultiplicationLayer (const IConnectableLayer *, const char *) override
 
void VisitNormalizationLayer (const IConnectableLayer *, const NormalizationDescriptor &, const char *) override
 
void VisitOutputLayer (const IConnectableLayer *, LayerBindingId, const char *) override
 
void VisitPadLayer (const IConnectableLayer *, const PadDescriptor &, const char *) override
 
void VisitPermuteLayer (const IConnectableLayer *, const PermuteDescriptor &, const char *) override
 
void VisitPooling2dLayer (const IConnectableLayer *, const Pooling2dDescriptor &, const char *) override
 
void VisitPreluLayer (const IConnectableLayer *, const char *) override
 
void VisitQuantizeLayer (const IConnectableLayer *, const char *) override
 
void VisitQuantizedLstmLayer (const IConnectableLayer *, const QuantizedLstmInputParams &, const char *) override
 
void VisitReshapeLayer (const IConnectableLayer *, const ReshapeDescriptor &, const char *) override
 
void VisitResizeBilinearLayer (const IConnectableLayer *, const ResizeBilinearDescriptor &, const char *) override
 
void VisitResizeLayer (const IConnectableLayer *, const ResizeDescriptor &, const char *) override
 
void VisitRsqrtLayer (const IConnectableLayer *, const char *) override
 
void VisitSliceLayer (const IConnectableLayer *, const SliceDescriptor &, const char *) override
 
void VisitSoftmaxLayer (const IConnectableLayer *, const SoftmaxDescriptor &, const char *) override
 
void VisitSpaceToBatchNdLayer (const IConnectableLayer *, const SpaceToBatchNdDescriptor &, const char *) override
 
void VisitSpaceToDepthLayer (const IConnectableLayer *, const SpaceToDepthDescriptor &, const char *) override
 
void VisitSplitterLayer (const IConnectableLayer *, const ViewsDescriptor &, const char *) override
 
void VisitStackLayer (const IConnectableLayer *, const StackDescriptor &, const char *) override
 
void VisitStandInLayer (const IConnectableLayer *, const StandInDescriptor &, const char *) override
 
void VisitStridedSliceLayer (const IConnectableLayer *, const StridedSliceDescriptor &, const char *) override
 
void VisitSubtractionLayer (const IConnectableLayer *, const char *) override
 
void VisitSwitchLayer (const IConnectableLayer *, const char *) override
 
void VisitTransposeConvolution2dLayer (const IConnectableLayer *, const TransposeConvolution2dDescriptor &, const ConstTensor &, const Optional< ConstTensor > &, const char *) override
 
- 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 ()
 

Detailed Description

Visitor class implementation to gather the TensorInfo for LayerBindingID for creation of ConstTensor for Refine.

Definition at line 46 of file QuantizationDataSet.hpp.

Member Function Documentation

◆ GetTensorInfo()

armnn::TensorInfo GetTensorInfo ( armnn::LayerBindingId  layerBindingId)

Definition at line 59 of file QuantizationDataSet.cpp.

Referenced by main().

60 {
61  auto iterator = m_TensorInfos.find(layerBindingId);
62  if (iterator != m_TensorInfos.end())
63  {
64  return m_TensorInfos.at(layerBindingId);
65  }
66  else
67  {
68  throw armnn::Exception("Could not retrieve tensor info for binding ID " + std::to_string(layerBindingId));
69  }
70 }
armnn::Exception
Base class for all ArmNN exceptions so that users can filter to just those.
Definition: Exceptions.hpp:46

◆ VisitInputLayer()

void VisitInputLayer ( const armnn::IConnectableLayer *  layer,
armnn::LayerBindingId  id,
const char *  name 
)
virtual

Function that an InputLayer should call back to when its Accept(ILayerVisitor&) function is invoked.

Parameters
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 51 of file QuantizationDataSet.cpp.

References IConnectableLayer::GetOutputSlot(), and IOutputSlot::GetTensorInfo().

54 {
55  boost::ignore_unused(name);
56  m_TensorInfos.emplace(id, layer->GetOutputSlot(0).GetTensorInfo());
57 }
armnn::IOutputSlot::GetTensorInfo
virtual const TensorInfo & GetTensorInfo() const =0
armnn::IConnectableLayer::GetOutputSlot
virtual const IOutputSlot & GetOutputSlot(unsigned int index) const =0
The documentation for this class was generated from the following files: