diff options
| author | James Ward <james.ward@arm.com> | 2022-11-16 15:36:37 +0000 |
|---|---|---|
| committer | James Ward <james.ward@arm.com> | 2022-11-21 12:08:36 +0000 |
| commit | 990cc4be0b2c33f470e42ce7761e06a5b072afe6 (patch) | |
| tree | b1ebeab2440422934f8b82a614a8cb313ae26bb6 | |
| parent | 52ac9d6e4afa916acc4cab0617b940f4ab7aaf4f (diff) | |
| download | reference_model-990cc4be0b2c33f470e42ce7761e06a5b072afe6.tar.gz | |
Create MI tests for Scatter/Gather: SCATTER, GATHER
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I44c0843fa049546582ca1111795342d5035a329c
| -rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 74 |
1 files changed, 74 insertions, 0 deletions
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index 0614dfc..9846c29 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -155,6 +155,43 @@ "tosa-mi" ] }, + "gather": { + "group": "scatter_gather", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "5,72" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "2,65536,1", + "--target-shape", + "1,2,65535", + "--target-shape", + "5000,1,1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] + }, "max_pool2d": { "group": "tensor", "generator_args": [ @@ -208,5 +245,42 @@ "profile": [ "tosa-mi" ] + }, + "scatter": { + "group": "scatter_gather", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "5,56" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,65541,1", + "--target-shape", + "1,1,65531", + "--target-shape", + "2989,6,1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "profile": [ + "tosa-mi" + ] } }
\ No newline at end of file |