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

#include <ConstTensorLayerVisitor.hpp>

Inheritance diagram for TestConstantLayerVisitor:
TestLayerVisitor LayerVisitorBase< VisitorNoThrowPolicy > ILayerVisitor

Public Member Functions

 TestConstantLayerVisitor (const ConstTensor &input, const char *name=nullptr)
 
void VisitConstantLayer (const IConnectableLayer *layer, const ConstTensor &input, const char *name=nullptr)
 
- Public Member Functions inherited from TestLayerVisitor
 TestLayerVisitor (const char *name)
 
- Public Member Functions inherited from LayerVisitorBase< 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 TestLayerVisitor
virtual ~TestLayerVisitor ()
 
void CheckLayerName (const char *name)
 
void CheckLayerPointer (const IConnectableLayer *layer)
 
void CheckConstTensors (const ConstTensor &expected, const ConstTensor &actual)
 
void CheckOptionalConstTensors (const Optional< ConstTensor > &expected, const Optional< ConstTensor > &actual)
 
- Protected Member Functions inherited from LayerVisitorBase< VisitorNoThrowPolicy >
 LayerVisitorBase ()
 
virtual ~LayerVisitorBase ()
 
- Protected Member Functions inherited from ILayerVisitor
 ILayerVisitor ()
 
virtual ~ILayerVisitor ()
 

Detailed Description

Definition at line 170 of file ConstTensorLayerVisitor.hpp.

Constructor & Destructor Documentation

◆ TestConstantLayerVisitor()

TestConstantLayerVisitor ( const ConstTensor &  input,
const char *  name = nullptr 
)
inlineexplicit

Definition at line 173 of file ConstTensorLayerVisitor.hpp.

175  : TestLayerVisitor(name)
176  , m_Input(input)
177  {}
armnn::TestLayerVisitor::TestLayerVisitor
TestLayerVisitor(const char *name)
Definition: TestLayerVisitor.hpp:30

Member Function Documentation

◆ VisitConstantLayer()

void VisitConstantLayer ( const IConnectableLayer *  layer,
const ConstTensor &  input,
const char *  name = nullptr 
)
inlinevirtual

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.

Parameters
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 179 of file ConstTensorLayerVisitor.hpp.

References TestLayerVisitor::CheckConstTensors(), TestLayerVisitor::CheckLayerName(), and TestLayerVisitor::CheckLayerPointer().

182  {
183  CheckLayerPointer(layer);
184  CheckLayerName(name);
185  CheckConstTensors(m_Input, input);
186  }
armnn::TestLayerVisitor::CheckLayerName
void CheckLayerName(const char *name)
Definition: TestLayerVisitor.cpp:12
armnn::TestLayerVisitor::CheckLayerPointer
void CheckLayerPointer(const IConnectableLayer *layer)
Definition: TestLayerVisitor.cpp:28
armnn::TestLayerVisitor::CheckConstTensors
void CheckConstTensors(const ConstTensor &expected, const ConstTensor &actual)
Definition: TestLayerVisitor.cpp:33
The documentation for this class was generated from the following file: