diff options
author | James Ward <james.ward@arm.com> | 2023-02-07 20:11:41 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-22 16:27:57 +0000 |
commit | ae25b93e7a1852ed9eb50c647bff03f17055fbf8 (patch) | |
tree | 0b9af04786bc1bcea00ecfd5ef166450c793cb43 | |
parent | 6587242465f1f2e54b40c681e8dc599ec2c1ba58 (diff) | |
download | reference_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.json | 55 |
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": [ |