diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 4e3180f0ff..f416912618 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -103,7 +103,8 @@ enum LayerType : uint { Normalization = 19, Pad = 20, Rsqrt = 21, - Floor = 22 + Floor = 22, + BatchNormalization = 23 } // Base layer table to be used as part of other layers @@ -340,14 +341,30 @@ table PadDescriptor { padList:[uint]; } + table RsqrtLayer { base:LayerBase; } +table BatchNormalizationLayer { + base:LayerBase; + descriptor:BatchNormalizationDescriptor; + mean:ConstTensor; + variance:ConstTensor; + beta:ConstTensor; + gamma:ConstTensor; +} + +table BatchNormalizationDescriptor { + eps:float; + dataLayout:DataLayout; +} + union Layer { ActivationLayer, AdditionLayer, BatchToSpaceNdLayer, + BatchNormalizationLayer, ConstantLayer, Convolution2dLayer, DepthwiseConvolution2dLayer, |