diff options
| author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-10-13 13:54:23 +0100 |
|---|---|---|
| committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-10-13 13:54:23 +0100 |
| commit | 4129b45c7464197b7a3415c6865282352a7fa40e (patch) | |
| tree | b64a2c824b3b102150b842f460e2005ab801f514 /operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3 | |
| parent | 2caed563b3a5a394c38c42835ef3f0ce1dbece43 (diff) | |
| download | conformance_tests-4129b45c7464197b7a3415c6865282352a7fa40e.tar.gz | |
Updating ARGMAX tests for FP16 and accumulator types
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I96337ce765a04870b548f7905ab869c2e0287808
Diffstat (limited to 'operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3')
3 files changed, 87 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/desc.json b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/desc.json new file mode 100644 index 000000000..2c04c0be8 --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/desc.json @@ -0,0 +1,22 @@ +{ + "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": "Axis smaller than zero", + "profile": [ + "tosa-bi", + "tosa-mi" + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/input-0.json new file mode 100644 index 000000000..7b6c960de --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/input-0.json @@ -0,0 +1,15 @@ +{ + "type": "int32", + "data": [ + -8765, + 26776, + -324, + 13088, + -2791, + -15344, + 17316, + 30747, + -15498, + 18268 + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/test.json new file mode 100644 index 000000000..bf615c25c --- /dev/null +++ b/operators/tensor/argmax/argmax_ERRORIF_AxisSmallerZero_10_i16_axis-3/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: -3 + }, + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ], + tensors: [ + { + name: "input-0", + shape: [ + 10 + ], + type: "INT16" + }, + { + name: "result-0", + shape: [ + 10 + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] +} |