ArmNN
 25.02
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Classes | Typedefs | Functions
armnnSerializer Namespace Reference

Classes

class  ISerializer
 
class  SerializerStrategy
 

Typedefs

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

Functions

serializer::ActivationFunction GetFlatBufferActivationFunction (armnn::ActivationFunction function)
 
serializer::ArgMinMaxFunction GetFlatBufferArgMinMaxFunction (armnn::ArgMinMaxFunction function)
 
serializer::ScatterNdFunction GetFlatBufferScatterNdFunction (armnn::ScatterNdFunction 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::BinaryOperation GetFlatBufferBinaryOperation (armnn::BinaryOperation binaryOperation)
 
armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation (armnn::UnaryOperation unaryOperation)
 
armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm (armnn::PoolingAlgorithm poolingAlgorithm)
 
armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding (armnn::OutputShapeRounding outputShapeRounding)
 
armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod (armnn::PaddingMethod paddingMethod)
 
armnnSerializer::PaddingMode GetFlatBufferPaddingMode (armnn::PaddingMode paddingMode)
 
armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel (armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel)
 
armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod (armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod)
 
armnnSerializer::ResizeMethod GetFlatBufferResizeMethod (armnn::ResizeMethod method)
 
armnnSerializer::ReduceOperation GetFlatBufferReduceOperation (armnn::ReduceOperation reduceOperation)
 

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 55 of file Serializer.cpp.

56 {
57  switch (function)
58  {
59  case armnn::ActivationFunction::Sigmoid:
60  return serializer::ActivationFunction::ActivationFunction_Sigmoid;
61  case armnn::ActivationFunction::TanH:
62  return serializer::ActivationFunction::ActivationFunction_TanH;
63  case armnn::ActivationFunction::Linear:
64  return serializer::ActivationFunction::ActivationFunction_Linear;
65  case armnn::ActivationFunction::ReLu:
66  return serializer::ActivationFunction::ActivationFunction_ReLu;
67  case armnn::ActivationFunction::BoundedReLu:
68  return serializer::ActivationFunction::ActivationFunction_BoundedReLu;
69  case armnn::ActivationFunction::LeakyReLu:
70  return serializer::ActivationFunction::ActivationFunction_LeakyReLu;
71  case armnn::ActivationFunction::Abs:
72  return serializer::ActivationFunction::ActivationFunction_Abs;
73  case armnn::ActivationFunction::Sqrt:
74  return serializer::ActivationFunction::ActivationFunction_Sqrt;
75  case armnn::ActivationFunction::Square:
76  return serializer::ActivationFunction::ActivationFunction_Square;
77  case armnn::ActivationFunction::Elu:
78  return serializer::ActivationFunction::ActivationFunction_Elu;
79  case armnn::ActivationFunction::HardSwish:
80  return serializer::ActivationFunction::ActivationFunction_HardSwish;
81  case armnn::ActivationFunction::Gelu:
82  return serializer::ActivationFunction::ActivationFunction_Gelu;
83  default:
84  return serializer::ActivationFunction::ActivationFunction_Sigmoid;
85  }
86 }
armnn::ActivationFunction::BoundedReLu
@ BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.

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

◆ GetFlatBufferArgMinMaxFunction()

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

Definition at line 88 of file Serializer.cpp.

89 {
90  switch (function)
91  {
92  case armnn::ArgMinMaxFunction::Max:
93  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max;
94  case armnn::ArgMinMaxFunction::Min:
95  default:
96  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min;
97  }
98 }

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

◆ GetFlatBufferBinaryOperation()

armnnSerializer::BinaryOperation GetFlatBufferBinaryOperation ( armnn::BinaryOperation  binaryOperation)

Definition at line 110 of file SerializerUtils.cpp.

111 {
112  switch (binaryOperation)
113  {
114  case armnn::BinaryOperation::Add:
115  return armnnSerializer::BinaryOperation::BinaryOperation_Add;
116  case armnn::BinaryOperation::Div:
117  return armnnSerializer::BinaryOperation::BinaryOperation_Div;
118  case armnn::BinaryOperation::FloorDiv:
119  return armnnSerializer::BinaryOperation::BinaryOperation_Div;
120  case armnn::BinaryOperation::Maximum:
121  return armnnSerializer::BinaryOperation::BinaryOperation_Maximum;
122  case armnn::BinaryOperation::Minimum:
123  return armnnSerializer::BinaryOperation::BinaryOperation_Minimum;
124  case armnn::BinaryOperation::Mul:
125  return armnnSerializer::BinaryOperation::BinaryOperation_Mul;
126  case armnn::BinaryOperation::Sub:
127  return armnnSerializer::BinaryOperation::BinaryOperation_Sub;
128  case armnn::BinaryOperation::SqDiff:
129  return armnnSerializer::BinaryOperation::BinaryOperation_SqDiff;
130  case armnn::BinaryOperation::Power:
131  return armnnSerializer::BinaryOperation::BinaryOperation_Power;
132  default:
133  throw armnn::InvalidArgumentException("Elementwise Binary operation unknown");
134  }
135 }

References armnn::Add, armnn::Div, armnn::FloorDiv, armnn::Maximum, armnn::Minimum, armnn::Mul, armnn::Power, armnn::SqDiff, and armnn::Sub.

◆ GetFlatBufferComparisonOperation()

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

Definition at line 11 of file SerializerUtils.cpp.

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 }

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

