Create MI tests for Tensors: AVG_POOL2D, FULLY_CONNECTED, MATMUL
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I874c87745b929f9808a5e45490042420a57b22f7
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": [