operators/tensor/argmax/argmax_14x27x3x22_bf16_axis0/test.json

{
  version: {
    _major: 0,
    _minor: 50,
    _patch: 0,
    _draft: false
  },
  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"
      ]
    }
  ]
}