diff options
author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-06-28 15:07:10 +0100 |
---|---|---|
committer | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-07-01 10:53:47 +0100 |
commit | 6fb339a7d202a9c64d8c7843d630fe8ab7be9f33 (patch) | |
tree | 08f7119b14c042bc009cae4ca90c0c49c5c15387 /src/armnnSerializer/SerializerUtils.cpp | |
parent | a9075df5b704e4f4432bf26027e3ba671d4596f0 (diff) | |
download | armnn-6fb339a7d202a9c64d8c7843d630fe8ab7be9f33.tar.gz |
IVGCVSW-3364 Add serialization support for Resize layer
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: I3b1af816cefc1760f63324f365de93f899c9c9ee
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index bfe795c8c4..9790d6e23b 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -124,4 +124,17 @@ armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorith } } +armnnSerializer::ResizeMethod GetFlatBufferResizeMethod(armnn::ResizeMethod method) +{ + switch (method) + { + case armnn::ResizeMethod::NearestNeighbor: + return armnnSerializer::ResizeMethod_NearestNeighbor; + case armnn::ResizeMethod::Bilinear: + return armnnSerializer::ResizeMethod_Bilinear; + default: + return armnnSerializer::ResizeMethod_NearestNeighbor; + } +} + } // namespace armnnSerializer
\ No newline at end of file |