3 files changed, 123 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/desc.json
new file mode 100644
index 000000000..199bfa93c
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/desc.json
@@ -0,0 +1,22 @@
+{
+  "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": 2,
+  "expected_failure": true,
+  "expected_failure_desc": "Output data type not supported for this configuration of operator",
+  "profile": [
+    "tosa-bi",
+    "tosa-mi"
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/input-0.json
new file mode 100644
index 000000000..dd322f6a7
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/input-0.json
@@ -0,0 +1,50 @@
+{
+  "type": "int32",
+  "data": [
+    [
+      -7813,
+      26310,
+      29383,
+      -15573,
+      -21635,
+      -15284,
+      -20132
+    ],
+    [
+      -27189,
+      2361,
+      -30424,
+      18877,
+      3519,
+      -17013,
+      -12042
+    ],
+    [
+      -26971,
+      -22168,
+      12567,
+      31378,
+      -3287,
+      -5299,
+      10724
+    ],
+    [
+      -23378,
+      9828,
+      -12208,
+      -15344,
+      9040,
+      -2361,
+      12987
+    ],
+    [
+      -22448,
+      18032,
+      8493,
+      25992,
+      20310,
+      -13267,
+      4199
+    ]
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json
new file mode 100644
index 000000000..49f5c6b4c
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_5x7_i16_axis0/test.json
@@ -0,0 +1,51 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 0
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            5,
+            7
+          ],
+          type: "INT16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            7
+          ],
+          type: "INT16"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}