diff options
author | Won Jeon <won.jeon@arm.com> | 2023-08-10 10:33:01 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2023-08-18 15:21:15 -0700 |
commit | a21b2e88d19d8cb11a9120d40bacbb594d600565 (patch) | |
tree | 3bc8a40db72a31c1e552a3bd6339627a1175686e /verif/conformance | |
parent | e0247481eb1f83f6eb7161d3f7ac2690b180952a (diff) | |
download | reference_model-a21b2e88d19d8cb11a9120d40bacbb594d600565.tar.gz |
Add DIM operator to reference model
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b
Diffstat (limited to 'verif/conformance')
-rw-r--r-- | verif/conformance/test_select.py | 7 | ||||
-rw-r--r-- | verif/conformance/tosa_base_profile_ops_info.json | 54 | ||||
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 39 |
3 files changed, 100 insertions, 0 deletions
diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py index f4e2a61..b7bbfc3 100644 --- a/verif/conformance/test_select.py +++ b/verif/conformance/test_select.py @@ -506,6 +506,13 @@ class DepthwiseConv2dOperator(Operator): param_names = ["kernel", "shape", "type", "accum_type", "stride", "pad", "dilation"] +class DimOeprator(Operator): + """Test selector for the DIM operator.""" + + name = "dim" + param_names = ["shape", "type", "axis"] + + class EqualOperator(Operator): """Test selector for the EQUAL operator.""" diff --git a/verif/conformance/tosa_base_profile_ops_info.json b/verif/conformance/tosa_base_profile_ops_info.json index 4e3cd03..772602b 100644 --- a/verif/conformance/tosa_base_profile_ops_info.json +++ b/verif/conformance/tosa_base_profile_ops_info.json @@ -1317,6 +1317,60 @@ } } }, + "dim": { + "group": "data_layout", + "profile": [ + "tosa-bi", + "tosa-mi" + ], + "generation": { + "standard": { + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + }, + "8k_level": { + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, "equal": { "group": "comparison", "profile": [ diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index 07b6af3..7388835 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -787,6 +787,45 @@ } } }, + "dim": { + "group": "data_layout", + "profile": [ + "tosa-mi" + ], + "generation": { + "standard": { + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,65", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, "equal": { "group": "comparison", "profile": [ |