Create MI tests for Tensors: ARGMAX, MAX_POOL2D

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

3 files changed, 177 insertions, 0 deletions

diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/desc.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/desc.json
new file mode 100644
index 000000000..59c8fbad4
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/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 stride dimension is smaller than zero",
+  "profile": [
+    "tosa-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/input-0.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/input-0.json
new file mode 100644
index 000000000..b0e38aa5c
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/input-0.json
@@ -0,0 +1,84 @@
+{
+  "type": "float16",
+  "data": [
+    [
+      [
+        [
+          0.0028705596923828125,
+          0.427978515625,
+          0.01253509521484375,
+          0.90966796875,
+          0.6416015625,
+          0.5830078125,
+          0.2015380859375,
+          0.69970703125,
+          0.67724609375,
+          0.060455322265625,
+          0.880859375,
+          0.57958984375,
+          0.79248046875
+        ],
+        [
+          0.262939453125,
+          0.1220703125,
+          0.70263671875,
+          0.7939453125,
+          0.385986328125,
+          0.0236968994140625,
+          0.818359375,
+          0.2060546875,
+          0.339111328125,
+          0.62841796875,
+          0.0999755859375,
+          0.072021484375,
+          0.0257110595703125
+        ],
+        [
+          0.91650390625,
+          0.311279296875,
+          0.66064453125,
+          0.9892578125,
+          0.66943359375,
+          0.52490234375,
+          0.52099609375,
+          0.376953125,
+          0.599609375,
+          0.958984375,
+          0.7880859375,
+          0.42431640625,
+          0.33251953125
+        ],
+        [
+          0.88330078125,
+          0.1357421875,
+          0.849609375,
+          0.255126953125,
+          0.7158203125,
+          0.80615234375,
+          0.47705078125,
+          0.1922607421875,
+          0.8447265625,
+          0.367431640625,
+          0.32080078125,
+          0.51025390625,
+          0.43505859375
+        ],
+        [
+          0.68310546875,
+          0.13720703125,
+          0.435791015625,
+          0.8916015625,
+          0.03607177734375,
+          0.457275390625,
+          0.66064453125,
+          0.8876953125,
+          0.59375,
+          0.82470703125,
+          0.22265625,
+          0.352783203125,
+          0.94189453125
+        ]
+      ]
+    ]
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/test.json b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/test.json
new file mode 100644
index 000000000..b3255f6e3
--- /dev/null
+++ b/operators/tensor/max_pool2d/max_pool2d_ERRORIF_StrideSmallerOne_1x1x5x13_f16_st-2-2_kern22_pad0000/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: [
+              0,
+              0,
+              0,
+              0
+            ],
+            kernel: [
+              2,
+              2
+            ],
+            stride: [
+              -2,
+              -2
+            ],
+            input_zp: 0,
+            output_zp: 0,
+            accum_dtype: "UNKNOWN"
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            1,
+            1,
+            5,
+            13
+          ],
+          type: "FP16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            1,
+            1,
+            1,
+            13
+          ],
+          type: "FP16"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}