Add generated tests for 2D input and 3D indices in gather (#4344)

* Add generated tests for 2D input and 3D indices in gather

This commit contains a new generated test for 2D input and 3D indices in gather.

gather_ex_2D_3D_float_1 for [1,1,2] shape indices with axis 0
gather_ex_2D_3D_float_2 for [1,2,1] shape indices with axis 1

* Update wrong comments

Signed-off-by: Jiyoung Yun <jy910.yun@samsung.com>

diff --git a/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_1.mod.py b/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_1.mod.py
new file mode 100644 (file)
index 0000000..652016a
--- /dev/null
+++ b/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_1.mod.py
@@ -0,0 +1,22 @@
+# model
+model = Model()
+i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 12 float32s
+i2 = Input("op2", "TENSOR_INT32", "{1,1,2}") # another vector of 2 int32s
+axis = Int32Scalar("axis", 0)
+i3 = Output("op3", "TENSOR_FLOAT32", "{1,1,2,4}") # a vector of 8 float32s
+model = model.Operation("GATHER_EX", i1, i2, axis).To(i3)
+
+# Example 1. Input in operand 0,
+input0 = {i1: # input 0
+          [3.123456789123456789, 4.123456789123456789, 5.123456789123456789, 6.123456789123456789,
+          7.123456789123456789, 8.123456789123456789, 9.123456789123456789, 1.123456789123456789,
+          2.123456789123456789, 18.123456789123456789, 19.123456789123456789, 11.123456789123456789],
+          i2: # input 1
+          [1, 0]}
+
+output0 = {i3: # output 0
+           [7.123456789123456789, 8.123456789123456789, 9.123456789123456789, 1.123456789123456789,
+            3.123456789123456789, 4.123456789123456789, 5.123456789123456789, 6.123456789123456789]}
+
+# Instantiate an example
+Example((input0, output0))

diff --git a/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_2.mod.py b/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_2.mod.py
new file mode 100644 (file)
index 0000000..850d330
--- /dev/null
+++ b/tests/nnapi/specs/Ex/gather_ex_2D_3D_float_2.mod.py
@@ -0,0 +1,23 @@
+# model
+model = Model()
+i1 = Input("op1", "TENSOR_FLOAT32", "{3,4}") # a vector of 12 float32s
+i2 = Input("op2", "TENSOR_INT32", "{1,2,1}") # another vector of 2 int32s
+axis = Int32Scalar("axis", 1)
+i3 = Output("op3", "TENSOR_FLOAT32", "{3,1,2,1}") # a vector of 6 float32s
+model = model.Operation("GATHER_EX", i1, i2, axis).To(i3)
+
+# Example 1. Input in operand 0,
+input0 = {i1: # input 0
+          [3.123456789123456789, 4.123456789123456789, 5.123456789123456789, 6.123456789123456789,
+          7.123456789123456789, 8.123456789123456789, 9.123456789123456789, 1.123456789123456789,
+          2.123456789123456789, 18.123456789123456789, 19.123456789123456789, 11.123456789123456789],
+          i2: # input 1
+          [1, 0]}
+
+output0 = {i3: # output 0
+           [4.123456789123456789, 3.123456789123456789,
+            8.123456789123456789, 7.123456789123456789,
+            18.123456789123456789, 2.123456789123456789]}
+
+# Instantiate an example
+Example((input0, output0))