4 files changed, 137 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_56_i16_axis0/Conformance-result-0.json b/operators/tensor/argmax/argmax_56_i16_axis0/Conformance-result-0.json
new file mode 100644
index 000000000..07bc834e8
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_i16_axis0/Conformance-result-0.json
@@ -0,0 +1,6 @@
+{
+  "type": "int32",
+  "data": [
+    17
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_i16_axis0/desc.json b/operators/tensor/argmax/argmax_56_i16_axis0/desc.json
new file mode 100644
index 000000000..3cd43b509
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_i16_axis0/desc.json
@@ -0,0 +1,20 @@
+{
+  "tosa_file": "test.json",
+  "ifm_name": [
+    "input-0"
+  ],
+  "ifm_file": [
+    "input-0.npy"
+  ],
+  "ofm_name": [
+    "result-0"
+  ],
+  "ofm_file": [
+    "result-0.npy"
+  ],
+  "expected_return_code": 0,
+  "expected_failure": false,
+  "expected_result_file": [
+    "Conformance-result-0.npy"
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_i16_axis0/input-0.json b/operators/tensor/argmax/argmax_56_i16_axis0/input-0.json
new file mode 100644
index 000000000..3a9c563cd
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_i16_axis0/input-0.json
@@ -0,0 +1,61 @@
+{
+  "type": "int32",
+  "data": [
+    -3560,
+    25763,
+    11660,
+    18244,
+    17032,
+    -20013,
+    -8920,
+    -2181,
+    -145,
+    -29898,
+    3051,
+    -22657,
+    15949,
+    11996,
+    27690,
+    16040,
+    -8740,
+    30638,
+    -5843,
+    -11415,
+    26577,
+    -8490,
+    -27765,
+    -1996,
+    19377,
+    -20351,
+    -2429,
+    -24254,
+    12221,
+    -1593,
+    -11129,
+    -17898,
+    4227,
+    11128,
+    28855,
+    -4119,
+    -22238,
+    21802,
+    8504,
+    13124,
+    -26397,
+    -12297,
+    17553,
+    21774,
+    -4246,
+    19973,
+    22393,
+    -7375,
+    26089,
+    -13873,
+    -17069,
+    11960,
+    8962,
+    -23610,
+    21806,
+    -19667
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_i16_axis0/test.json b/operators/tensor/argmax/argmax_56_i16_axis0/test.json
new file mode 100644
index 000000000..0d7914b7a
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_i16_axis0/test.json
@@ -0,0 +1,50 @@
+{
+  version: {
+    _major: 0,
+    _minor: 30,
+    _patch: 0,
+    _draft: false
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 0
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            56
+          ],
+          type: "INT16"
+        },
+        {
+          name: "result-0",
+          shape: [
+
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}