Compute Library
 22.11
DataLayerVisitor Class Referencefinal

Graph printer visitor. More...

#include <DataLayerVisitor.h>

Collaboration diagram for DataLayerVisitor:
[legend]

Public Types

using LayerData = std::map< std::string, std::string >
 

Public Member Functions

 DataLayerVisitor ()=default
 Default Constructor. More...
 
const LayerDatalayer_data () const
 
void visit (ConvolutionLayerNode &n) override
 Visit ConvolutionLayerNode. More...
 
void visit (DepthwiseConvolutionLayerNode &n) override
 Visit DepthwiseConvolutionLayerNode. More...
 
void visit (FusedConvolutionBatchNormalizationNode &n) override
 Visit FusedConvolutionBatchNormalizationNode. More...
 
void visit (FusedConvolutionBatchNormalizationWithPostOpsNode &n) override
 Visit FusedConvolutionBatchNormalizationWithPostOpsNode. More...
 
void visit (FusedDepthwiseConvolutionBatchNormalizationNode &n) override
 Visit FusedDepthwiseConvolutionBatchNormalizationNode. More...
 
void visit (OutputNode &n) override
 Visit OutputNode. More...
 
void default_visit (INode &n) override
 Function to be overloaded by the client and implement default behavior for the non-overloaded visitors. More...
 
- Public Member Functions inherited from DefaultNodeVisitor
virtual ~DefaultNodeVisitor ()=default
 Default destructor. More...
 
- Public Member Functions inherited from INodeVisitor
virtual ~INodeVisitor ()=default
 Default destructor. More...
 
virtual void visit (INode &n)=0
 Visit INode. More...
 
virtual void visit (ActivationLayerNode &n)=0
 Visit ActivationLayerNode. More...
 
virtual void visit (BatchNormalizationLayerNode &n)=0
 Visit BatchNormalizationLayerNode. More...
 
virtual void visit (ConcatenateLayerNode &n)=0
 Visit ConcatenateLayerNode. More...
 
virtual void visit (ConstNode &n)=0
 Visit ConstNode. More...
 
virtual void visit (DequantizationLayerNode &n)=0
 Visit DequantizationLayerNode. More...
 
virtual void visit (DetectionOutputLayerNode &n)=0
 Visit DetectionOutputLayerNode. More...
 
virtual void visit (DetectionPostProcessLayerNode &n)=0
 Visit DetectionPostProcessLayerNode. More...
 
virtual void visit (EltwiseLayerNode &n)=0
 Visit EltwiseLayerNode. More...
 
virtual void visit (FlattenLayerNode &n)=0
 Visit FlattenLayerNode. More...
 
virtual void visit (FullyConnectedLayerNode &n)=0
 Visit FullyConnectedLayerNode. More...
 
virtual void visit (FusedConvolutionWithPostOpNode &n)=0
 Visit FusedConvolutionWithPostOpNode. More...
 
virtual void visit (InputNode &n)=0
 Visit InputNode. More...
 
virtual void visit (NormalizationLayerNode &n)=0
 Visit NormalizationLayerNode. More...
 
virtual void visit (PermuteLayerNode &n)=0
 Visit PermuteLayerNode. More...
 
virtual void visit (PReluLayerNode &n)=0
 Visit PreluLayerNode. More...
 
virtual void visit (PoolingLayerNode &n)=0
 Visit PoolingLayerNode. More...
 
virtual void visit (PrintLayerNode &n)=0
 Visit PrintLayerNode. More...
 
virtual void visit (PriorBoxLayerNode &n)=0
 Visit PriorBoxLayerNode. More...
 
virtual void visit (QuantizationLayerNode &n)=0
 Visit QuantizationLayerNode. More...
 
virtual void visit (ReshapeLayerNode &n)=0
 Visit ReshapeLayerNode. More...
 
virtual void visit (SoftmaxLayerNode &n)=0
 Visit SoftmaxLayerNode. More...
 
virtual void visit (SplitLayerNode &n)=0
 Visit SplitLayerNode. More...
 
