diff options
-rw-r--r-- | verif/conformance/test_select.py | 12 | ||||
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 114 |
2 files changed, 126 insertions, 0 deletions
diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py index 99588a9..696b466 100644 --- a/verif/conformance/test_select.py +++ b/verif/conformance/test_select.py @@ -684,6 +684,12 @@ class SelectOperator(Operator): name = "select" +class SigmoidOperator(Operator): + """Test selector for the SIGMOID operator.""" + + name = "sigmoid" + + class SliceOperator(Operator): """Test selector for the SLICE operator.""" @@ -703,6 +709,12 @@ class TableOperator(Operator): name = "table" +class TanhOperator(Operator): + """Test selector for the TANH operator.""" + + name = "tanh" + + class TileOperator(Operator): """Test selector for the TILE operator.""" diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index ce15b2f..4328b9f 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -1183,5 +1183,119 @@ "profile": [ "tosa-mi" ] + }, + "sigmoid": { + "group": "activation", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,19", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,2,65538,2,1", + "--target-shape", + "1,3,1,65533" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] + }, + "tanh": { + "group": "activation", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "18,60", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,24", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,65535,2,1,1", + "--target-shape", + "1,65540,1,2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] } }
\ No newline at end of file |