diff options
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index db4bd66..dc4e45b 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -575,6 +575,58 @@ "tosa-mi" ] }, + "conv3d": { + "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,7,18,5,4", + "--target-shape", + "1,6,12,17,3", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,2,65539,1,2", + "--target-shape", + "1,65532,1,1,3", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ], + "profile": [ + "tosa-mi" + ] + }, "depthwise_conv2d": { "group": "tensor", "generator_negative_dim_range": "1,10", |