8K level: rank 6 ARGMAX tests

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ia7c3d89b176a2ac13fd05c734a3d8b7753fa8409

1 files changed, 64 insertions, 0 deletions

diff --git a/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json b/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json
new file mode 100644
index 000000000..5689f9f9b
--- /dev/null
+++ b/operators/tensor/argmax/argmax_5x9x7x7x7x8_i8_axis5/test.json
@@ -0,0 +1,64 @@
+{
+  version: {
+    _major: 0,
+    _minor: 70,
+    _patch: 0,
+    _draft: true
+  },
+  regions: [
+    {
+      name: "main",
+      blocks: [
+        {
+          name: "main",
+          operators: [
+            {
+              op: "ARGMAX",
+              attribute_type: "AxisAttribute",
+              attribute: {
+                axis: 5
+              },
+              inputs: [
+                "input-0"
+              ],
+              outputs: [
+                "result-0"
+              ]
+            }
+          ],
+          tensors: [
+            {
+              name: "input-0",
+              shape: [
+                5,
+                9,
+                7,
+                7,
+                7,
+                8
+              ],
+              type: "INT8"
+            },
+            {
+              name: "result-0",
+              shape: [
+                5,
+                9,
+                7,
+                7,
+                7
+              ],
+              type: "INT32"
+            }
+          ],
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ]
+    }
+  ]
+}