1 files changed, 22 insertions, 0 deletions
diff --git a/tests/nnapi/specs/Ex/gather_ex_2D_quant8.mod.py b/tests/nnapi/specs/Ex/gather_ex_2D_quant8.mod.py
new file mode 100644
index 000000000..8e4eb04ec
--- /dev/null
+++ b/tests/nnapi/specs/Ex/gather_ex_2D_quant8.mod.py
@@ -0,0 +1,22 @@
+# model
+model = Model()
+i1 = Input("op1", "TENSOR_QUANT8_ASYMM", "{3,4}, 0.5f, 1") # a vector of 12 qasymm8s
+i2 = Input("op2", "TENSOR_INT32", "{2}") # another vector of 2 int32s
+axis = Int32Scalar("axis", 0)
+i3 = Output("op3", "TENSOR_QUANT8_ASYMM", "{2,4}, 0.5f, 1") # a vector of 8 qasymm8s
+model = model.Operation("GATHER_EX", i1, i2, axis).To(i3)
+
+# Example 1. Input in operand 0,
+input0 = {i1: # input 0
+          [7, 9, 11, 13,
+          15, 17, 19, 3,
+          5, 37, 39, 23],
+          i2: # input 1
+          [2, 0]}
+
+output0 = {i3: # output 0
+           [5, 37, 39, 23,
+           7, 9, 11, 13]}
+
+# Instantiate an example
+Example((input0, output0))