| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2017 Arm Ltd. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | #pragma once |
| 6 | |
| Matthew Bentham | ff130e2 | 2020-01-17 11:47:42 +0000 | [diff] [blame] | 7 | #include <armnn/Types.hpp> |
| Matthew Bentham | 268509a | 2019-02-25 13:58:24 +0000 | [diff] [blame] | 8 | #include <ArmnnSchema_generated.h> |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 9 | |
| 10 | namespace armnnSerializer |
| 11 | { |
| 12 | |
| Aron Virginas-Tar | e80ebd1 | 2019-10-17 16:11:54 +0100 | [diff] [blame] | 13 | armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation(armnn::ComparisonOperation comparisonOperation); |
| 14 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 15 | armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType); |
| Mike Kelly | a0766c3 | 2019-02-19 17:22:07 +0000 | [diff] [blame] | 16 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 17 | armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 18 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 19 | armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 20 | |
| josh minor | 4a3c610 | 2020-01-06 16:40:46 -0600 | [diff] [blame^] | 21 | armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation(armnn::UnaryOperation unaryOperation); |
| 22 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 23 | armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 24 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 25 | armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding( |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 26 | armnn::OutputShapeRounding outputShapeRounding); |
| 27 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 28 | armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 29 | |
| Nina Drozd | 5772878 | 2019-02-27 10:53:27 +0000 | [diff] [blame] | 30 | armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel( |
| 31 | armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel); |
| 32 | |
| 33 | armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod( |
| 34 | armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod); |
| 35 | |
| FinnWilliamsArm | 6fb339a | 2019-06-28 15:07:10 +0100 | [diff] [blame] | 36 | armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod method); |
| 37 | |
| Matthew Bentham | 268509a | 2019-02-25 13:58:24 +0000 | [diff] [blame] | 38 | } // namespace armnnSerializer |