Create MI tests for Tensors: ARGMAX, MAX_POOL2D

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

3 files changed, 141 insertions, 0 deletions

diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/desc.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/desc.json
new file mode 100644
index 000000000..38f1a00dc
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/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": "At least one pad is larger than kernel dimension",
+  "profile": [
+    "tosa-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/input-0.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/input-0.json
new file mode 100644
index 000000000..2e7eac096
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/input-0.json
@@ -0,0 +1,48 @@
+{
+  "type": "float16",
+  "data": [
+    [
+      [
+        [
+          0.05816650390625,
+          0.189453125,
+          0.395751953125,
+          0.341064453125,
+          0.525390625,
+          0.2666015625,
+          0.79736328125,
+          0.0074920654296875,
+          0.79150390625,
+          0.393310546875,
+          0.00513458251953125
+        ],
+        [
+          0.84716796875,
+          0.44677734375,
+          0.724609375,
+          0.87841796875,
+          0.0343017578125,
+          0.26220703125,
+          0.118896484375,
+          0.0980224609375,
+          0.94580078125,
+          0.8056640625,
+          0.36474609375
+        ],
+        [
+          0.6875,
+          0.91943359375,
+          0.2208251953125,
+          0.01415252685546875,
+          0.82421875,
+          0.72265625,
+          0.1602783203125,
+          0.48974609375,
+          0.160400390625,
+          0.6025390625,
+          0.6474609375
+        ]
+      ]
+    ]
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/test.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/test.json
new file mode 100644
index 000000000..95dac3004
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_PadLargerEqualKernel_1x1x3x11_f16_st11_kern22_pad2232/test.json
@@ -0,0 +1,72 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "MAX_POOL2D",
+          attribute_type: "PoolAttribute",
+          attribute: {
+            pad: [
+              2,
+              2,
+              3,
+              2
+            ],
+            kernel: [
+              2,
+              2
+            ],
+            stride: [
+              1,
+              1
+            ],
+            input_zp: 0,
+            output_zp: 0,
+            accum_dtype: "UNKNOWN"
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            1,
+            1,
+            3,
+            11
+          ],
+          type: "FP16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            1,
+            4,
+            7,
+            11
+          ],
+          type: "FP16"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}