aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs9
1 files changed, 8 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index dc14069798..db70e7bd4c 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -92,7 +92,8 @@ enum LayerType : uint {
DepthwiseConvolution2d = 8,
Activation = 9,
Permute = 10,
- FullyConnected = 11
+ FullyConnected = 11,
+ Constant = 12
}
// Base layer table to be used as part of other layers
@@ -125,6 +126,11 @@ table AdditionLayer {
base:LayerBase;
}
+table ConstantLayer {
+ base:LayerBase;
+ input:ConstTensor;
+}
+
table Convolution2dLayer {
base:LayerBase;
descriptor:Convolution2dDescriptor;
@@ -251,6 +257,7 @@ table PermuteDescriptor {
union Layer {
ActivationLayer,
AdditionLayer,
+ ConstantLayer,
Convolution2dLayer,
DepthwiseConvolution2dLayer,
FullyConnectedLayer,