From 7037332896b3151963a3e83f65acb40590c74199 Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Tue, 9 May 2023 13:02:04 +0100
Subject: 8K level: rank 6 ARGMAX tests

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ia7c3d89b176a2ac13fd05c734a3d8b7753fa8409
---
 .../desc.json                                      | 21 +++++++++
 .../input-0.json                                   | 16 +++++++
 .../test.json                                      | 55 ++++++++++++++++++++++
 3 files changed, 92 insertions(+)
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/desc.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/input-0.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/test.json

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

diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/desc.json
new file mode 100644
index 000000000..d57e8394e
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/desc.json
@@ -0,0 +1,21 @@
+{
+  "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-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/input-0.json
new file mode 100644
index 000000000..c6bbb4f2f
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/input-0.json
@@ -0,0 +1,16 @@
+{
+  "type": "float32",
+  "data": [
+    0.86328125,
+    0.6484375,
+    0.82421875,
+    0.60546875,
+    0.57421875,
+    0.77734375,
+    0.87109375,
+    0.73828125,
+    0.1533203125,
+    0.1298828125,
+    0.9609375
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/test.json
new file mode 100644
index 000000000..0178f4b69
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_11_bf16_axis0/test.json
@@ -0,0 +1,55 @@
+{
+  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: [
+                11
+              ],
+              type: "BF16"
+            },
+            {
+              name: "result-0",
+              shape: [
+
+              ],
+              type: "INT48"
+            }
+          ],
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ]
+    }
+  ]
+}
-- 
cgit v1.2.1