diff options
Diffstat (limited to 'operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0')
3 files changed, 86 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/desc.json new file mode 100644 index 000000000..2fc8edae4 --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/desc.json @@ -0,0 +1,21 @@ +{ + "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": 2, + "expected_failure": true, + "expected_failure_desc": "Mismatch between output shape provided and expected output shape", + "profile": [ + "tosa-mi" + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/input-0.json new file mode 100644 index 000000000..5dcb2ff21 --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/input-0.json @@ -0,0 +1,15 @@ +{ + "type": "float16", + "data": [ + 0.60009765625, + 0.2376708984375, + 0.54150390625, + 0.60009765625, + 0.61865234375, + 0.6826171875, + 0.06903076171875, + 0.70361328125, + 0.154541015625, + 0.67724609375 + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json new file mode 100644 index 000000000..4efc31ab3 --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_ArgmaxOutputRankMismatch_10_f16_axis0/test.json @@ -0,0 +1,50 @@ +{ + 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: [ + 10 + ], + type: "FP16" + }, + { + name: "result-0", + shape: [ + 1 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] +} |