From 57728788f65656e4fa08923d12bee0de34a72fc7 Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Wed, 27 Feb 2019 10:53:27 +0000 Subject: IVGCVSW-2700 Serialize/de-serialize the Normalization layer Change-Id: Ib307ec6c28beb6c158d337678e67a2484c495a06 Signed-off-by: Nina Drozd --- src/armnnSerializer/SerializerUtils.cpp | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) (limited to 'src/armnnSerializer/SerializerUtils.cpp') diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp index 592f258b81..bfe795c8c4 100644 --- a/src/armnnSerializer/SerializerUtils.cpp +++ b/src/armnnSerializer/SerializerUtils.cpp @@ -96,4 +96,32 @@ armnnSerializer::PaddingMethod GetFlatBufferPaddingMethod(armnn::PaddingMethod p } } +armnnSerializer::NormalizationAlgorithmChannel GetFlatBufferNormalizationAlgorithmChannel( + armnn::NormalizationAlgorithmChannel normalizationAlgorithmChannel) +{ + switch (normalizationAlgorithmChannel) + { + case armnn::NormalizationAlgorithmChannel::Across: + return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across; + case armnn::NormalizationAlgorithmChannel::Within: + return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Within; + default: + return armnnSerializer::NormalizationAlgorithmChannel::NormalizationAlgorithmChannel_Across; + } +} + +armnnSerializer::NormalizationAlgorithmMethod GetFlatBufferNormalizationAlgorithmMethod( + armnn::NormalizationAlgorithmMethod normalizationAlgorithmMethod) +{ + switch (normalizationAlgorithmMethod) + { + case armnn::NormalizationAlgorithmMethod::LocalBrightness: + return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness; + case armnn::NormalizationAlgorithmMethod::LocalContrast: + return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalContrast; + default: + return armnnSerializer::NormalizationAlgorithmMethod::NormalizationAlgorithmMethod_LocalBrightness; + } +} + } // namespace armnnSerializer \ No newline at end of file -- cgit v1.2.1