ArmNN
 25.11
Loading...
Searching...
No Matches
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 GetFlatBufferActivationFunction ( armnn::ActivationFunction function)

Definition at line 55 of file Serializer.cpp.

56{
57 switch (function)
58 {
60 return serializer::ActivationFunction::ActivationFunction_Sigmoid;
62 return serializer::ActivationFunction::ActivationFunction_TanH;
64 return serializer::ActivationFunction::ActivationFunction_Linear;
66 return serializer::ActivationFunction::ActivationFunction_ReLu;
68 return serializer::ActivationFunction::ActivationFunction_BoundedReLu;
70 return serializer::ActivationFunction::ActivationFunction_LeakyReLu;
72 return serializer::ActivationFunction::ActivationFunction_Abs;
74 return serializer::ActivationFunction::ActivationFunction_Sqrt;
76 return serializer::ActivationFunction::ActivationFunction_Square;
78 return serializer::ActivationFunction::ActivationFunction_Elu;
80 return serializer::ActivationFunction::ActivationFunction_HardSwish;
82 return serializer::ActivationFunction::ActivationFunction_Gelu;
83 default:
84 return serializer::ActivationFunction::ActivationFunction_Sigmoid;
85 }
86}
@ BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.
Definition Types.hpp:92

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

Referenced by GetFlatBufferActivationFunction().

◆ GetFlatBufferArgMinMaxFunction()

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

Definition at line 88 of file Serializer.cpp.

89{
90 switch (function)
91 {
93 return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max;
95 default:
96 return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min;
97 }
98}

References GetFlatBufferArgMinMaxFunction(), armnn::Max, and armnn::Min.

Referenced by GetFlatBufferArgMinMaxFunction().

◆ GetFlatBufferBinaryOperation()

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

Definition at line 110 of file SerializerUtils.cpp.

111{
112 switch (binaryOperation)
113 {
115 return armnnSerializer::BinaryOperation::BinaryOperation_Add;
117 return armnnSerializer::BinaryOperation::BinaryOperation_Div;
119 return armnnSerializer::BinaryOperation::BinaryOperation_Div;
121 return armnnSerializer::BinaryOperation::BinaryOperation_Maximum;
123 return armnnSerializer::BinaryOperation::BinaryOperation_Minimum;
125 return armnnSerializer::BinaryOperation::BinaryOperation_Mul;
127 return armnnSerializer::BinaryOperation::BinaryOperation_Sub;
129 return armnnSerializer::BinaryOperation::BinaryOperation_SqDiff;
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 {
16 return armnnSerializer::ComparisonOperation::ComparisonOperation_Equal;
18 return armnnSerializer::ComparisonOperation::ComparisonOperation_Greater;
20 return armnnSerializer::ComparisonOperation::ComparisonOperation_GreaterOrEqual;
22 return armnnSerializer::ComparisonOperation::ComparisonOperation_Less;
24 return armnnSerializer::ComparisonOperation::ComparisonOperation_LessOrEqual;
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 {
51 return armnnSerializer::ConstTensorData::ConstTensorData_IntData;
54 return armnnSerializer::ConstTensorData::ConstTensorData_ShortData;
59 return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
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 {
99 return armnnSerializer::DataLayout::DataLayout_NHWC;
101 return armnnSerializer::DataLayout::DataLayout_NDHWC;
103 return armnnSerializer::DataLayout::DataLayout_NCDHW;
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 {
72 return armnnSerializer::DataType::DataType_Float32;
74 return armnnSerializer::DataType::DataType_Float16;
76 return armnnSerializer::DataType::DataType_Signed32;
78 return armnnSerializer::DataType::DataType_Signed64;
80 return armnnSerializer::DataType::DataType_QSymmS16;
82 return armnnSerializer::DataType::DataType_QAsymmS8;
84 return armnnSerializer::DataType::DataType_QAsymmU8;
86 return armnnSerializer::DataType::DataType_QSymmS8;
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 {
37 return armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalAnd;
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 {
221 return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
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 {
235 return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
237 return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
238 default:
239 return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
240 }
241}
@ LocalContrast
Jarret 2009: Local Contrast Normalization.
Definition Types.hpp:219
@ LocalBrightness
Krichevsky 2012: Local Brightness Normalization.
Definition Types.hpp:217

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 {
183 return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
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 {
195 return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
197 default:
198 return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
199 }
200}
@ Exclude
The padding fields don't count and are ignored.
Definition Types.hpp:194
@ IgnoreValue
The padding fields count, but are ignored.
Definition Types.hpp:192

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 {
207 return armnnSerializer::PaddingMode::PaddingMode_Reflect;
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 {
169 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
171 return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
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 {
261 return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
263 return armnnSerializer::ReduceOperation::ReduceOperation_Max;
265 return armnnSerializer::ReduceOperation::ReduceOperation_Mean;
267 return armnnSerializer::ReduceOperation::ReduceOperation_Min;
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 {
248 return armnnSerializer::ResizeMethod_NearestNeighbor;
250 return armnnSerializer::ResizeMethod_Bilinear;
251 default:
252 return armnnSerializer::ResizeMethod_NearestNeighbor;
253 }
254}

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

◆ GetFlatBufferScatterNdFunction()

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

Definition at line 100 of file Serializer.cpp.

101{
102 switch (function)
103 {
105 return serializer::ScatterNdFunction::ScatterNdFunction_Update;
107 return serializer::ScatterNdFunction::ScatterNdFunction_Add;
109 return serializer::ScatterNdFunction::ScatterNdFunction_Sub;
111 return serializer::ScatterNdFunction::ScatterNdFunction_Max;
113 return serializer::ScatterNdFunction::ScatterNdFunction_Min;
114 default:
115 return serializer::ScatterNdFunction::ScatterNdFunction_Update;
116 }
117}

References armnn::Add, GetFlatBufferScatterNdFunction(), armnn::Max, armnn::Min, armnn::Sub, and armnn::Update.

Referenced by GetFlatBufferScatterNdFunction().

◆ GetFlatBufferUnaryOperation()

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

Definition at line 137 of file SerializerUtils.cpp.

138{
139 switch (unaryOperation)
140 {
142 return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
144 return armnnSerializer::UnaryOperation::UnaryOperation_Ceil;
146 return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
148 return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
150 return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
152 return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
154 return armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot;
156 return armnnSerializer::UnaryOperation::UnaryOperation_Log;
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.