diff options
| author | James Ward <james.ward@arm.com> | 2023-01-27 17:27:34 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-06 19:02:05 +0000 |
| commit | 8384a6f384bda904bad68431adca6e5b4deb9324 (patch) | |
| tree | a00c701b695a13d46585694731422d5633664215 | |
| parent | 9165b9ad6bf80a924edc91bcd88e193e78d590b4 (diff) | |
| download | reference_model-8384a6f384bda904bad68431adca6e5b4deb9324.tar.gz | |
Create MI tests for Tensors: AVG_POOL2D, FULLY_CONNECTED, MATMUL
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I874c87745b929f9808a5e45490042420a57b22f7
| -rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 105 |
1 files changed, 99 insertions, 6 deletions
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index 81cb666..7b05477 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -185,9 +185,9 @@ "--fp-values-range", "-2.0,2.0", "--target-shape", - "1,49,42,28", + "1,11,7,39", "--target-shape", - "3,11,44,3", + "1,21,36,7", "--allow-pooling-and-conv-oversizes" ], [ @@ -196,9 +196,9 @@ "--fp-values-range", "-2.0,2.0", "--target-shape", - "1,65535,5,1", + "1,2,65533,2", "--target-shape", - "1,3,65537,1", + "1,65539,2,1", "--allow-pooling-and-conv-oversizes" ] ], @@ -206,8 +206,7 @@ }, "permutes": [ "shape", - "type", - "accum_type" + "type" ], "sparsity": { "pad": 17 @@ -682,6 +681,52 @@ "tosa-mi" ] }, + "fully_connected": { + "group": "tensor", + "generator_negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0" + ], + [ + "--target-dtype", + "fp32", + "--target-shape", + "1,296", + "--target-shape", + "65540,2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "3,16", + "--target-shape", + "1,23" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] + }, "floor": { "group": "ew_unary", "generator_args": [ @@ -1005,6 +1050,54 @@ "tosa-mi" ] }, + "matmul": { + "group": "tensor", + "generator_negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "13,70" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,19" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "4,65530,1", + "--target-shape", + "1,65538,3" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] + }, "max_pool2d": { "group": "tensor", "generator_args": [ |