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