ArmNN
 22.11
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)
 
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::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.

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 GetFlatBufferArgMinMaxFunction().

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;
81  default:
82  return serializer::ActivationFunction::ActivationFunction_Sigmoid;
83  }
84 }
min(a, max(b, input)) ReLu1 & ReLu6.

◆ GetFlatBufferArgMinMaxFunction()

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

Definition at line 86 of file Serializer.cpp.

References armnn::BFloat16, armnn::Boolean, armnn::Float16, armnn::Float32, OriginsDescriptor::GetConcatAxis(), TensorInfo::GetDataType(), TensorShape::GetDimensionality(), TensorShape::GetDimensionSpecificity(), GetFlatBufferActivationFunction(), GetFlatBufferComparisonOperation(), GetFlatBufferConstTensorData(), GetFlatBufferDataLayout(), GetFlatBufferDataType(), GetFlatBufferLogicalBinaryOperation(), GetFlatBufferNormalizationAlgorithmChannel(), GetFlatBufferNormalizationAlgorithmMethod(), GetFlatBufferOutputShapeRounding(), GetFlatBufferPaddingMethod(), GetFlatBufferPaddingMode(), GetFlatBufferPoolingAlgorithm(), GetFlatBufferReduceOperation(), GetFlatBufferResizeMethod(), GetFlatBufferUnaryOperation(), IConnectableLayer::GetGuid(), BaseTensor< MemoryType >::GetInfo(), BaseTensor< MemoryType >::GetMemoryArea(), IConnectableLayer::GetName(), BaseTensor< MemoryType >::GetNumBytes(), TensorShape::GetNumDimensions(), OriginsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumDimensions(), OriginsDescriptor::GetNumViews(), ViewsDescriptor::GetNumViews(), ViewsDescriptor::GetOrigins(), TensorInfo::GetQuantizationDim(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetQuantizationScales(), TensorInfo::GetShape(), PermutationVector::GetSize(), OriginsDescriptor::GetViewOrigin(), ViewsDescriptor::GetViewOrigin(), ViewsDescriptor::GetViewSizes(), TensorInfo::HasPerAxisQuantization(), armnn::IgnoreUnused(), ActivationDescriptor::m_A, LstmDescriptor::m_ActivationFunc, BatchMatMulDescriptor::m_AdjointX, BatchMatMulDescriptor::m_AdjointY, ResizeDescriptor::m_AlignCorners, NormalizationDescriptor::m_Alpha, ArgMinMaxDescriptor::m_Axis, SoftmaxDescriptor::m_Axis, GatherDescriptor::m_Axis, MeanDescriptor::m_Axis, StackDescriptor::m_Axis, ChannelShuffleDescriptor::m_Axis, ActivationDescriptor::m_B, SliceDescriptor::m_Begin, StridedSliceDescriptor::m_Begin, StridedSliceDescriptor::m_BeginMask, SoftmaxDescriptor::m_Beta, NormalizationDescriptor::m_Beta, InstanceNormalizationDescriptor::m_Beta, FullyConnectedDescriptor::m_BiasEnabled, Convolution2dDescriptor::m_BiasEnabled, Convolution3dDescriptor::m_BiasEnabled, DepthwiseConvolution2dDescriptor::m_BiasEnabled, TransposeConvolution2dDescriptor::m_BiasEnabled, BatchToSpaceNdDescriptor::m_BlockShape, SpaceToBatchNdDescriptor::m_BlockShape, SpaceToDepthDescriptor::m_BlockSize, QLstmDescriptor::m_CellClip, QLstmDescriptor::m_CellIntermediateScale, LstmDescriptor::m_CifgEnabled, QLstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, FullyConnectedDescriptor::m_ConstantWeights, BatchToSpaceNdDescriptor::m_Crops, Pooling2dDescriptor::m_DataLayout, Pooling3dDescriptor::m_DataLayout, Convolution2dDescriptor::m_DataLayout, Convolution3dDescriptor::m_DataLayout, DepthwiseConvolution2dDescriptor::m_DataLayout, NormalizationDescriptor::m_DataLayout, L2NormalizationDescriptor::m_DataLayout, BatchNormalizationDescriptor::m_DataLayout, InstanceNormalizationDescriptor::m_DataLayout, BatchToSpaceNdDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, SpaceToBatchNdDescriptor::m_DataLayout, SpaceToDepthDescriptor::m_DataLayout, StridedSliceDescriptor::m_DataLayout, TransposeConvolution2dDescriptor::m_DataLayout, BatchMatMulDescriptor::m_DataLayoutX, BatchMatMulDescriptor::m_DataLayoutY, DetectionPostProcessDescriptor::m_DetectionsPerClass, Convolution2dDescriptor::m_DilationX, Convolution3dDescriptor::m_DilationX, DepthwiseConvolution2dDescriptor::m_DilationX, Convolution2dDescriptor::m_DilationY, Convolution3dDescriptor::m_DilationY, DepthwiseConvolution2dDescriptor::m_DilationY, Convolution3dDescriptor::m_DilationZ, PermuteDescriptor::m_DimMappings, TransposeDescriptor::m_DimMappings, StridedSliceDescriptor::m_EllipsisMask, StridedSliceDescriptor::m_End, StridedSliceDescriptor::m_EndMask, L2NormalizationDescriptor::m_Eps, BatchNormalizationDescriptor::m_Eps, InstanceNormalizationDescriptor::m_Eps, QLstmDescriptor::m_ForgetIntermediateScale, ActivationDescriptor::m_Function, ArgMinMaxDescriptor::m_Function, InstanceNormalizationDescriptor::m_Gamma, ResizeDescriptor::m_HalfPixelCenters, QLstmDescriptor::m_HiddenStateScale, QLstmDescriptor::m_HiddenStateZeroPoint, QLstmDescriptor::m_InputIntermediateScale, StackDescriptor::m_InputShape, NormalizationDescriptor::m_K, MeanDescriptor::m_KeepDims, ReduceDescriptor::m_KeepDims, LstmDescriptor::m_LayerNormEnabled, QLstmDescriptor::m_LayerNormEnabled, DetectionPostProcessDescriptor::m_MaxClassesPerDetection, DetectionPostProcessDescriptor::m_MaxDetections, ResizeDescriptor::m_Method, StridedSliceDescriptor::m_NewAxisMask, DetectionPostProcessDescriptor::m_NmsIouThreshold, DetectionPostProcessDescriptor::m_NmsScoreThreshold, NormalizationDescriptor::m_NormChannelType, NormalizationDescriptor::m_NormMethodType, NormalizationDescriptor::m_NormSize, DetectionPostProcessDescriptor::m_NumClasses, ChannelShuffleDescriptor::m_NumGroups, StackDescriptor::m_NumInputs, StandInDescriptor::m_NumInputs, StandInDescriptor::m_NumOutputs, ComparisonDescriptor::m_Operation, ElementwiseUnaryDescriptor::m_Operation, LogicalBinaryDescriptor::m_Operation, QLstmDescriptor::m_OutputIntermediateScale, Pooling2dDescriptor::m_OutputShapeRounding, Pooling3dDescriptor::m_OutputShapeRounding, Pooling3dDescriptor::m_PadBack, Convolution3dDescriptor::m_PadBack, Pooling2dDescriptor::m_PadBottom, Pooling3dDescriptor::m_PadBottom, Convolution2dDescriptor::m_PadBottom, Convolution3dDescriptor::m_PadBottom, DepthwiseConvolution2dDescriptor::m_PadBottom, TransposeConvolution2dDescriptor::m_PadBottom, Pooling2dDescriptor::m_PaddingMethod, Pooling3dDescriptor::m_PaddingMethod, PadDescriptor::m_PaddingMode, Pooling3dDescriptor::m_PadFront, Convolution3dDescriptor::m_PadFront, Pooling2dDescriptor::m_PadLeft, Pooling3dDescriptor::m_PadLeft, Convolution2dDescriptor::m_PadLeft, Convolution3dDescriptor::m_PadLeft, DepthwiseConvolution2dDescriptor::m_PadLeft, TransposeConvolution2dDescriptor::m_PadLeft, SpaceToBatchNdDescriptor::m_PadList, PadDescriptor::m_PadList, Pooling2dDescriptor::m_PadRight, Pooling3dDescriptor::m_PadRight, Convolution2dDescriptor::m_PadRight, Convolution3dDescriptor::m_PadRight, DepthwiseConvolution2dDescriptor::m_PadRight, TransposeConvolution2dDescriptor::m_PadRight, Pooling2dDescriptor::m_PadTop, Pooling3dDescriptor::m_PadTop, Convolution2dDescriptor::m_PadTop, Convolution3dDescriptor::m_PadTop, DepthwiseConvolution2dDescriptor::m_PadTop, TransposeConvolution2dDescriptor::m_PadTop, PadDescriptor::m_PadValue, LstmDescriptor::m_PeepholeEnabled, QLstmDescriptor::m_PeepholeEnabled, Pooling3dDescriptor::m_PoolDepth, Pooling2dDescriptor::m_PoolHeight, Pooling3dDescriptor::m_PoolHeight, Pooling2dDescriptor::m_PoolType, Pooling3dDescriptor::m_PoolType, Pooling2dDescriptor::m_PoolWidth, Pooling3dDescriptor::m_PoolWidth, QLstmDescriptor::m_ProjectionClip, LstmDescriptor::m_ProjectionEnabled, QLstmDescriptor::m_ProjectionEnabled, ReduceDescriptor::m_ReduceOperation, DetectionPostProcessDescriptor::m_ScaleH, DetectionPostProcessDescriptor::m_ScaleW, DetectionPostProcessDescriptor::m_ScaleX, DetectionPostProcessDescriptor::m_ScaleY, StridedSliceDescriptor::m_ShrinkAxisMask, SliceDescriptor::m_Size, StridedSliceDescriptor::m_Stride, Pooling2dDescriptor::m_StrideX, Pooling3dDescriptor::m_StrideX, Convolution2dDescriptor::m_StrideX, Convolution3dDescriptor::m_StrideX, DepthwiseConvolution2dDescriptor::m_StrideX, TransposeConvolution2dDescriptor::m_StrideX, Pooling2dDescriptor::m_StrideY, Pooling3dDescriptor::m_StrideY, Convolution2dDescriptor::m_StrideY, Convolution3dDescriptor::m_StrideY, DepthwiseConvolution2dDescriptor::m_StrideY, TransposeConvolution2dDescriptor::m_StrideY, Pooling3dDescriptor::m_StrideZ, Convolution3dDescriptor::m_StrideZ, ResizeDescriptor::m_TargetHeight, ReshapeDescriptor::m_TargetShape, ResizeDescriptor::m_TargetWidth, LstmDescriptor::m_TimeMajor, FullyConnectedDescriptor::m_TransposeWeightMatrix, BatchMatMulDescriptor::m_TransposeX, BatchMatMulDescriptor::m_TransposeY, DetectionPostProcessDescriptor::m_UseRegularNms, FillDescriptor::m_Value, ReduceDescriptor::m_vAxis, armnn::Max, armnn::Min, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, armnn::QSymmS8, armnn::Signed32, armnn::Signed64, and OptionalReferenceSwitch< IsReference, T >::value().

87 {
88  switch (function)
89  {
91  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Max;
93  default:
94  return serializer::ArgMinMaxFunction::ArgMinMaxFunction_Min;
95  }
96 }

◆ 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 GetFlatBufferArgMinMaxFunction().

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 }