◆ GetFlatBufferConstTensorData()

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

Definition at line 45 of file SerializerUtils.cpp.

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::QAsymmS8:
56  case armnn::DataType::QAsymmU8:
57  case armnn::DataType::QSymmS8:
58  case armnn::DataType::Boolean:
59  return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
60  case armnn::DataType::Signed64:
61  return armnnSerializer::ConstTensorData::ConstTensorData_LongData;
62  default:
63  return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
64  }
65 }

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

◆ GetFlatBufferDataLayout()

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

Definition at line 94 of file SerializerUtils.cpp.

95 {
96  switch (dataLayout)
97  {
98  case armnn::DataLayout::NHWC:
99  return armnnSerializer::DataLayout::DataLayout_NHWC;
100  case armnn::DataLayout::NDHWC:
101  return armnnSerializer::DataLayout::DataLayout_NDHWC;
102  case armnn::DataLayout::NCDHW:
103  return armnnSerializer::DataLayout::DataLayout_NCDHW;
104  case armnn::DataLayout::NCHW:
105  default:
106  return armnnSerializer::DataLayout::DataLayout_NCHW;
107  }
108 }

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

◆ GetFlatBufferDataType()

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

Definition at line 67 of file SerializerUtils.cpp.

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

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

◆ GetFlatBufferLogicalBinaryOperation()

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

Definition at line 31 of file SerializerUtils.cpp.

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 }

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

◆ GetFlatBufferNormalizationAlgorithmChannel()

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

Definition at line 215 of file SerializerUtils.cpp.

217 {
218  switch (normalizationAlgorithmChannel)
219  {
220  case armnn::NormalizationAlgorithmChannel::Across:
221  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
222  case armnn::NormalizationAlgorithmChannel::Within:
223  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
224  default:
225  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
226  }
227 }

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

◆ GetFlatBufferNormalizationAlgorithmMethod()

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

Definition at line 229 of file SerializerUtils.cpp.

231 {
232  switch (normalizationAlgorithmMethod)
233  {
234  case armnn::NormalizationAlgorithmMethod::LocalBrightness:
235  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
236  case armnn::NormalizationAlgorithmMethod::LocalContrast:
237  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
238  default:
239  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
240  }
241 }
armnn::NormalizationAlgorithmMethod::LocalContrast
@ LocalContrast
Jarret 2009: Local Contrast Normalization.
armnn::NormalizationAlgorithmMethod::LocalBrightness
@ LocalBrightness
Krichevsky 2012: Local Brightness Normalization.

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

◆ GetFlatBufferOutputShapeRounding()

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

Definition at line 178 of file SerializerUtils.cpp.

179 {
180  switch (outputShapeRounding)
181  {
182  case armnn::OutputShapeRounding::Ceiling:
183  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
184  case armnn::OutputShapeRounding::Floor:
185  default:
186  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
187  }
188 }

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

◆ GetFlatBufferPaddingMethod()

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

Definition at line 190 of file SerializerUtils.cpp.

191 {
192  switch (paddingMethod)
193  {
194  case armnn::PaddingMethod::IgnoreValue:
195  return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
196  case armnn::PaddingMethod::Exclude:
197  default:
198  return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
199  }
200 }
armnn::PaddingMethod::Exclude
@ Exclude
The padding fields don't count and are ignored.
armnn::PaddingMethod::IgnoreValue
@ IgnoreValue
The padding fields count, but are ignored.

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

◆ GetFlatBufferPaddingMode()

armnnSerializer::PaddingMode GetFlatBufferPaddingMode ( armnn::PaddingMode  paddingMode)

Definition at line 202 of file SerializerUtils.cpp.

