ArmNN
 20.02
SerializerUtils.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <armnn/Types.hpp>
8 #include <ArmnnSchema_generated.h>
9 
10 namespace armnnSerializer
11 {
12 
14 
15 armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType);
16 
18 
20 
22 
24 
26  armnn::OutputShapeRounding outputShapeRounding);
27 
29 
31  armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel);
32 
34  armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod);
35 
37 
38 } // namespace armnnSerializer
armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation(armnn::UnaryOperation comparisonOperation)
DataLayout
Definition: Types.hpp:49
armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding(armnn::OutputShapeRounding outputShapeRounding)
NormalizationAlgorithmChannel
Definition: Types.hpp:126
armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod(armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod)
PoolingAlgorithm
Definition: Types.hpp:96
armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType)
armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType)
PaddingMethod
The padding method modifies the output of pooling layers.
Definition: Types.hpp:118
ComparisonOperation
Definition: Types.hpp:77
armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel(armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel)
DataType
Definition: Types.hpp:32
armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout)
OutputShapeRounding
Definition: Types.hpp:140
armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation(armnn::ComparisonOperation comparisonOperation)
armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod method)
ResizeMethod
Definition: Types.hpp:103
UnaryOperation
Definition: Types.hpp:87
armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod)
NormalizationAlgorithmMethod
Definition: Types.hpp:132
armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm)