{ | |
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: [ | |
54, | |
38 | |
], | |
type: "FP32" | |
}, | |
{ | |
name: "result-0", | |
shape: [ | |
38 | |
], | |
type: "INT32" | |
} | |
], | |
inputs: [ | |
"input-0" | |
], | |
outputs: [ | |
"result-0" | |
] | |
} | |
] | |
} | |
] | |
} |