1 // 2 // Copyright © 2017,2019-2021,2023 Arm Ltd and Contributors. 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::BinaryOperation GetFlatBufferBinaryOperation(armnn::BinaryOperation binaryOperation); 22 23 armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation(armnn::UnaryOperation unaryOperation); 24 25 armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm); 26 27 armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding( 28 armnn::OutputShapeRounding outputShapeRounding); 29 30 armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod); 31 32 armnnSerializer::PaddingMode GetFlatBufferPaddingMode(armnn::PaddingMode paddingMode); 33 34 armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel( 35 armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel); 36 37 armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod( 38 armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod); 39 40 armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod method); 41 42 armnnSerializer::LogicalBinaryOperation GetFlatBufferLogicalBinaryOperation( 43 armnn::LogicalBinaryOperation logicalBinaryOperation); 44 45 armnnSerializer::ReduceOperation GetFlatBufferReduceOperation(armnn::ReduceOperation reduceOperation); 46 47 } // namespace armnnSerializer 48