operators/tensor/argmax/argmax_59x34_bf16_axis0/test.json

{
  version: {
    _major: 0,
    _minor: 60,
    _patch: 0,
    _draft: false
  },
  regions: [
    {
      name: "main",
      blocks: [
        {
          name: "main",
          operators: [
            {
              op: "ARGMAX",
              attribute_type: "AxisAttribute",
              attribute: {
                axis: 0
              },
              inputs: [
                "input-0"
              ],
              outputs: [
                "result-0"
              ]
            }
          ],
          tensors: [
            {
              name: "input-0",
              shape: [
                59,
                34
              ],
              type: "BF16"
            },
            {
              name: "result-0",
              shape: [
                34
              ],
              type: "INT32"
            }
          ],
          inputs: [
            "input-0"
          ],
          outputs: [
            "result-0"
          ]
        }
      ]
    }
  ]
}