diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index c1e4e3826a..7d64dca6f3 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -94,7 +94,8 @@ enum LayerType : uint { Permute = 10, FullyConnected = 11, Constant = 12, - SpaceToBatchNd = 13 + SpaceToBatchNd = 13, + BatchToSpaceNd = 14 } // Base layer table to be used as part of other layers @@ -266,9 +267,21 @@ table SpaceToBatchNdDescriptor { dataLayout:DataLayout; } +table BatchToSpaceNdLayer { + base:LayerBase; + descriptor:BatchToSpaceNdDescriptor; +} + +table BatchToSpaceNdDescriptor { + blockShape:[uint]; + crops:[uint]; + dataLayout:DataLayout; +} + union Layer { ActivationLayer, AdditionLayer, + BatchToSpaceNdLayer, ConstantLayer, Convolution2dLayer, DepthwiseConvolution2dLayer, |