ArmNN
 20.11
Classes | Typedefs | Functions
armnnSerializer Namespace Reference

Classes

class  ISerializer
 
class  Serializer
 
class  SerializerVisitor
 

Typedefs

using ISerializerPtr = std::unique_ptr< ISerializer, void(*)(ISerializer *serializer)>
 

Functions

serializer::ActivationFunction GetFlatBufferActivationFunction (armnn::ActivationFunction function)
 
serializer::ArgMinMaxFunction GetFlatBufferArgMinMaxFunction (armnn::ArgMinMaxFunction function)
 
armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation (armnn::ComparisonOperation comparisonOperation)
 
armnnSerializer::LogicalBinaryOperation GetFlatBufferLogicalBinaryOperation (armnn::LogicalBinaryOperation logicalBinaryOperation)
 
armnnSerializer::ConstTensorData GetFlatBufferConstTensorData (armnn::DataType dataType)
 
armnnSerializer::DataType GetFlatBufferDataType (armnn::DataType dataType)
 
armnnSerializer::DataLayout GetFlatBufferDataLayout (armnn::DataLayout dataLayout)
 
armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation (armnn::UnaryOperation comparisonOperation)
 
armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm (armnn::PoolingAlgorithm poolingAlgorithm)
 
armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding (armnn::OutputShapeRounding outputShapeRounding)
 
armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod (armnn::PaddingMethod paddingMethod)
 
armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel (armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel)
 
armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod (armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod)
 
armnnSerializer::ResizeMethod GetFlatBufferResizeMethod (armnn::ResizeMethod method)
 

Typedef Documentation

◆ ISerializerPtr

using ISerializerPtr = std::unique_ptr<ISerializer, void(*)(ISerializer* serializer)>

Definition at line 15 of file ISerializer.hpp.

Function Documentation

◆ GetFlatBufferActivationFunction()

serializer::ActivationFunction armnnSerializer::GetFlatBufferActivationFunction ( armnn::ActivationFunction  function)

Definition at line 26 of file Serializer.cpp.

References armnn::Abs, armnn::BoundedReLu, armnn::Elu, armnn::HardSwish, armnn::LeakyReLu, armnn::Linear, armnn::ReLu, armnn::Sigmoid, armnn::Sqrt, armnn::Square, and armnn::TanH.

Referenced by SerializerVisitor::VisitActivationLayer().

27 {
28  switch (function)
29  {
30  case armnn::ActivationFunction::Sigmoid:
31  return serializer::ActivationFunction::ActivationFunction_Sigmoid;
32  case armnn::ActivationFunction::TanH:
33  return serializer::ActivationFunction::ActivationFunction_TanH;
34  case armnn::ActivationFunction::Linear:
35  return serializer::ActivationFunction::ActivationFunction_Linear;
36  case armnn::ActivationFunction::ReLu:
37  return serializer::ActivationFunction::ActivationFunction_ReLu;
38  case armnn::ActivationFunction::BoundedReLu:
39  return serializer::ActivationFunction::ActivationFunction_BoundedReLu;
40  case armnn::ActivationFunction::LeakyReLu:
41  return serializer::ActivationFunction::ActivationFunction_LeakyReLu;
42  case armnn::ActivationFunction::Abs:
43  return serializer::ActivationFunction::ActivationFunction_Abs;
44  case armnn::ActivationFunction::Sqrt:
45  return serializer::ActivationFunction::ActivationFunction_Sqrt;
46  case armnn::ActivationFunction::Square:
47  return serializer::ActivationFunction::ActivationFunction_Square;
48  case armnn::ActivationFunction::Elu:
49  return serializer::ActivationFunction::ActivationFunction_Elu;
50  case armnn::ActivationFunction::HardSwish:
51  return serializer::ActivationFunction::ActivationFunction_HardSwish;
52  default:
53  return serializer::ActivationFunction::ActivationFunction_Sigmoid;
54  }
55 }
armnn::ActivationFunction::BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.

◆ GetFlatBufferArgMinMaxFunction()

serializer::ArgMinMaxFunction armnnSerializer::GetFlatBufferArgMinMaxFunction ( armnn::ArgMinMaxFunction  function)

Definition at line 57 of file Serializer.cpp.

References armnn::Max, and armnn::Min.

Referenced by SerializerVisitor::VisitArgMinMaxLayer().

58 {
59  switch (function)
60  {
61  case armnn::ArgMinMaxFunction::Max:
62  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max;
63  case armnn::ArgMinMaxFunction::Min:
64  default:
65  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min;
66  }
67 }

◆ GetFlatBufferComparisonOperation()

armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation ( armnn::ComparisonOperation  comparisonOperation)

Definition at line 11 of file SerializerUtils.cpp.

References armnn::Equal, armnn::Greater, armnn::GreaterOrEqual, armnn::Less, armnn::LessOrEqual, and armnn::NotEqual.

Referenced by SerializerVisitor::VisitComparisonLayer().

