diff options
author | James Ward <james.ward@arm.com> | 2022-11-16 12:53:49 +0000 |
---|---|---|
committer | James Ward <james.ward@arm.com> | 2022-11-21 12:07:10 +0000 |
commit | 52ac9d6e4afa916acc4cab0617b940f4ab7aaf4f (patch) | |
tree | ed3cee18ef7c4820b83f5da26ad5b16b41ba51f5 /verif/conformance/tosa_main_profile_ops_info.json | |
parent | a633bc61b156f3a4a2cd9f313a738ace459befc1 (diff) | |
download | reference_model-52ac9d6e4afa916acc4cab0617b940f4ab7aaf4f.tar.gz |
Create MI tests for Tensors: ARGMAX, MAX_POOL2D
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I908d88ddf7c52d98ce6d533396b5c439f15750f5
Diffstat (limited to 'verif/conformance/tosa_main_profile_ops_info.json')
-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 |