◆ GetFlatBufferConstTensorData()

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

Definition at line 45 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

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 }

◆ GetFlatBufferDataLayout()

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

Definition at line 94 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

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 }

◆ GetFlatBufferDataType()

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

Definition at line 67 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

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 }

◆ GetFlatBufferLogicalBinaryOperation()

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

Definition at line 31 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

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 }

◆ GetFlatBufferNormalizationAlgorithmChannel()

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

Definition at line 186 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

188 {
189  switch (normalizationAlgorithmChannel)
190  {
192  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
194  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within;
195  default:
196  return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across;
197  }
198 }

◆ GetFlatBufferNormalizationAlgorithmMethod()

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

Definition at line 200 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

202 {
203  switch (normalizationAlgorithmMethod)
204  {
206  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
208  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast;
209  default:
210  return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness;
211  }
212 }
Jarret 2009: Local Contrast Normalization.
Krichevsky 2012: Local Brightness Normalization.

◆ GetFlatBufferOutputShapeRounding()

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

Definition at line 149 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

150 {
151  switch (outputShapeRounding)
152  {
154  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Ceiling;
156  default:
157  return armnnSerializer::OutputShapeRounding::OutputShapeRounding_Floor;
158  }
159 }

◆ GetFlatBufferPaddingMethod()

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

