diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-02-21 09:01:41 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-02-21 09:57:44 +0000 |
commit | dbb0c0ca0c8425886ee3a2095e0ced07099134f9 (patch) | |
tree | 2d85dd36b0c23c2fda9f6898a7ab8a9f74f4520c /src/armnnSerializer/Schema.fbs | |
parent | 5324782821dce525cf7c1636c659f998fae2fb85 (diff) | |
download | armnn-dbb0c0ca0c8425886ee3a2095e0ced07099134f9.tar.gz |
IVGCVSW-2639 Add Serializer & Deserializer for Fully Connected
* Added FullyConnectedLayer to Serializer Schema Schema.fbs
* Added FullyConnected serialization and deserialization support
* Added FullyConnected serialization and deserialization unit tests
Change-Id: I8ef14f9728158f849fa4d1a8d05a1a4170cd5b41
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
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, |