diff options
author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-18 15:24:53 +0000 |
---|---|---|
committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-19 11:52:27 +0000 |
commit | 3166c3edeb64d834ba27031ddd39b5b1f940b604 (patch) | |
tree | 2789010d0878d64442f51ba0edbd8f159d1a32a0 /src/armnnSerializer/SerializerUtils.hpp | |
parent | a6b504a8925174739f5a064cf77d1563cca38708 (diff) | |
download | armnn-3166c3edeb64d834ba27031ddd39b5b1f940b604.tar.gz |
IVGCVSW-2645 Add Serializer & Deserializer for Pooling2d
Change-Id: Iba41da3cccd539a0175f2ed0ff9a8b6a23c5fb6f
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
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 |