203 {
204  switch (paddingMode)
205  {
206  case armnn::PaddingMode::Reflect:
207  return armnnSerializer::PaddingMode::PaddingMode_Reflect;
208  case armnn::PaddingMode::Symmetric:
209  return armnnSerializer::PaddingMode::PaddingMode_Symmetric;
210  default:
211  return armnnSerializer::PaddingMode::PaddingMode_Constant;
212  }
213 }

References armnn::Reflect, and armnn::Symmetric.

◆ GetFlatBufferPoolingAlgorithm()

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

Definition at line 164 of file SerializerUtils.cpp.

165 {
166  switch (poolingAlgorithm)
167  {
168  case armnn::PoolingAlgorithm::Average:
169  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
170  case armnn::PoolingAlgorithm::L2:
171  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
172  case armnn::PoolingAlgorithm::Max:
173  default:
174  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
175  }
176 }

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

◆ GetFlatBufferReduceOperation()

armnnSerializer::ReduceOperation GetFlatBufferReduceOperation ( armnn::ReduceOperation  reduceOperation)

Definition at line 256 of file SerializerUtils.cpp.

257 {
258  switch (reduceOperation)
259  {
260  case armnn::ReduceOperation::Sum:
261  return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
262  case armnn::ReduceOperation::Max:
263  return armnnSerializer::ReduceOperation::ReduceOperation_Max;
264  case armnn::ReduceOperation::Mean:
265  return armnnSerializer::ReduceOperation::ReduceOperation_Mean;
266  case armnn::ReduceOperation::Min:
267  return armnnSerializer::ReduceOperation::ReduceOperation_Min;
268  case armnn::ReduceOperation::Prod:
269  return armnnSerializer::ReduceOperation::ReduceOperation_Prod;
270  default:
271  return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
272  }
273 }

References armnn::Max, armnn::Mean, armnn::Min, armnn::Prod, and armnn::Sum.

◆ GetFlatBufferResizeMethod()

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

Definition at line 243 of file SerializerUtils.cpp.

244 {
245  switch (method)
246  {
247  case armnn::ResizeMethod::NearestNeighbor:
248  return armnnSerializer::ResizeMethod_NearestNeighbor;
249  case armnn::ResizeMethod::Bilinear:
250  return armnnSerializer::ResizeMethod_Bilinear;
251  default:
252  return armnnSerializer::ResizeMethod_NearestNeighbor;
253  }
254 }

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

◆ GetFlatBufferScatterNdFunction()

serializer::ScatterNdFunction armnnSerializer::GetFlatBufferScatterNdFunction ( armnn::ScatterNdFunction  function)

Definition at line 100 of file Serializer.cpp.

101 {
102  switch (function)
103  {
104  case armnn::ScatterNdFunction::Update:
105  return serializer::ScatterNdFunction::ScatterNdFunction_Update;
106  case armnn::ScatterNdFunction::Add:
107  return serializer::ScatterNdFunction::ScatterNdFunction_Add;
108  case armnn::ScatterNdFunction::Sub:
109  return serializer::ScatterNdFunction::ScatterNdFunction_Sub;
110  case armnn::ScatterNdFunction::Max:
111  return serializer::ScatterNdFunction::ScatterNdFunction_Max;
112  case armnn::ScatterNdFunction::Min:
113  return serializer::ScatterNdFunction::ScatterNdFunction_Min;
114  default:
115  return serializer::ScatterNdFunction::ScatterNdFunction_Update;
116  }
117 }

References armnn::Add, armnn::Max, armnn::Min, armnn::Sub, and armnn::Update.

◆ GetFlatBufferUnaryOperation()

armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation ( armnn::UnaryOperation  unaryOperation)

Definition at line 137 of file SerializerUtils.cpp.

138 {
139  switch (unaryOperation)
140  {
141  case armnn::UnaryOperation::Abs:
142  return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
143  case armnn::UnaryOperation::Ceil:
144  return armnnSerializer::UnaryOperation::UnaryOperation_Ceil;
145  case armnn::UnaryOperation::Rsqrt:
146  return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
147  case armnn::UnaryOperation::Sqrt:
148  return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
149  case armnn::UnaryOperation::Exp:
150  return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
151  case armnn::UnaryOperation::Neg:
152  return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
153  case armnn::UnaryOperation::LogicalNot:
154  return armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot;
155  case armnn::UnaryOperation::Log:
156  return armnnSerializer::UnaryOperation::UnaryOperation_Log;
157  case armnn::UnaryOperation::Sin:
158  return armnnSerializer::UnaryOperation::UnaryOperation_Sin;
159  default:
160  throw armnn::InvalidArgumentException("Elementwise Unary operation unknown");
161  }
162 }

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