virtual void visit (StackLayerNode &n)=0
 Visit StackLayerNode. More...
 

Detailed Description

Graph printer visitor.

Definition at line 36 of file DataLayerVisitor.h.

Member Typedef Documentation

◆ LayerData

using LayerData = std::map<std::string, std::string>

Definition at line 39 of file DataLayerVisitor.h.

Constructor & Destructor Documentation

◆ DataLayerVisitor()

DataLayerVisitor ( )
default

Default Constructor.

Member Function Documentation

◆ default_visit()

void default_visit ( INode n)
overridevirtual

Function to be overloaded by the client and implement default behavior for the non-overloaded visitors.

Implements DefaultNodeVisitor.

Definition at line 155 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

156 {
157  _layer_data.clear();
158  add_generic_layer_data<INode>(_layer_data, n);
159 }

◆ layer_data()

const DataLayerVisitor::LayerData & layer_data ( ) const

Definition at line 161 of file DataLayerVisitor.cpp.

Referenced by SchedulerClock< output_timestamps >::test_start().

162 {
163  return _layer_data;
164 }

◆ visit() [1/6]

void visit ( ConvolutionLayerNode n)
overridevirtual

Visit ConvolutionLayerNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 111 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

112 {
113  _layer_data.clear();
114  add_generic_layer_data<ConvolutionLayerNode>(_layer_data, n);
115  add_convolution_layer_data<ConvolutionLayerNode>(_layer_data, n);
116  add_convolution_layer_method<ConvolutionLayerNode>(_layer_data, n);
117 }

◆ visit() [2/6]

void visit ( DepthwiseConvolutionLayerNode n)
overridevirtual

Visit DepthwiseConvolutionLayerNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 119 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

120 {
121  _layer_data.clear();
122  add_generic_layer_data<DepthwiseConvolutionLayerNode>(_layer_data, n);
123  add_convolution_layer_data<DepthwiseConvolutionLayerNode>(_layer_data, n);
124 }

◆ visit() [3/6]

void visit ( FusedConvolutionBatchNormalizationNode n)
overridevirtual

Visit FusedConvolutionBatchNormalizationNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 126 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

127 {
128  _layer_data.clear();
129  add_generic_layer_data<FusedConvolutionBatchNormalizationNode>(_layer_data, n);
130  add_convolution_layer_data<FusedConvolutionBatchNormalizationNode>(_layer_data, n);
131  add_convolution_layer_method<FusedConvolutionBatchNormalizationNode>(_layer_data, n);
132 }

◆ visit() [4/6]

void visit ( FusedConvolutionBatchNormalizationWithPostOpsNode n)
overridevirtual

Visit FusedConvolutionBatchNormalizationWithPostOpsNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 134 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

135 {
136  _layer_data.clear();
137  add_generic_layer_data<FusedConvolutionBatchNormalizationWithPostOpsNode>(_layer_data, n);
138  add_convolution_layer_data<FusedConvolutionBatchNormalizationWithPostOpsNode>(_layer_data, n);
139  add_convolution_layer_method<FusedConvolutionBatchNormalizationWithPostOpsNode>(_layer_data, n);
140 }

◆ visit() [5/6]

void visit ( FusedDepthwiseConvolutionBatchNormalizationNode n)
overridevirtual

Visit FusedDepthwiseConvolutionBatchNormalizationNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 142 of file DataLayerVisitor.cpp.

References arm_compute::test::validation::n.

143 {
144  _layer_data.clear();
145  add_generic_layer_data<FusedDepthwiseConvolutionBatchNormalizationNode>(_layer_data, n);
146  add_convolution_layer_data<FusedDepthwiseConvolutionBatchNormalizationNode>(_layer_data, n);
147 }

◆ visit() [6/6]

void visit ( OutputNode n)
overridevirtual

Visit OutputNode.

Parameters
[in]nNode to visit.

Implements INodeVisitor.

Definition at line 149 of file DataLayerVisitor.cpp.

References ARM_COMPUTE_UNUSED.

150 {
151  _layer_data.clear();
153 }
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Definition: Error.h:152

The documentation for this class was generated from the following files: