4 files changed, 136 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_52_f32_axis0/Conformance-result-0.json b/operators/tensor/argmax/argmax_52_f32_axis0/Conformance-result-0.json
new file mode 100644
index 000000000..a76a8b1e4
--- /dev/null
+++ b/operators/tensor/argmax/argmax_52_f32_axis0/Conformance-result-0.json
@@ -0,0 +1,6 @@
+{
+  "type": "int32",
+  "data": [
+    4
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_52_f32_axis0/desc.json b/operators/tensor/argmax/argmax_52_f32_axis0/desc.json
new file mode 100644
index 000000000..d418e4060
--- /dev/null
+++ b/operators/tensor/argmax/argmax_52_f32_axis0/desc.json
@@ -0,0 +1,23 @@
+{
+  "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": 0,
+  "expected_failure": false,
+  "expected_result_file": [
+    "Conformance-result-0.npy"
+  ],
+  "profile": [
+    "tosa-mi"
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_52_f32_axis0/input-0.json b/operators/tensor/argmax/argmax_52_f32_axis0/input-0.json
new file mode 100644
index 000000000..655da1649
--- /dev/null
+++ b/operators/tensor/argmax/argmax_52_f32_axis0/input-0.json
@@ -0,0 +1,57 @@
+{
+  "type": "float32",
+  "data": [
+    1.847590684890747,
+    1.6343227624893188,
+    0.7988285422325134,
+    -0.9365201592445374,
+    1.876705527305603,
+    1.1150035858154297,
+    0.8675607442855835,
+    -0.20255398750305176,
+    -0.9110337495803833,
+    -1.614436149597168,
+    1.6104096174240112,
+    -0.17689484357833862,
+    -1.1905465126037598,
+    -0.7761735320091248,
+    0.3168782889842987,
+    -1.2929089069366455,
+    1.426457166671753,
+    1.0340781211853027,
+    0.8778518438339233,
+    -0.27162784337997437,
+    0.5092353820800781,
+    0.3363918662071228,
+    0.5993863940238953,
+    -1.6622227430343628,
+    -0.3367703855037689,
+    -1.833543300628662,
+    -0.024036722257733345,
+    -0.6805551648139954,
+    -1.4219032526016235,
+    -1.586388111114502,
+    0.35057827830314636,
+    -1.3176281452178955,
+    1.7004804611206055,
+    0.32424455881118774,
+    -0.6125207543373108,
+    0.3636619746685028,
+    -1.9087845087051392,
+    1.8342368602752686,
+    -0.070786252617836,
+    1.1309409141540527,
+    -1.6690800189971924,
+    -0.05336667597293854,
+    -0.037172023206949234,
+    1.7513058185577393,
+    0.286912202835083,
+    -0.10604239255189896,
+    -0.9320973753929138,
+    -0.6737239956855774,
+    0.08268961310386658,
+    -0.24435415863990784,
+    -1.913551688194275,
+    1.3051676750183105
+  ]
+}
+\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_52_f32_axis0/test.json b/operators/tensor/argmax/argmax_52_f32_axis0/test.json
new file mode 100644
index 000000000..44cc6ffaa
--- /dev/null
+++ b/operators/tensor/argmax/argmax_52_f32_axis0/test.json
@@ -0,0 +1,50 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 0
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            52
+          ],
+          type: "FP32"
+        },
+        {
+          name: "result-0",
+          shape: [
+
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}