diff options
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.hpp')
-rw-r--r-- | src/armnnSerializer/SerializerUtils.hpp | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.hpp b/src/armnnSerializer/SerializerUtils.hpp new file mode 100644 index 0000000000..72a8806560 --- /dev/null +++ b/src/armnnSerializer/SerializerUtils.hpp @@ -0,0 +1,25 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// +#pragma once + +#include <armnn/ArmNN.hpp> + +#include <Schema_generated.h> + +namespace armnnSerializer +{ + +armnn::armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType); + +armnn::armnnSerializer::DataLayout GetFlatBufferDataLayout(armnn::DataLayout dataLayout); + +armnn::armnnSerializer::PoolingAlgorithm GetFlatBufferPoolingAlgorithm(armnn::PoolingAlgorithm poolingAlgorithm); + +armnn::armnnSerializer::OutputShapeRounding GetFlatBufferOutputShapeRounding( + armnn::OutputShapeRounding outputShapeRounding); + +armnn::armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod paddingMethod); + +} // namespace armnnSerializer
\ No newline at end of file |