From a25d672b327ad02b3813592803e41f5909aa0d5a Mon Sep 17 00:00:00 2001
From: James Ward <james.ward@arm.com>
Date: Fri, 18 Nov 2022 15:10:49 +0000
Subject: Create MI tests for Tensors: ARGMAX, MAX_POOL2D

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I3682311cbdffa52fa88ae1b5779c615e2f062807
---
 .../argmax/argmax_14x27x3x22_bf16_axis3/test.json  | 55 ++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json

(limited to 'operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json')

diff --git a/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json
new file mode 100644
index 000000000..6824c2239
--- /dev/null
+++ b/operators/tensor/argmax/argmax_14x27x3x22_bf16_axis3/test.json
@@ -0,0 +1,55 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 3
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            14,
+            27,
+            3,
+            22
+          ],
+          type: "BF16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            14,
+            27,
+            3
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}
-- 
cgit v1.2.1