diff options
Diffstat (limited to 'verif')
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index 656cdde..0614dfc 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -1,4 +1,66 @@ { + "argmax": { + "group": "tensor", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,3,65535,1", + "--target-shape", + "1,65537,1,2", + "--num-rand-permutations", + "2" + ] + ], + "params": { + }, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] + }, "avg_pool2d": { "group": "tensor", "generator_args": [ @@ -92,5 +154,59 @@ "profile": [ "tosa-mi" ] + }, + "max_pool2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,47,37,25", + "--target-shape", + "1,15,13,38", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,65534,4,1", + "--target-shape", + "1,6,65536,1", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "f32", + "f16", + "bf16" + ], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ], + "sparsity": { + "pad": 11 + }, + "profile": [ + "tosa-mi" + ] } }
\ No newline at end of file |