{ 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: [ 1, 6, 7, 15, 4 ], type: "BF16" }, { name: "result-0", shape: [ 6, 7, 15, 4 ], type: "INT32" } ], inputs: [ "input-0" ], outputs: [ "result-0" ] } ] }