aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
authorTamas Nyiri <tamas.nyiri@arm.com>2021-11-05 14:55:33 +0000
committerTamas Nyiri <tamas.nyiri@arm.com>2021-11-17 11:32:55 +0000
commitd998a1cfbcc7b4ed9922532ffe19c51283fcae68 (patch)
tree53317946bb5845f4f253ebfd53b668c6698fcfee /src/armnnSerializer/ArmnnSchema.fbs
parent7b885b3cce70154596b1994b013ea91527117c26 (diff)
downloadarmnn-d998a1cfbcc7b4ed9922532ffe19c51283fcae68.tar.gz
IVGCVSW-6510 Serialization + Deserialization implementation
Subtask of story: IVGCVSW-6164 Add a Pooling3d FrontEnd and Ref Implementation * Add serialization support * Add deserialization support * Add corresponding unit tests Change-Id: I7cce5421f0a9b7c47a03524e733f3315131ba125 Signed-off-by: Tamas Nyiri <tamas.nyiri@arm.com>
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs26
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 {