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 13 armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation(armnn::ComparisonOperation comparisonOperation); 14 15 armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType); 16 17 armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType); 18 19 armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout); 20 21 armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation(armnn::UnaryOperation unaryOperation); 22 23 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm); 24 25 armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding( 26 armnn::OutputShapeRounding outputShapeRounding); 27 28 armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod); 29 30 armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel( 31 armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel); 32 33 armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod( 34 armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod); 35 36 armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod method); 37 38 armnnSerializer::LogicalBinaryOperation GetFlatBufferLogicalBinaryOperation( 39 armnn::LogicalBinaryOperation logicalBinaryOperation); 40 41 } // namespace armnnSerializer 42