commit 0cfcf235c4bcd2ae570eea8bc2677f471281b8e6
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Mon Sep 09 17:16:24 2019 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Sep 11 08:57:05 2019 +0000
tree dbeba0ddeb240f91099f6d436757cd365dc8ab2b
parent 4cd29a046c3d46917d84d12feb668969af23a39e

IVGCVSW-3724 Adding serialization support for ArgMinMax

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I21210c843c3b8800ccc68d4f3095259d0a233bd1

src/armnnSerializer/ArmnnSchema.fbs

diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 2b3ec49..97d2780 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -22,6 +22,11 @@
     Square = 9
 }
 
+enum ArgMinMaxFunction : byte {
+    Min = 0,
+    Max = 1
+}
+
 enum DataType : byte {
     Float16 = 0,
     Float32 = 1,
@@ -133,7 +138,8 @@
     Resize = 43,
     Stack = 44,
     QuantizedLstm = 45,
-    Abs = 46
+    Abs = 46,
+    ArgMinMax = 47
 }
 
 // Base layer table to be used as part of other layers
@@ -171,6 +177,16 @@
     base:LayerBase;
 }
 
+table ArgMinMaxLayer {
+    base:LayerBase;
+    descriptor:ArgMinMaxDescriptor;
+}
+
+table ArgMinMaxDescriptor{
+    function:ArgMinMaxFunction;
+    axis:int;
+}
+
 table ConstantLayer {
     base:LayerBase;
     input:ConstTensor;
@@ -693,7 +709,8 @@
     TransposeConvolution2dLayer,
     ResizeLayer,
     StackLayer,
-    AbsLayer
+    AbsLayer,
+    ArgMinMaxLayer
 }
 
 table AnyLayer {