diff options
Diffstat (limited to 'src/armnnSerializer/Schema.fbs')
-rw-r--r-- | src/armnnSerializer/Schema.fbs | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs index 94ca23b0cd..dc14069798 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -91,7 +91,8 @@ enum LayerType : uint { Convolution2d = 7, DepthwiseConvolution2d = 8, Activation = 9, - Permute = 10 + Permute = 10, + FullyConnected = 11 } // Base layer table to be used as part of other layers @@ -142,6 +143,18 @@ table Convolution2dDescriptor { dataLayout:DataLayout = NCHW; } +table FullyConnectedLayer { + base:LayerBase; + descriptor:FullyConnectedDescriptor; + weights:ConstTensor; + biases:ConstTensor; +} + +table FullyConnectedDescriptor { + biasEnabled:bool = false; + transposeWeightsMatrix:bool = false; +} + table InputLayer { base:BindableLayerBase; } @@ -240,6 +253,7 @@ union Layer { AdditionLayer, Convolution2dLayer, DepthwiseConvolution2dLayer, + FullyConnectedLayer, InputLayer, MultiplicationLayer, OutputLayer, |