diff options
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 362dd5c653..c8ffce48bc 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -180,6 +180,7 @@ enum LayerType : uint { UnidirectionalSequenceLstm = 63, ChannelShuffle = 64, Convolution3d = 65, + Pooling3d = 66, } // Base layer table to be used as part of other layers @@ -453,6 +454,11 @@ table Pooling2dLayer { descriptor:Pooling2dDescriptor; } +table Pooling3dLayer { + base:LayerBase; + descriptor:Pooling3dDescriptor; +} + enum PoolingAlgorithm : byte { Max = 0, Average = 1, @@ -484,6 +490,25 @@ table Pooling2dDescriptor { dataLayout:DataLayout; } +table Pooling3dDescriptor { + poolType:PoolingAlgorithm; + padLeft:uint; + padRight:uint; + padTop:uint; + padBottom:uint; + padFront:uint; + padBack:uint; + poolWidth:uint; + poolHeight:uint; + poolDepth:uint; + strideX:uint; + strideY:uint; + strideZ:uint; + outputShapeRounding:OutputShapeRounding; + paddingMethod:PaddingMethod; + dataLayout:DataLayout; +} + table QuantizeLayer { base:LayerBase; } @@ -1046,6 +1071,7 @@ union Layer { UnidirectionalSequenceLstmLayer, ChannelShuffleLayer, Convolution3dLayer, + Pooling3dLayer, } table AnyLayer { |