Create MI tests for Tensors: ARGMAX, MAX_POOL2D

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I3682311cbdffa52fa88ae1b5779c615e2f062807

3 files changed, 157 insertions, 0 deletions

diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/desc.json
new file mode 100644
index 000000000..d57e8394e
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_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_10x6_bf16_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/input-0.json
new file mode 100644
index 000000000..aac158dbe
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/input-0.json
@@ -0,0 +1,85 @@
+{
+  "type": "float32",
+  "data": [
+    [
+      0.06884765625,
+      0.875,
+      0.875,
+      0.84375,
+      0.2021484375,
+      0.1630859375
+    ],
+    [
+      0.96875,
+      0.10595703125,
+      0.78515625,
+      0.8046875,
+      0.9296875,
+      0.84375
+    ],
+    [
+      0.671875,
+      0.384765625,
+      0.25390625,
+      0.953125,
+      0.2333984375,
+      0.546875
+    ],
+    [
+      0.052001953125,
+      0.93359375,
+      0.19140625,
+      0.22265625,
+      0.953125,
+      0.396484375
+    ],
+    [
+      0.5625,
+      0.494140625,
+      0.50390625,
+      0.63671875,
+      0.95703125,
+      0.0277099609375
+    ],
+    [
+      0.044921875,
+      0.46875,
+      0.80078125,
+      0.4453125,
+      0.451171875,
+      0.43359375
+    ],
+    [
+      0.361328125,
+      0.031982421875,
+      0.1064453125,
+      0.85546875,
+      0.99609375,
+      0.0810546875
+    ],
+    [
+      0.55078125,
+      0.91796875,
+      0.7109375,
+      0.40234375,
+      0.6875,
+      0.11962890625
+    ],
+    [
+      0.31640625,
+      0.455078125,
+      0.173828125,
+      0.416015625,
+      0.99609375,
+      0.2216796875
+    ],
+    [
+      0.12890625,
+      0.1787109375,
+      0.53125,
+      0.2021484375,
+      0.04150390625,
+      0.01611328125
+    ]
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_axis0/test.json
new file mode 100644
index 000000000..6a3101840
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputType_10x6_bf16_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: [
+            10,
+            6
+          ],
+          type: "BF16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            6
+          ],
+          type: "INT48"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}