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

References armnn::Boolean, armnn::Float16, armnn::Float32, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, 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::QSymmS8:
43  case armnn::DataType::Boolean:
44  return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
45  default:
46  return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
47  }
48 }

◆ GetFlatBufferDataLayout()

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

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

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

◆ GetFlatBufferDataType()

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

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

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

◆ GetFlatBufferNormalizationAlgorithmChannel()

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

Definition at line 144 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitNormalizationLayer().

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

◆ GetFlatBufferNormalizationAlgorithmMethod()

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

Definition at line 158 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitNormalizationLayer().

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

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

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

◆ GetFlatBufferPaddingMethod()

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

Definition at line 132 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

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

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

Referenced by SerializerVisitor::VisitPooling2dLayer().

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

◆ GetFlatBufferResizeMethod()

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

Definition at line 172 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitResizeLayer().

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

◆ GetFlatBufferUnaryOperation()

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

Definition at line 87 of file SerializerUtils.cpp.

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

Referenced by SerializerVisitor::VisitElementwiseUnaryLayer().

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