aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/SerializerUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp13
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