diff options
author | Nina Drozd <nina.drozd@arm.com> | 2019-02-27 10:53:27 +0000 |
---|---|---|
committer | Nina Drozd <nina.drozd@arm.com> | 2019-03-01 10:52:26 +0000 |
commit | 57728788f65656e4fa08923d12bee0de34a72fc7 (patch) | |
tree | 4d84f00762483cb26f6c2e880a07eb8b7699b7da /src/armnnSerializer/SerializerUtils.cpp | |
parent | 377351e5420304668da92da4ee00a012923619d1 (diff) | |
download | armnn-57728788f65656e4fa08923d12bee0de34a72fc7.tar.gz |
IVGCVSW-2700 Serialize/de-serialize the Normalization layer
Change-Id: Ib307ec6c28beb6c158d337678e67a2484c495a06
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r-- | src/armnnSerializer/SerializerUtils.cpp | 28 |
1 files changed, 28 insertions, 0 deletions
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 |