3 files changed, 133 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/desc.json
new file mode 100644
index 000000000..992afc22b
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_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": "Input data type not supported for this operator",
+  "profile": [
+    "tosa-bi",
+    "tosa-mi"
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/input-0.json
new file mode 100644
index 000000000..fa6489725
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/input-0.json
@@ -0,0 +1,55 @@
+{
+  "type": "int64",
+  "data": [
+    [
+      74004711791105,
+      115948743448071,
+      110132723124812
+    ],
+    [
+      -74275077411099,
+      -21822440320907,
+      61447363562558
+    ],
+    [
+      -103494339845735,
+      129299819790654,
+      69511024151637
+    ],
+    [
+      92560761594434,
+      114685398905668,
+      -90648490837828
+    ],
+    [
+      -21918178679045,
+      42691248193943,
+      90764737270298
+    ],
+    [
+      66992285361977,
+      -106529524820847,
+      41003010594236
+    ],
+    [
+      -57968732935161,
+      -26318526907244,
+      139113340624813
+    ],
+    [
+      -96530207496223,
+      -111356979913411,
+      -25573654732685
+    ],
+    [
+      102492934810913,
+      -95079918739684,
+      -118232768813108
+    ],
+    [
+      -22841994337105,
+      37238597353371,
+      -97195417767857
+    ]
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/test.json
new file mode 100644
index 000000000..e3e5bbe91
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10x3_i48_axis0/test.json
@@ -0,0 +1,56 @@
+{
+  version: {
+    _major: 0,
+    _minor: 70,
+    _patch: 0,
+    _draft: true
+  },
+  regions: [
+    {
+      name: "main",
+      blocks: [
+        {
+          name: "main",
+          operators: [
+            {
+              op: "ARGMAX",
+              attribute_type: "AxisAttribute",
+              attribute: {
+                axis: 0
+              },
+              inputs: [
+                "input-0"
+              ],
+              outputs: [
+                "result-0"
+              ]
+            }
+          ],
+          tensors: [
+            {
+              name: "input-0",
+              shape: [
+                10,
+                3
+              ],
+              type: "INT48"
+            },
+            {
+              name: "result-0",
+              shape: [
+                3
+              ],
+              type: "INT32"
+            }
+          ],
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ]
+    }
+  ]
+}