diff options
Diffstat (limited to 'operators/tensor/argmax/argmax_1x7x6x4x4_i8_axis1/test.json')
-rw-r--r-- | operators/tensor/argmax/argmax_1x7x6x4x4_i8_axis1/test.json | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_1x7x6x4x4_i8_axis1/test.json b/operators/tensor/argmax/argmax_1x7x6x4x4_i8_axis1/test.json new file mode 100644 index 000000000..ba8984bae --- /dev/null +++ b/operators/tensor/argmax/argmax_1x7x6x4x4_i8_axis1/test.json @@ -0,0 +1,62 @@ +{ + version: { + _major: 0, + _minor: 70, + _patch: 0, + _draft: true + }, + regions: [ + { + name: "main", + blocks: [ + { + name: "main", + operators: [ + { + op: "ARGMAX", + attribute_type: "AxisAttribute", + attribute: { + axis: 1 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 1, + 7, + 6, + 4, + 4 + ], + type: "INT8" + }, + { + name: "result-0", + shape: [ + 1, + 6, + 4, + 4 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] + } + ] +} |