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/ArmnnSchema.fbs | |
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/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 31 |
1 files changed, 29 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index b59adcf82b..cde0087d6f 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -99,7 +99,8 @@ enum LayerType : uint { Division = 15, Minimum = 16, Equal = 17, - Maximum = 18 + Maximum = 18, + Normalization = 19 } // Base layer table to be used as part of other layers @@ -298,6 +299,31 @@ table BatchToSpaceNdDescriptor { dataLayout:DataLayout; } +enum NormalizationAlgorithmChannel : byte { + Across = 0, + Within = 1 +} + +enum NormalizationAlgorithmMethod : byte { + LocalBrightness = 0, + LocalContrast = 1 +} + +table NormalizationLayer { + base:LayerBase; + descriptor:NormalizationDescriptor; +} + +table NormalizationDescriptor { + normChannelType:NormalizationAlgorithmChannel = Across; + normMethodType:NormalizationAlgorithmMethod = LocalBrightness; + normSize:uint; + alpha:float; + beta:float; + k:float; + dataLayout:DataLayout = NCHW; +} + union Layer { ActivationLayer, AdditionLayer, @@ -317,7 +343,8 @@ union Layer { DivisionLayer, MinimumLayer, EqualLayer, - MaximumLayer + MaximumLayer, + NormalizationLayer } table AnyLayer { |