From 4129b45c7464197b7a3415c6865282352a7fa40e Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Thu, 13 Oct 2022 13:54:23 +0100
Subject: Updating ARGMAX tests for FP16 and accumulator types

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I96337ce765a04870b548f7905ab869c2e0287808
---
 .../desc.json                                      |  22 ++
 .../input-0.json                                   | 291 +++++++++++++++++++++
 .../test.json                                      |  54 ++++
 3 files changed, 367 insertions(+)
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/desc.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/input-0.json
 create mode 100644 operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json

(limited to 'operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12')

diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/desc.json b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/desc.json
new file mode 100644
index 000000000..db4d1a620
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/desc.json
@@ -0,0 +1,22 @@
+{
+  "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": "Axis larger than rank",
+  "profile": [
+    "tosa-bi",
+    "tosa-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/input-0.json
new file mode 100644
index 000000000..9b48e975e
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/input-0.json
@@ -0,0 +1,291 @@
+{
+  "type": "int32",
+  "data": [
+    [
+      [
+        974,
+        21281
+      ],
+      [
+        32106,
+        20295
+      ],
+      [
+        -10509,
+        23207
+      ],
+      [
+        -31934,
+        -11397
+      ],
+      [
+        -19700,
+        1838
+      ],
+      [
+        25664,
+        17914
+      ]
+    ],
+    [
+      [
+        32657,
+        -23791
+      ],
+      [
+        16527,
+        14705
+      ],
+      [
+        -2853,
+        -26380
+      ],
+      [
+        -25920,
+        27082
+      ],
+      [
+        -13421,
+        8998
+      ],
+      [
+        29199,
+        -439
+      ]
+    ],
+    [
+      [
+        -19011,
+        -1001
+      ],
+      [
+        -17729,
+        -20883
+      ],
+      [
+        20380,
+        -10098
+      ],
+      [
+        12111,
+        14884
+      ],
+      [
+        3238,
+        2683
+      ],
+      [
+        -20937,
+        1920
+      ]
+    ],
+    [
+      [
+        16484,
+        -19973
+      ],
+      [
+        -17563,
+        12475
+      ],
+      [
+        31575,
+        -7076
+      ],
+      [
+        21517,
+        -20792
+      ],
+      [
+        4569,
+        29083
+      ],
+      [
+        -3831,
+        445
+      ]
+    ],
+    [
+      [
+        -22605,
+        1837
+      ],
+      [
+        5612,
+        31547
+      ],
+      [
+        -19598,
+        -8756
+      ],
+      [
+        9650,
+        29033
+      ],
+      [
+        -17179,
+        12774
+      ],
+      [
+        24860,
+        8722
+      ]
+    ],
+    [
+      [
+        -4797,
+        -17406
+      ],
+      [
+        -13891,
+        -29377
+      ],
+      [
+        -22805,
+        -4078
+      ],
+      [
+        -23561,
+        -32102
+      ],
+      [
+        12627,
+        -28115
+      ],
+      [
+        -23520,
+        6247
+      ]
+    ],
+    [
+      [
+        27644,
+        14710
+      ],
+      [
+        -30637,
+        -21762
+      ],
+      [
+        15752,
+        15237
+      ],
+      [
+        -7142,
+        29473
+      ],
+      [
+        25458,
+        15936
+      ],
+      [
+        16267,
+        18659
+      ]
+    ],
+    [
+      [
+        -23041,
+        13153
+      ],
+      [
+        28829,
+        -18040
+      ],
+      [
+        -31809,
+        -9117
+      ],
+      [
+        -15099,
+        29443
+      ],
+      [
+        -3547,
+        10565
+      ],
+      [
+        31584,
+        18941
+      ]
+    ],
+    [
+      [
+        16323,
+        30716
+      ],
+      [
+        12376,
+        -17601
+      ],
+      [
+        14862,
+        -11686
+      ],
+      [
+        -11081,
+        9489
+      ],
+      [
+        -12093,
+        -31172
+      ],
+      [
+        32739,
+        4667
+      ]
+    ],
+    [
+      [
+        11141,
+        20230
+      ],
+      [
+        8815,
+        7238
+      ],
+      [
+        19126,
+        -13410
+      ],
+      [
+        21524,
+        14686
+      ],
+      [
+        -20834,
+        -23198
+      ],
+      [
+        24883,
+        25871
+      ]
+    ],
+    [
+      [
+        32347,
+        -26155
+      ],
+      [
+        25927,
+        23821
+      ],
+      [
+        17537,
+        -144
+      ],
+      [
+        557,
+        -4677
+      ],
+      [
+        -30851,
+        43
+      ],
+      [
+        -17226,
+        -19648
+      ]
+    ]
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json
new file mode 100644
index 000000000..705ea85a6
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_AxisLargerRank_11x6x2_i16_axis12/test.json
@@ -0,0 +1,54 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 12
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            11,
+            6,
+            2
+          ],
+          type: "INT16"
+        },
+        {
+          name: "result-0",
+          shape: [
+            11,
+            6,
+            2
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}
-- 
cgit v1.2.1