From 52ac9d6e4afa916acc4cab0617b940f4ab7aaf4f Mon Sep 17 00:00:00 2001
From: James Ward <james.ward@arm.com>
Date: Wed, 16 Nov 2022 12:53:49 +0000
Subject: Create MI tests for Tensors: ARGMAX, MAX_POOL2D

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I908d88ddf7c52d98ce6d533396b5c439f15750f5
---
 verif/conformance/tosa_main_profile_ops_info.json | 116 ++++++++++++++++++++++
 1 file changed, 116 insertions(+)

diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index 656cdde..0614dfc 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -1,4 +1,66 @@
 {
+    "argmax": {
+        "group": "tensor",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "32,64",
+                "--target-rank",
+                "1",
+                "--target-rank",
+                "2",
+                "--target-rank",
+                "3",
+                "--num-rand-permutations",
+                "2"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,32",
+                "--target-rank",
+                "4",
+                "--num-rand-permutations",
+                "2"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,3,65535,1",
+                "--target-shape",
+                "1,65537,1,2",
+                "--num-rand-permutations",
+                "2"
+            ]
+        ],
+        "params": {
+        },
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "avg_pool2d": {
         "group": "tensor",
         "generator_args": [
@@ -92,5 +154,59 @@
         "profile": [
             "tosa-mi"
         ]
+    },
+    "max_pool2d": {
+        "group": "tensor",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,47,37,25",
+                "--target-shape",
+                "1,15,13,38",
+                "--allow-pooling-and-conv-oversizes"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,65534,4,1",
+                "--target-shape",
+                "1,6,65536,1",
+                "--max-conv-dilation",
+                "1",
+                "--allow-pooling-and-conv-oversizes"
+            ]
+        ],
+        "params": {
+            "shape": [],
+            "type": [
+                "f32",
+                "f16",
+                "bf16"
+            ],
+            "stride": [],
+            "pad": []
+        },
+        "permutes": [
+            "shape",
+            "type",
+            "pad"
+        ],
+        "sparsity": {
+            "pad": 11
+        },
+        "profile": [
+            "tosa-mi"
+        ]
     }
 }
\ No newline at end of file
-- 
cgit v1.2.1