aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2023-02-07 20:11:41 +0000
committerEric Kunze <eric.kunze@arm.com>2023-02-22 16:27:57 +0000
commitae25b93e7a1852ed9eb50c647bff03f17055fbf8 (patch)
tree0b9af04786bc1bcea00ecfd5ef166450c793cb43
parent6587242465f1f2e54b40c681e8dc599ec2c1ba58 (diff)
downloadreference_model-ae25b93e7a1852ed9eb50c647bff03f17055fbf8.tar.gz
Create MI tests for Tensors: DEPTHWISE_CONV2D
Signed-off-by: James Ward <james.ward@arm.com> Change-Id: I6c2087c94c2c024aabc2086d40cbca9c719c69dd
-rw-r--r--verif/conformance/tosa_main_profile_ops_info.json55
1 files changed, 55 insertions, 0 deletions
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index 675ec9f..db4bd66 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -575,6 +575,61 @@
"tosa-mi"
]
},
+ "depthwise_conv2d": {
+ "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-shape",
+ "1,17,31,4",
+ "--target-shape",
+ "1,37,11,5",
+ "--allow-pooling-and-conv-oversizes"
+ ],
+ [
+ "--target-dtype",
+ "fp32",
+ "--fp-values-range",
+ "-2.0,2.0",
+ "--target-shape",
+ "1,1,65531,2",
+ "--target-shape",
+ "1,65540,1,3",
+ "--max-conv-dilation",
+ "1",
+ "--allow-pooling-and-conv-oversizes"
+ ]
+ ],
+ "params": {
+ "shape": [],
+ "type": [
+ "f16xf16",
+ "bf16xbf16",
+ "f32xf32"
+ ],
+ "kernel": [],
+ "stride": [],
+ "pad": [],
+ "dilation": []
+ },
+ "permutes": [
+ "kernel",
+ "shape",
+ "type",
+ "pad"
+ ],
+ "profile": [
+ "tosa-mi"
+ ]
+ },
"equal": {
"group": "comparison",
"generator_args": [