diff options
author | James Ward <james.ward@arm.com> | 2022-11-18 15:10:49 +0000 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2022-11-18 15:10:49 +0000 |
commit | a25d672b327ad02b3813592803e41f5909aa0d5a (patch) | |
tree | 42143632c6bd0dff88743cc5284b8e4a25c0918a /operators/tensor/argmax/argmax_56_bf16_axis0 | |
parent | 874bb4fe493b2958a21e9a84c22828e82d4ff167 (diff) | |
download | conformance_tests-a25d672b327ad02b3813592803e41f5909aa0d5a.tar.gz |
Create MI tests for Tensors: ARGMAX, MAX_POOL2D
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I3682311cbdffa52fa88ae1b5779c615e2f062807
Diffstat (limited to 'operators/tensor/argmax/argmax_56_bf16_axis0')
4 files changed, 140 insertions, 0 deletions
diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json b/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json new file mode 100644 index 000000000..d576d4dc2 --- /dev/null +++ b/operators/tensor/argmax/argmax_56_bf16_axis0/Conformance-result-0.json @@ -0,0 +1,6 @@ +{ + "type": "int32", + "data": [ + 29 + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json b/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json new file mode 100644 index 000000000..d418e4060 --- /dev/null +++ b/operators/tensor/argmax/argmax_56_bf16_axis0/desc.json @@ -0,0 +1,23 @@ +{ + "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-mi" + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json b/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json new file mode 100644 index 000000000..f44785938 --- /dev/null +++ b/operators/tensor/argmax/argmax_56_bf16_axis0/input-0.json @@ -0,0 +1,61 @@ +{ + "type": "float32", + "data": [ + -0.75, + 1.328125, + 1.21875, + -0.44921875, + -0.84375, + 0.7265625, + -1.4375, + -1.1953125, + -1.96875, + 1.140625, + 0.65625, + 0.8203125, + 1.1171875, + -0.1640625, + 0.2734375, + -1.4375, + -1.5390625, + 0.671875, + -0.115234375, + 0.259765625, + 1.0546875, + 0.53515625, + 0.2138671875, + 0.236328125, + -0.78125, + -1.875, + -0.251953125, + -1.140625, + -0.365234375, + 1.40625, + -1.0625, + -1.765625, + -0.87109375, + -0.82421875, + 0.64453125, + 0.2275390625, + 1.1328125, + 0.65625, + -0.373046875, + 1.25, + -1.328125, + -1.90625, + -1.6328125, + 0.88671875, + -0.15234375, + -1.3515625, + 0.004150390625, + -1.390625, + 0.78515625, + -0.21484375, + -0.474609375, + -0.79296875, + 0.51953125, + -0.55078125, + -1.6484375, + -1.5234375 + ] +}
\ No newline at end of file diff --git a/operators/tensor/argmax/argmax_56_bf16_axis0/test.json b/operators/tensor/argmax/argmax_56_bf16_axis0/test.json new file mode 100644 index 000000000..7a15987fc --- /dev/null +++ b/operators/tensor/argmax/argmax_56_bf16_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: [ + 56 + ], + type: "BF16" + }, + { + name: "result-0", + shape: [ + + ], + type: "INT32" + } + ], + inputs: [ + "input-0" + ], + outputs: [ + "result-0" + ] + } + ] +} |