From 4129b45c7464197b7a3415c6865282352a7fa40e Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Thu, 13 Oct 2022 13:54:23 +0100
Subject: Updating ARGMAX tests for FP16 and accumulator types

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I96337ce765a04870b548f7905ab869c2e0287808
---
 .../desc.json                                      | 22 ++++++++++
 .../input-0.json                                   | 15 +++++++
 .../test.json                                      | 50 ++++++++++++++++++++++
 3 files changed, 87 insertions(+)
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/desc.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/input-0.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json

(limited to 'operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0')

diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/desc.json
new file mode 100644
index 000000000..992afc22b
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_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_10_b_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/input-0.json
new file mode 100644
index 000000000..2a6289573
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/input-0.json
@@ -0,0 +1,15 @@
+{
+  "type": "bool",
+  "data": [
+    true,
+    true,
+    false,
+    true,
+    true,
+    true,
+    false,
+    true,
+    true,
+    false
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json
new file mode 100644
index 000000000..d0a53bfc9
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongInputType_10_b_axis0/test.json
@@ -0,0 +1,50 @@
+{
+  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: [
+            10
+          ],
+          type: "BOOL"
+        },
+        {
+          name: "result-0",
+          shape: [
+
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}
-- 
cgit v1.2.1