diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-06-30 14:27:56 +0100 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2022-07-12 11:28:15 +0100 |
commit | 0ecfa37738d56cbb50af584e9bf077052094f460 (patch) | |
tree | c3b6d68ddd458ec6a1a409245ca860ed07522ce4 /verif/conformance/tosa_base_profile_ops_info.json | |
parent | 6c6467f3f07687f0eebd049f1fa3ce21c516f184 (diff) | |
download | reference_model-0ecfa37738d56cbb50af584e9bf077052094f460.tar.gz |
Add conformance generator scripts
Change-Id: I5cb16ea1f47ee454f03a1d5182827e3fd9f49128
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Diffstat (limited to 'verif/conformance/tosa_base_profile_ops_info.json')
-rw-r--r-- | verif/conformance/tosa_base_profile_ops_info.json | 2220 |
1 files changed, 2220 insertions, 0 deletions
diff --git a/verif/conformance/tosa_base_profile_ops_info.json b/verif/conformance/tosa_base_profile_ops_info.json new file mode 100644 index 0000000..79e50ff --- /dev/null +++ b/verif/conformance/tosa_base_profile_ops_info.json @@ -0,0 +1,2220 @@ +{ + "abs": { + "group": "ew_unary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "add": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "argmax": { + "group": "tensor", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-shape", + "1,3,1,65535", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8", + "--num-rand-permutations", + "2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "arithmetic_right_shift": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "rounding" + ] + }, + "avg_pool2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,49,42,28", + "--target-shape", + "3,11,44,3", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,65535,5,1", + "--target-shape", + "1,3,65537,1", + "--target-dtype", + "int8", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "i8", + "i16" + ], + "kernel": [], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ] + }, + "bitwise_and": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "bitwise_not": { + "group": "ew_unary", + "generator_args": [ + [ + "--tensor-dim-range", + "2,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65537,1", + "--target-shape", + "1,65538,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "bitwise_or": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "3,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65538,1", + "--target-shape", + "1,65537,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "bitwise_xor": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "4,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65536,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "cast": { + "group": "type_conversion", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,1,65535,1,1", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8" + ] + ], + "params": { + "shape": [], + "type": [], + "output_type": [ + "outINT8", + "outINT16", + "outINT32", + "outBOOL" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ] + }, + "clamp": { + "group": "activation", + "generator_args": [ + [ + "--tensor-dim-range", + "20,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,1,65536", + "--target-shape", + "1,3,65537,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "clz": { + "group": "ew_unary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "concat": { + "group": "data_layout", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ], + [ + "--target-shape", + "1,1,65535,1,1", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8", + "--num-const-inputs-concat", + "2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "cond_if": { + "group": "control_flow", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32" + ], + [ + "--target-shape", + "3,22,17,1,2", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "variant", + "shape", + "cond" + ] + }, + "const": { + "group": "data_nodes", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,1,65537,1,1", + "--target-shape", + "1,65530,1,2", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "conv2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,49,42,28", + "--target-shape", + "1,11,44,13", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,65535,4,1", + "--target-shape", + "1,5,65536,1", + "--target-dtype", + "int8", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "conv3d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,9,21,14,1", + "--target-shape", + "1,9,11,12,3", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,1,65535,3,1", + "--target-shape", + "1,1,3,65536,1", + "--target-dtype", + "int8", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ] + }, + "depthwise_conv2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,9,44,3", + "--target-shape", + "1,11,33,4", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,65532,2,1", + "--target-shape", + "1,4,65537,1", + "--target-dtype", + "int8", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "equal": { + "group": "comparison", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65501,1", + "--target-shape", + "1,65541,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "fully_connected": { + "group": "tensor", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16" + ], + [ + "--target-shape", + "444,1", + "--target-shape", + "3,65538", + "--tensor-dim-range", + "1,2", + "--target-dtype", + "int8" + ], + [ + "--target-shape", + "3,16", + "--target-shape", + "1,23", + "--tensor-dim-range", + "100,200", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "gather": { + "group": "scatter_gather", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,64" + ], + [ + "--target-shape", + "2,65533,1", + "--target-shape", + "1,2,65533", + "--target-shape", + "5000,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "greater": { + "group": "comparison", + "generator_args": [ + [ + "--filter", + "^greater$", + "--target-dtype", + "int32", + "--tensor-dim-range", + "10,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--filter", + "^greater$", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--filter", + "^greater$", + "--target-shape", + "1,2,65530,1", + "--target-shape", + "1,65539,1,2,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "greater_equal": { + "group": "comparison", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "20,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,3,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "identity": { + "group": "data_nodes", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,61", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,1,65536,1,1", + "--target-shape", + "1,65531,1,2", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "intdiv": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65533,1", + "--target-shape", + "1,1,65539,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_and": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65538,1", + "--target-shape", + "1,2,1,65536" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_left_shift": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65532,1,1", + "--target-shape", + "1,2,1,65538", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_not": { + "group": "ew_unary", + "generator_args": [ + [ + "--tensor-dim-range", + "2,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65537,1", + "--target-shape", + "1,2,1,65535" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_or": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "3,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,14", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65536,1", + "--target-shape", + "1,2,1,65537" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_right_shift": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "25,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,25", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,1,65540,3", + "--target-shape", + "2,1,65530,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "logical_xor": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "4,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "3,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,65539,1", + "--target-shape", + "1,2,1,65534" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "matmul": { + "group": "tensor", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "16,64" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16" + ], + [ + "--target-shape", + "1,65532,1", + "--target-shape", + "1,2,65538", + "--target-shape", + "2000,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "maximum": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "max_pool2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,4,75,3", + "--target-shape", + "2,11,44,1", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,65535,5,1", + "--target-shape", + "1,3,65537,1", + "--target-shape", + "33333,3,2,1", + "--target-dtype", + "int8", + "--allow-pooling-and-conv-oversizes" + ] + ], + "params": { + "shape": [], + "type": [ + "i8", + "i16" + ], + "kernel": [], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ] + }, + "minimum": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "mul": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65500,1", + "--target-shape", + "1,68539,1,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "perm", + "shift" + ] + }, + "negate": { + "group": "ew_unary", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "22,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,22", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "3,1,65540,1", + "--target-shape", + "1,1,2,1,65531", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "pad": { + "group": "data_layout", + "generator_args": [ + [ + "--tensor-dim-range", + "1,64", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "4" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-rank", + "5", + "--target-dtype", + "int16" + ], + [ + "--target-shape", + "1,1,65537,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ], + "sparsity": { + "pad": 15 + }, + "preselected": [ + { + "shape": "6", + "type": "i8", + "pad": "pad00" + } + ] + }, + "reduce_all": { + "group": "reduction", + "generator_args": [ + [ + "--tensor-dim-range", + "1,32" + ], + [ + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + }, + "reduce_any": { + "group": "reduction", + "generator_args": [ + [ + "--tensor-dim-range", + "1,32" + ], + [ + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + }, + "reduce_max": { + "group": "reduction", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,32" + ], + [ + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + }, + "reduce_min": { + "group": "reduction", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,24" + ], + [ + "--target-shape", + "1,1,65538,1", + "--target-shape", + "2,65535,1,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + }, + "reduce_sum": { + "group": "reduction", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,32" + ], + [ + "--target-shape", + "3,65532,1,1", + "--target-shape", + "65536,1,1,2", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + }, + "rescale": { + "group": "type_conversion", + "generator_args": [ + [ + "--tensor-dim-range", + "16,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-rank", + "4" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1", + "--target-dtype", + "int8" + ] + ], + "params": { + "scale": [ + "sc0", + "sc0", + "sc1", + "sc1", + "sc1", + "sc1" + ], + "double_round": [ + "dr0", + "dr0", + "dr0", + "dr0", + "dr1", + "dr1" + ], + "per_channel": [ + "pc0", + "pc1", + "pc0", + "pc1", + "pc0", + "pc1", + "pc1" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ] + }, + "reshape": { + "group": "data_layout", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--num-rand-permutations", + "2" + ], + [ + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8", + "--num-rand-permutations", + "2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "rank", + "perm" + ] + }, + "resize": { + "group": "image", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16" + ], + [ + "--target-shape", + "1,3,16383,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "reverse": { + "group": "data_layout", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,32" + ], + [ + "--target-shape", + "1,1,65535", + "--target-shape", + "1,65540,1,2", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "scatter": { + "group": "scatter_gather", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,64" + ], + [ + "--target-shape", + "1,65540,1", + "--target-shape", + "1,1,65533", + "--target-shape", + "3000,7,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "select": { + "group": "ew_ternary", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8", + "--num-rand-permutations", + "2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "slice": { + "group": "data_layout", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--target-dtype", + "int8", + "--num-rand-permutations", + "2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "perm" + ] + }, + "sub": { + "group": "ew_binary", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "table": { + "group": "ew_binary", + "generator_args": [ + [ + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65537,1,1,3", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tile": { + "group": "data_layout", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,32" + ], + [ + "--target-shape", + "65533", + "--target-shape", + "1,2,65540,1", + "--target-dtype", + "int8" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "transpose": { + "group": "data_layout", + "generator_args": [ + [ + "--filter", + "^transpose$", + "--target-dtype", + "int8", + "--target-rank", + "1", + "--target-rank", + "2", + "--num-rand-permutations", + "2" + ], + [ + "--filter", + "^transpose$", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--filter", + "^transpose$", + "--target-shape", + "1,65537,1,2", + "--target-shape", + "65535,1,1,1", + "--target-dtype", + "int8", + "--num-rand-permutations", + "1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "perm" + ] + }, + "transpose_conv2d": { + "group": "tensor", + "generator_args": [ + [ + "--target-shape", + "1,49,33,1", + "--target-shape", + "2,11,33,3", + "--target-dtype", + "int16", + "--target-dtype", + "int8", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-shape", + "1,65536,1,1", + "--target-dtype", + "int8", + "--max-conv-dilation", + "1" + ] + ], + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "out_shape": [] + }, + "permutes": [ + "shape", + "type", + "stride", + "pad" + ] + }, + "while_loop": { + "group": "control_flow", + "generator_args": [ + [ + "--target-dtype", + "int32" + ], + [ + "--target-shape", + "3,22,17,1,2", + "--target-shape", + "1,65537,2,1", + "--target-dtype", + "int32" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type" + ] + } +}
\ No newline at end of file |