diff options
Diffstat (limited to 'operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0')
4 files changed, 637 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/Conformance-result-0.json b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/Conformance-result-0.json new file mode 100644 index 000000000..c316b9cfd --- /dev/null +++ b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/Conformance-result-0.json @@ -0,0 +1,63 @@ +{ + "type": "int32", + "data": [ + [ + [ + [ + 0 + ], + [ + 2 + ] + ], + [ + [ + 4 + ], + [ + 3 + ] + ], + [ + [ + 1 + ], + [ + 1 + ] + ], + [ + [ + 4 + ], + [ + 1 + ] + ], + [ + [ + 7 + ], + [ + 5 + ] + ], + [ + [ + 5 + ], + [ + 2 + ] + ], + [ + [ + 1 + ], + [ + 0 + ] + ] + ] + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/desc.json b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/desc.json new file mode 100644 index 000000000..279cf7380 --- /dev/null +++ b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/desc.json @@ -0,0 +1,27 @@ +{ + "tosa_file": "test.json", + "ifm_name": [ + "input-0" + ], + "ifm_file": [ + "input-0.npy" + ], + "ofm_name": [ + "result-0" + ], + "ofm_file": [ + "result-0.npy" + ], + "expected_return_code": 0, + "expected_failure": false, + "expected_result_file": [ + "Conformance-result-0.npy" + ], + "profile": [ + "tosa-bi", + "tosa-mi" + ], + "tag": [ + "8k_level" + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/input-0.json b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/input-0.json new file mode 100644 index 000000000..9c3981944 --- /dev/null +++ b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/input-0.json @@ -0,0 +1,485 @@ +{ + "type": "int32", + "data": [ + [ + [ + [ + [ + 30212 + ], + [ + 27076 + ] + ], + [ + [ + 19541 + ], + [ + -16599 + ] + ], + [ + [ + 17976 + ], + [ + 15500 + ] + ], + [ + [ + -13796 + ], + [ + 11663 + ] + ], + [ + [ + -16190 + ], + [ + 1513 + ] + ], + [ + [ + 19447 + ], + [ + -20413 + ] + ], + [ + [ + -10686 + ], + [ + 19348 + ] + ] + ] + ], + [ + [ + [ + [ + -3029 + ], + [ + 30187 + ] + ], + [ + [ + -32247 + ], + [ + 14538 + ] + ], + [ + [ + 20591 + ], + [ + 27384 + ] + ], + [ + [ + -22495 + ], + [ + 31731 + ] + ], + [ + [ + -7060 + ], + [ + -9697 + ] + ], + [ + [ + -9468 + ], + [ + 9058 + ] + ], + [ + [ + 10439 + ], + [ + -1709 + ] + ] + ] + ], + [ + [ + [ + [ + -28113 + ], + [ + 31891 + ] + ], + [ + [ + -18213 + ], + [ + -26621 + ] + ], + [ + [ + 6382 + ], + [ + -5743 + ] + ], + [ + [ + 7573 + ], + [ + 24611 + ] + ], + [ + [ + -26390 + ], + [ + -14153 + ] + ], + [ + [ + 1086 + ], + [ + 18860 + ] + ], + [ + [ + 6595 + ], + [ + -7682 + ] + ] + ] + ], + [ + [ + [ + [ + 7212 + ], + [ + 31455 + ] + ], + [ + [ + -1308 + ], + [ + 25135 + ] + ], + [ + [ + 18347 + ], + [ + -13021 + ] + ], + [ + [ + -16417 + ], + [ + 17707 + ] + ], + [ + [ + -19171 + ], + [ + -10692 + ] + ], + [ + [ + -17342 + ], + [ + 6970 + ] + ], + [ + [ + -13492 + ], + [ + 10386 + ] + ] + ] + ], + [ + [ + [ + [ + 28497 + ], + [ + 21600 + ] + ], + [ + [ + 29357 + ], + [ + -31166 + ] + ], + [ + [ + -2920 + ], + [ + -18340 + ] + ], + [ + [ + 27385 + ], + [ + 13888 + ] + ], + [ + [ + -11782 + ], + [ + 4163 + ] + ], + [ + [ + 6909 + ], + [ + -289 + ] + ], + [ + [ + -1681 + ], + [ + -29150 + ] + ] + ] + ], + [ + [ + [ + [ + -3116 + ], + [ + -9944 + ] + ], + [ + [ + 20856 + ], + [ + 23927 + ] + ], + [ + [ + 16411 + ], + [ + -30744 + ] + ], + [ + [ + 22362 + ], + [ + -6788 + ] + ], + [ + [ + -5481 + ], + [ + 29173 + ] + ], + [ + [ + 24071 + ], + [ + -29434 + ] + ], + [ + [ + -9186 + ], + [ + 16364 + ] + ] + ] + ], + [ + [ + [ + [ + 2545 + ], + [ + -15886 + ] + ], + [ + [ + 1623 + ], + [ + 24026 + ] + ], + [ + [ + -18135 + ], + [ + 16126 + ] + ], + [ + [ + 15385 + ], + [ + 25277 + ] + ], + [ + [ + -13243 + ], + [ + -18618 + ] + ], + [ + [ + 3123 + ], + [ + 2050 + ] + ], + [ + [ + -5929 + ], + [ + 18302 + ] + ] + ] + ], + [ + [ + [ + [ + 19598 + ], + [ + -16666 + ] + ], + [ + [ + 19367 + ], + [ + -17646 + ] + ], + [ + [ + -6054 + ], + [ + -30987 + ] + ], + [ + [ + -10460 + ], + [ + 30049 + ] + ], + [ + [ + -3765 + ], + [ + 13763 + ] + ], + [ + [ + -25695 + ], + [ + 8884 + ] + ], + [ + [ + -8417 + ], + [ + 15990 + ] + ] + ] + ] + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/test.json b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/test.json new file mode 100644 index 000000000..b14143307 --- /dev/null +++ b/operators/tensor/argmax/argmax_8x1x7x2x1_i16_axis0/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: 0 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 8, + 1, + 7, + 2, + 1 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 1, + 7, + 2, + 1 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] + } + ] +} |