diff options
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 |