diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 5a001de545..1204ad8f0f 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -121,7 +121,8 @@ enum LayerType : uint { Dequantize = 36, Merge = 37, Switch = 38, - Concat = 39 + Concat = 39, + SpaceToDepth = 40 } // Base layer table to be used as part of other layers @@ -338,6 +339,16 @@ table SpaceToBatchNdDescriptor { dataLayout:DataLayout; } +table SpaceToDepthLayer { + base:LayerBase; + descriptor:SpaceToDepthDescriptor; +} + +table SpaceToDepthDescriptor { + blockSize:uint; + dataLayout:DataLayout; +} + table SubtractionLayer { base:LayerBase; } @@ -584,7 +595,8 @@ union Layer { DequantizeLayer, MergeLayer, SwitchLayer, - ConcatLayer + ConcatLayer, + SpaceToDepthLayer } table AnyLayer { |