| 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 | |
| 7 | #include <armnn/ArmNN.hpp> |
| 8 | |
| Matthew Bentham | 268509a | 2019-02-25 13:58:24 +0000 | [diff] [blame] | 9 | #include <ArmnnSchema_generated.h> |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 10 | |
| 11 | namespace armnnSerializer |
| 12 | { |
| 13 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 14 | armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType); |
| Mike Kelly | a0766c3 | 2019-02-19 17:22:07 +0000 | [diff] [blame] | 15 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 16 | armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 17 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 18 | armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 19 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 20 | armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 21 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 22 | armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding( |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 23 | armnn::OutputShapeRounding outputShapeRounding); |
| 24 | |
| Derek Lamberti | 0028d1b | 2019-02-20 13:57:42 +0000 | [diff] [blame] | 25 | armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod); |
| Saoirse Stewart | 3166c3e | 2019-02-18 15:24:53 +0000 | [diff] [blame] | 26 | |
| Nina Drozd | 5772878 | 2019-02-27 10:53:27 +0000 | [diff] [blame] | 27 | armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel( |
| 28 | armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel); |
| 29 | |
| 30 | armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod( |
| 31 | armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod); |
| 32 | |
| Matthew Bentham | 268509a | 2019-02-25 13:58:24 +0000 | [diff] [blame] | 33 | } // namespace armnnSerializer |