commit 2605b236d103e1ba27069e0d668599042a4761af
author Finn Williams <Finn.Williams@arm.com> Wed Jun 10 15:53:46 2020 +0100
committer Finn Williams <Finn.Williams@arm.com> Mon Jul 06 21:50:50 2020 +0100
tree dc0a408acb5749a1f51544fd5857452377d05aab
parent c5789ca2e432075e2c92e7e0d99139c5329280e6

IVGCVSW-4624 Add a RANK Reference Implementation

 * Add Rank front end
 * Add Rank reference implementation
 * Add Rank serialization support
 * Add Scalar serialization support

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I06e4a468c2a84e79bae2e6c5348596bbbf853b4b

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 6a388db..e1b6e1f 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -59,6 +59,7 @@
     quantizationOffset:int = 0;
     quantizationScales:[float];
     quantizationDim:uint;
+    dimensionality:uint = 1;
 }
 
 struct Connection {
@@ -157,7 +158,8 @@
     ElementwiseUnary = 54,
     Transpose = 55,
     QLstm = 56,
-    Fill = 57
+    Fill = 57,
+    Rank = 58
 }
 
 // Base layer table to be used as part of other layers
@@ -859,6 +861,10 @@
     descriptor:StandInDescriptor;
 }
 
+table RankLayer {
+    base:LayerBase;
+}
+
 union Layer {
     ActivationLayer,
     AdditionLayer,
@@ -917,7 +923,8 @@
     ElementwiseUnaryLayer,
     TransposeLayer,
     QLstmLayer,
-    FillLayer
+    FillLayer,
+    RankLayer
 }
 
 table AnyLayer {