commit fc413c0c977e6c9680a2aa6546e977be0a2efdb9
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Wed Feb 13 15:41:52 2019 +0000
committer Aron Virginas-Tar <aron.virginas-tar@arm.com> Mon Feb 18 14:49:04 2019 +0000
tree c8eaeef557d1a8a0e5a18a7104f5b6d308c9efc2
parent 2ee88dfe7096f8f571ae7be9cbf0f49ededd89af

IVGCVSW-2644 Add Serializer & Deserializer for Softmax

Change-Id: Ifea2108e173d2b602162fe53b880a68e1c715510
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>

src/armnnSerializer/Schema.fbs

diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs
index 5d6388d..411b89a 100644
--- a/src/armnnSerializer/Schema.fbs
+++ b/src/armnnSerializer/Schema.fbs
@@ -66,7 +66,8 @@
     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 @@
     base:LayerBase;
 }
 
+table SoftmaxLayer {
+    base:LayerBase;
+    descriptor:SoftmaxDescriptor;
+}
+
+table SoftmaxDescriptor {
+    beta:float;
+}
+
 table OutputLayer {
     base:BindableLayerBase;
 }
@@ -104,7 +114,8 @@
     AdditionLayer,
     InputLayer,
     MultiplicationLayer,
-    OutputLayer
+    OutputLayer,
+    SoftmaxLayer
 }
 
 table AnyLayer {