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