Create MI tests for Tensors: ARGMAX, MAX_POOL2D

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

4 files changed, 140 insertions, 0 deletions

diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json b/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json
new file mode 100644
index 000000000..d576d4dc2
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json
@@ -0,0 +1,6 @@
+{
+  "type": "int32",
+  "data": [
+    29
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json b/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json
new file mode 100644
index 000000000..d418e4060
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json
@@ -0,0 +1,23 @@
+{
+  "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": 0,
+  "expected_failure": false,
+  "expected_result_file": [
+    "Conformance-result-0.npy"
+  ],
+  "profile": [
+    "tosa-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json b/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json
new file mode 100644
index 000000000..f44785938
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json
@@ -0,0 +1,61 @@
+{
+  "type": "float32",
+  "data": [
+    -0.75,
+    1.328125,
+    1.21875,
+    -0.44921875,
+    -0.84375,
+    0.7265625,
+    -1.4375,
+    -1.1953125,
+    -1.96875,
+    1.140625,
+    0.65625,
+    0.8203125,
+    1.1171875,
+    -0.1640625,
+    0.2734375,
+    -1.4375,
+    -1.5390625,
+    0.671875,
+    -0.115234375,
+    0.259765625,
+    1.0546875,
+    0.53515625,
+    0.2138671875,
+    0.236328125,
+    -0.78125,
+    -1.875,
+    -0.251953125,
+    -1.140625,
+    -0.365234375,
+    1.40625,
+    -1.0625,
+    -1.765625,
+    -0.87109375,
+    -0.82421875,
+    0.64453125,
+    0.2275390625,
+    1.1328125,
+    0.65625,
+    -0.373046875,
+    1.25,
+    -1.328125,
+    -1.90625,
+    -1.6328125,
+    0.88671875,
+    -0.15234375,
+    -1.3515625,
+    0.004150390625,
+    -1.390625,
+    0.78515625,
+    -0.21484375,
+    -0.474609375,
+    -0.79296875,
+    0.51953125,
+    -0.55078125,
+    -1.6484375,
+    -1.5234375
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/test.json b/operators/tensor/argmax/argmax_56_bf16_axis0/test.json
new file mode 100644
index 000000000..7a15987fc
--- /dev/null
+++ b/operators/tensor/argmax/argmax_56_bf16_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: [
+            56
+          ],
+          type: "BF16"
+        },
+        {
+          name: "result-0",
+          shape: [
+
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}