diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-02-13 15:41:52 +0000 |
---|---|---|
committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-18 14:49:04 +0000 |
commit | fc413c0c977e6c9680a2aa6546e977be0a2efdb9 (patch) | |
tree | c8eaeef557d1a8a0e5a18a7104f5b6d308c9efc2 /src/armnnSerializer/Schema.fbs | |
parent | 2ee88dfe7096f8f571ae7be9cbf0f49ededd89af (diff) | |
download | armnn-fc413c0c977e6c9680a2aa6546e977be0a2efdb9.tar.gz |
IVGCVSW-2644 Add Serializer & Deserializer for Softmax
Change-Id: Ifea2108e173d2b602162fe53b880a68e1c715510
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 | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs index 5d6388d944..411b89ad8e 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -66,7 +66,8 @@ enum LayerType : uint { Addition = 0, Input = 1, Multiplication = 2, - Output = 3 + Output = 3, + Softmax = 4 } // Base layer table to be used as part of other layers @@ -96,6 +97,15 @@ table MultiplicationLayer { base:LayerBase; } +table SoftmaxLayer { + base:LayerBase; + descriptor:SoftmaxDescriptor; +} + +table SoftmaxDescriptor { + beta:float; +} + table OutputLayer { base:BindableLayerBase; } @@ -104,7 +114,8 @@ union Layer { AdditionLayer, InputLayer, MultiplicationLayer, - OutputLayer + OutputLayer, + SoftmaxLayer } table AnyLayer { |