operators/tensor/argmax/argmax_14x27x3x22_bf16_axis0/test.json

{
  version: {
    _major: 0,
    _minor: 80,
    _patch: 0,
    _draft: true
  },
  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: [
                14,
                27,
                3,
                22
              ],
              type: "BF16"
            },
            {
              name: "result-0",
              shape: [
                27,
                3,
                22
              ],
              type: "INT32"
            }
          ],
          inputs: [
            "input-0"
          ],
          outputs: [
            "result-0"
          ]
        }
      ]
    }
  ]
}