diff options
author | James Ward <james.ward@arm.com> | 2022-12-12 13:02:44 +0000 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2022-12-15 12:45:27 +0000 |
commit | b45db9a696f5df7b233f374248f329c16ee7ae64 (patch) | |
tree | a01b74f62a38a9984f14b361f3d65d86aff66aa2 | |
parent | 2a27c8f5508cff3c748074c204965e138a434bd9 (diff) | |
download | reference_model-b45db9a696f5df7b233f374248f329c16ee7ae64.tar.gz |
Create MI tests for Activation: SIGMOID, TANH
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I75d6280ecb5219d50c91d14029e90ff3a2f6eb06
-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 |