12 {
13  switch (comparisonOperation)
14  {
15  case armnn::ComparisonOperation::Equal:
16  return armnnSerializer::ComparisonOperation::ComparisonOperation_Equal;
17  case armnn::ComparisonOperation::Greater:
18  return armnnSerializer::ComparisonOperation::ComparisonOperation_Greater;
19  case armnn::ComparisonOperation::GreaterOrEqual:
20  return armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual;
21  case armnn::ComparisonOperation::Less:
22  return armnnSerializer::ComparisonOperation::ComparisonOperation_Less;
23  case armnn::ComparisonOperation::LessOrEqual:
24  return armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual;
25  case armnn::ComparisonOperation::NotEqual:
26  default:
27  return armnnSerializer::ComparisonOperation::ComparisonOperation_NotEqual;
28  }
29 }

◆ GetFlatBufferConstTensorData()

armnnSerializer::ConstTensorData GetFlatBufferConstTensorData ( armnn::DataType  dataType)

Definition at line 45 of file SerializerUtils.cpp.

References armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, and armnn::Signed32.

Referenced by SerializerVisitor::VisitQuantizedLstmLayer().

46 {
47  switch (dataType)
48  {
49  case armnn::DataType::Float32:
50  case armnn::DataType::Signed32:
51  return armnnSerializer::ConstTensorData::ConstTensorData_IntData;
52  case armnn::DataType::Float16:
53  case armnn::DataType::QSymmS16:
54  return armnnSerializer::ConstTensorData::ConstTensorData_ShortData;
55  case armnn::DataType::QAsymmU8:
56  case armnn::DataType::QSymmS8:
57  case armnn::DataType::Boolean:
58  return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
59  default:
60  return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
61  }
62 }

◆ GetFlatBufferDataLayout()

armnnSerializer::DataLayout GetFlatBufferDataLayout ( armnn::DataLayout  dataLayout)

Definition at line 89 of file SerializerUtils.cpp.

References armnn::NCHW, and armnn::NHWC.

Referenced by SerializerVisitor::VisitBatchNormalizationLayer(), SerializerVisitor::VisitBatchToSpaceNdLayer(), SerializerVisitor::VisitConvolution2dLayer(), SerializerVisitor::VisitDepthToSpaceLayer(), SerializerVisitor::VisitDepthwiseConvolution2dLayer(), SerializerVisitor::VisitInstanceNormalizationLayer(), SerializerVisitor::VisitL2NormalizationLayer(), SerializerVisitor::VisitNormalizationLayer(), SerializerVisitor::VisitPooling2dLayer(), SerializerVisitor::VisitResizeBilinearLayer(), SerializerVisitor::VisitResizeLayer(), SerializerVisitor::VisitSpaceToBatchNdLayer(), SerializerVisitor::VisitSpaceToDepthLayer(), SerializerVisitor::VisitStridedSliceLayer(), and SerializerVisitor::VisitTransposeConvolution2dLayer().

90 {
91  switch (dataLayout)
92  {
93  case armnn::DataLayout::NHWC:
94  return armnnSerializer::DataLayout::DataLayout_NHWC;
95  case armnn::DataLayout::NCHW:
96  default:
97  return armnnSerializer::DataLayout::DataLayout_NCHW;
98  }
99 }

◆ GetFlatBufferDataType()

armnnSerializer::DataType GetFlatBufferDataType ( armnn::DataType  dataType)

Definition at line 64 of file SerializerUtils.cpp.

References armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, and armnn::Signed32.

Referenced by SerializerVisitor::VisitQuantizedLstmLayer().

65 {
66  switch (dataType)
67  {
68  case armnn::DataType::Float32:
69  return armnnSerializer::DataType::DataType_Float32;
70  case armnn::DataType::Float16:
71  return armnnSerializer::DataType::DataType_Float16;
72  case armnn::DataType::Signed32:
73  return armnnSerializer::DataType::DataType_Signed32;
74  case armnn::DataType::QSymmS16:
75  return armnnSerializer::DataType::DataType_QSymmS16;
76  case armnn::DataType::QAsymmS8:
77  return armnnSerializer::DataType::DataType_QAsymmS8;
78  case armnn::DataType::QAsymmU8:
79  return armnnSerializer::DataType::DataType_QAsymmU8;
80  case armnn::DataType::QSymmS8:
81  return armnnSerializer::DataType::DataType_QSymmS8;
82  case armnn::DataType::Boolean:
83  return armnnSerializer::DataType::DataType_Boolean;
84  default:
85  return armnnSerializer::DataType::DataType_Float16;
86  }
87 }

◆ GetFlatBufferLogicalBinaryOperation()

armnnSerializer::LogicalBinaryOperation GetFlatBufferLogicalBinaryOperation ( armnn::LogicalBinaryOperation  logicalBinaryOperation)

Definition at line 31 of file SerializerUtils.cpp.

References armnn::LogicalAnd, and armnn::LogicalOr.

Referenced by SerializerVisitor::VisitLogicalBinaryLayer().

33 {
34  switch (logicalBinaryOperation)
35  {
36  case armnn::LogicalBinaryOperation::LogicalAnd:
37  return armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalAnd;
38  case armnn::LogicalBinaryOperation::LogicalOr:
39  return armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalOr;
40  default:
41  throw armnn::InvalidArgumentException("Logical Binary operation unknown");
42  }
43 }