Definition at line 161 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

162 {
163  switch (paddingMethod)
164  {
166  return armnnSerializer::PaddingMethod::PaddingMethod_IgnoreValue;
168  default:
169  return armnnSerializer::PaddingMethod::PaddingMethod_Exclude;
170  }
171 }
The padding fields don&#39;t count and are ignored.
The padding fields count, but are ignored.

◆ GetFlatBufferPaddingMode()

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

Definition at line 173 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

174 {
175  switch (paddingMode)
176  {
178  return armnnSerializer::PaddingMode::PaddingMode_Reflect;
180  return armnnSerializer::PaddingMode::PaddingMode_Symmetric;
181  default:
182  return armnnSerializer::PaddingMode::PaddingMode_Constant;
183  }
184 }

◆ GetFlatBufferPoolingAlgorithm()

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

Definition at line 135 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

136 {
137  switch (poolingAlgorithm)
138  {
140  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Average;
142  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_L2;
144  default:
145  return armnnSerializer::PoolingAlgorithm::PoolingAlgorithm_Max;
146  }
147 }

◆ GetFlatBufferReduceOperation()

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

Definition at line 227 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

228 {
229  switch (reduceOperation)
230  {
232  return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
234  return armnnSerializer::ReduceOperation::ReduceOperation_Max;
236  return armnnSerializer::ReduceOperation::ReduceOperation_Mean;
238  return armnnSerializer::ReduceOperation::ReduceOperation_Min;
240  return armnnSerializer::ReduceOperation::ReduceOperation_Prod;
241  default:
242  return armnnSerializer::ReduceOperation::ReduceOperation_Sum;
243  }
244 }

◆ GetFlatBufferResizeMethod()

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

Definition at line 214 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

215 {
216  switch (method)
217  {
219  return armnnSerializer::ResizeMethod_NearestNeighbor;
221  return armnnSerializer::ResizeMethod_Bilinear;
222  default:
223  return armnnSerializer::ResizeMethod_NearestNeighbor;
224  }
225 }

◆ GetFlatBufferUnaryOperation()

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

Definition at line 110 of file SerializerUtils.cpp.

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

Referenced by GetFlatBufferArgMinMaxFunction().

111 {
112  switch (comparisonOperation)
113  {
115  return armnnSerializer::UnaryOperation::UnaryOperation_Abs;
117  return armnnSerializer::UnaryOperation::UnaryOperation_Rsqrt;
119  return armnnSerializer::UnaryOperation::UnaryOperation_Sqrt;
121  return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
123  return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
125  return armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot;
127  return armnnSerializer::UnaryOperation::UnaryOperation_Log;
129  return armnnSerializer::UnaryOperation::UnaryOperation_Sin;
130  default:
131  throw armnn::InvalidArgumentException("Unary operation unknown");
132  }
133 }