ArmNN
 20.02
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::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 27 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().

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

◆ GetFlatBufferArgMinMaxFunction()

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

Definition at line 58 of file Serializer.cpp.

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

Referenced by SerializerVisitor::VisitArgMinMaxLayer().

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

◆ 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 31 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitQuantizedLstmLayer().

32 {
33  switch (dataType)
34  {
35  case armnn::DataType::Float32:
36  case armnn::DataType::Signed32:
37  return armnnSerializer::ConstTensorData::ConstTensorData_IntData;
38  case armnn::DataType::Float16:
39  case armnn::DataType::QSymmS16:
40  return armnnSerializer::ConstTensorData::ConstTensorData_ShortData;
41  case armnn::DataType::QAsymmU8:
42  case armnn::DataType::Boolean:
43  return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
44  default:
45  return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
46  }
47 }

◆ GetFlatBufferDataLayout()

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

Definition at line 72 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().

73 {
74  switch (dataLayout)
75  {
76  case armnn::DataLayout::NHWC:
77  return armnnSerializer::DataLayout::DataLayout_NHWC;
78  case armnn::DataLayout::NCHW:
79  default:
80  return armnnSerializer::DataLayout::DataLayout_NCHW;
81  }
82 }

◆ GetFlatBufferDataType()

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

Definition at line 49 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::GetVersionTable(), and SerializerVisitor::VisitQuantizedLstmLayer().

50 {
51  switch (dataType)
52  {
53  case armnn::DataType::Float32:
54  return armnnSerializer::DataType::DataType_Float32;
55  case armnn::DataType::Float16:
56  return armnnSerializer::DataType::DataType_Float16;
57  case armnn::DataType::Signed32:
58  return armnnSerializer::DataType::DataType_Signed32;
59  case armnn::DataType::QSymmS16:
60  return armnnSerializer::DataType::DataType_QSymmS16;
61  case armnn::DataType::QAsymmS8:
62  return armnnSerializer::DataType::DataType_QAsymmS8;
63  case armnn::DataType::QAsymmU8:
64  return armnnSerializer::DataType::DataType_QAsymmU8;
65  case armnn::DataType::Boolean:
66  return armnnSerializer::DataType::DataType_Boolean;
67  default:
68  return armnnSerializer::DataType::DataType_Float16;
69  }
70 }

◆ GetFlatBufferNormalizationAlgorithmChannel()

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

Definition at line 141 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitNormalizationLayer().

143 {
144  switch (normalizationAlgorithmChannel)
145  {
146  case armnn::NormalizationAlgorithmChannel::Across:
147  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
148  case armnn::NormalizationAlgorithmChannel::Within:
149  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
150  default:
151  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
152  }
153 }

◆ GetFlatBufferNormalizationAlgorithmMethod()

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

Definition at line 155 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitNormalizationLayer().

157 {
158  switch (normalizationAlgorithmMethod)
159  {
160  case armnn::NormalizationAlgorithmMethod::LocalBrightness:
161  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
162  case armnn::NormalizationAlgorithmMethod::LocalContrast:
163  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
164  default:
165  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
166  }
167 }
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 117 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

118 {
119  switch (outputShapeRounding)
120  {
121  case armnn::OutputShapeRounding::Ceiling:
122  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
123  case armnn::OutputShapeRounding::Floor:
124  default:
125  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
126  }
127 }

◆ GetFlatBufferPaddingMethod()

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

Definition at line 129 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

130 {
131  switch (paddingMethod)
132  {
133  case armnn::PaddingMethod::IgnoreValue:
134  return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
135  case armnn::PaddingMethod::Exclude:
136  default:
137  return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
138  }
139 }
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 103 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

104 {
105  switch (poolingAlgorithm)
106  {
107  case armnn::PoolingAlgorithm::Average:
108  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
109  case armnn::PoolingAlgorithm::L2:
110  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
111  case armnn::PoolingAlgorithm::Max:
112  default:
113  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
114  }
115 }

◆ GetFlatBufferResizeMethod()

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

Definition at line 169 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitResizeLayer().

170 {
171  switch (method)
172  {
173  case armnn::ResizeMethod::NearestNeighbor:
174  return armnnSerializer::ResizeMethod_NearestNeighbor;
175  case armnn::ResizeMethod::Bilinear:
176  return armnnSerializer::ResizeMethod_Bilinear;
177  default:
178  return armnnSerializer::ResizeMethod_NearestNeighbor;
179  }
180 }

◆ GetFlatBufferUnaryOperation()

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

Definition at line 84 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitElementwiseUnaryLayer().

85 {
86  switch (comparisonOperation)
87  {
88  case armnn::UnaryOperation::Abs:
89  return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
90  case armnn::UnaryOperation::Rsqrt:
91  return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
92  case armnn::UnaryOperation::Sqrt:
93  return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
94  case armnn::UnaryOperation::Exp:
95  return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
96  case armnn::UnaryOperation::Neg:
97  return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
98  default:
99  throw armnn::InvalidArgumentException("Unary operation unknown");
100  }
101 }