◆ GetFlatBufferNormalizationAlgorithmChannel()

armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel ( armnn::NormalizationAlgorithmChannel  normalizationAlgorithmChannel)

Definition at line 160 of file SerializerUtils.cpp.

References armnn::Across, and armnn::Within.

Referenced by SerializerVisitor::VisitNormalizationLayer().

162 {
163  switch (normalizationAlgorithmChannel)
164  {
165  case armnn::NormalizationAlgorithmChannel::Across:
166  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
167  case armnn::NormalizationAlgorithmChannel::Within:
168  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
169  default:
170  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
171  }
172 }

◆ GetFlatBufferNormalizationAlgorithmMethod()

armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod ( armnn::NormalizationAlgorithmMethod  normalizationAlgorithmMethod)

Definition at line 174 of file SerializerUtils.cpp.

References armnn::LocalBrightness, and armnn::LocalContrast.

Referenced by SerializerVisitor::VisitNormalizationLayer().

176 {
177  switch (normalizationAlgorithmMethod)
178  {
179  case armnn::NormalizationAlgorithmMethod::LocalBrightness:
180  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
181  case armnn::NormalizationAlgorithmMethod::LocalContrast:
182  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
183  default:
184  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
185  }
186 }
armnn::NormalizationAlgorithmMethod::LocalContrast
Jarret 2009: Local Contrast Normalization.
armnn::NormalizationAlgorithmMethod::LocalBrightness
Krichevsky 2012: Local Brightness Normalization.

◆ GetFlatBufferOutputShapeRounding()

armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding ( armnn::OutputShapeRounding  outputShapeRounding)

Definition at line 136 of file SerializerUtils.cpp.

References armnn::Ceiling, and armnn::Floor.

Referenced by SerializerVisitor::VisitPooling2dLayer().

137 {
138  switch (outputShapeRounding)
139  {
140  case armnn::OutputShapeRounding::Ceiling:
141  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
142  case armnn::OutputShapeRounding::Floor:
143  default:
144  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
145  }
146 }

◆ GetFlatBufferPaddingMethod()

armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod ( armnn::PaddingMethod  paddingMethod)

Definition at line 148 of file SerializerUtils.cpp.

References armnn::Exclude, and armnn::IgnoreValue.

Referenced by SerializerVisitor::VisitPooling2dLayer().

149 {
150  switch (paddingMethod)
151  {
152  case armnn::PaddingMethod::IgnoreValue:
153  return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
154  case armnn::PaddingMethod::Exclude:
155  default:
156  return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
157  }
158 }
armnn::PaddingMethod::Exclude
The padding fields don&#39;t count and are ignored.
armnn::PaddingMethod::IgnoreValue
The padding fields count, but are ignored.

◆ GetFlatBufferPoolingAlgorithm()

armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm ( armnn::PoolingAlgorithm  poolingAlgorithm)

Definition at line 122 of file SerializerUtils.cpp.

References armnn::Average, armnn::L2, and armnn::Max.

Referenced by SerializerVisitor::VisitPooling2dLayer().

123 {
124  switch (poolingAlgorithm)
125  {
126  case armnn::PoolingAlgorithm::Average:
127  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
128  case armnn::PoolingAlgorithm::L2:
129  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
130  case armnn::PoolingAlgorithm::Max:
131  default:
132  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
133  }
134 }

◆ GetFlatBufferResizeMethod()

armnnSerializer::ResizeMethod GetFlatBufferResizeMethod ( armnn::ResizeMethod  method)

Definition at line 188 of file SerializerUtils.cpp.

References armnn::Bilinear, and armnn::NearestNeighbor.

Referenced by SerializerVisitor::VisitResizeLayer().

189 {
190  switch (method)
191  {
192  case armnn::ResizeMethod::NearestNeighbor:
193  return armnnSerializer::ResizeMethod_NearestNeighbor;
194  case armnn::ResizeMethod::Bilinear:
195  return armnnSerializer::ResizeMethod_Bilinear;
196  default:
197  return armnnSerializer::ResizeMethod_NearestNeighbor;
198  }
199 }

◆ GetFlatBufferUnaryOperation()

armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation ( armnn::UnaryOperation  comparisonOperation)

Definition at line 101 of file SerializerUtils.cpp.

References armnn::Abs, armnn::Exp, armnn::LogicalNot, armnn::Neg, armnn::Rsqrt, and armnn::Sqrt.

Referenced by SerializerVisitor::VisitElementwiseUnaryLayer().

102 {
103  switch (comparisonOperation)
104  {
105  case armnn::UnaryOperation::Abs:
106  return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
107  case armnn::UnaryOperation::Rsqrt:
108  return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
109  case armnn::UnaryOperation::Sqrt:
110  return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
111  case armnn::UnaryOperation::Exp:
112  return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
113  case armnn::UnaryOperation::Neg:
114  return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
115  case armnn::UnaryOperation::LogicalNot:
116  return armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot;
117  default:
118  throw armnn::InvalidArgumentException("Unary operation unknown");
119  }
120 }