| { |
| "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": [ |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,49,42,28", |
| "--target-shape", |
| "3,11,44,3", |
| "--allow-pooling-and-conv-oversizes" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,65535,5,1", |
| "--target-shape", |
| "1,3,65537,1", |
| "--allow-pooling-and-conv-oversizes" |
| ] |
| ], |
| "params": { |
| }, |
| "permutes": [ |
| "shape", |
| "type", |
| "accum_type" |
| ], |
| "sparsity": { |
| "pad": 17 |
| }, |
| "profile": [ |
| "tosa-mi" |
| ] |
| }, |
| "concat": { |
| "group": "data_layout", |
| "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", |
| "--num-const-inputs-concat", |
| "3" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--tensor-dim-range", |
| "1,16", |
| "--target-rank", |
| "4", |
| "--target-rank", |
| "5", |
| "--num-const-inputs-concat", |
| "1" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,1,65534,1,1", |
| "--target-shape", |
| "1,65537,1,2", |
| "--num-const-inputs-concat", |
| "2" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type" |
| ], |
| "profile": [ |
| "tosa-mi" |
| ] |
| }, |
| "conv2d": { |
| "group": "tensor", |
| "generator_args": [ |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,49,42,28", |
| "--target-shape", |
| "1,11,44,13", |
| "--allow-pooling-and-conv-oversizes" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,65535,4,1", |
| "--target-shape", |
| "1,5,65536,1", |
| "--max-conv-dilation", |
| "1", |
| "--allow-pooling-and-conv-oversizes" |
| ] |
| ], |
| "params": { |
| "shape": [], |
| "type": [], |
| "kernel": [], |
| "stride": [], |
| "pad": [], |
| "dilation": [] |
| }, |
| "permutes": [ |
| "kernel", |
| "shape", |
| "type", |
| "pad" |
| ], |
| "profile": [ |
| "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": [ |
| [ |
| "--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" |
| ] |
| }, |
| "pad": { |
| "group": "data_layout", |
| "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" |
| ], |
| [ |
| "--target-dtype", |
| "fp16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--tensor-dim-range", |
| "1,17", |
| "--target-rank", |
| "4" |
| ], |
| [ |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--tensor-dim-range", |
| "1,16", |
| "--target-rank", |
| "5" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,1,65539,1" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type" |
| ], |
| "sparsity": { |
| "pad": 21 |
| }, |
| "preselected": [ |
| { |
| "shape": "50", |
| "type": "bf16", |
| "pad": "pad11" |
| }, |
| { |
| "shape": "63x46", |
| "type": "bf16", |
| "pad": "pad1010" |
| }, |
| { |
| "shape": "6", |
| "type": "f16", |
| "pad": "pad01" |
| } |
| ], |
| "profile": [ |
| "tosa-mi" |
| ] |
| }, |
| "reshape": { |
| "group": "data_layout", |
| "generator_args": [ |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--num-rand-permutations", |
| "2" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,1,65536,1", |
| "--target-shape", |
| "1,65535,1,2", |
| "--num-rand-permutations", |
| "2" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type", |
| "rank", |
| "perm" |
| ], |
| "profile": [ |
| "tosa-mi" |
| ] |
| }, |
| "reverse": { |
| "group": "data_layout", |
| "generator_args": [ |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--tensor-dim-range", |
| "1,48" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,1,65537", |
| "--target-shape", |
| "1,65535,1,2" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type" |
| ], |
| "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" |
| ] |
| }, |
| "tile": { |
| "group": "data_layout", |
| "generator_args": [ |
| [ |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--tensor-dim-range", |
| "4,32" |
| ], |
| [ |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "65535", |
| "--target-shape", |
| "1,65543,2,1" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type" |
| ], |
| "profile": [ |
| "tosa-mi" |
| ] |
| }, |
| "transpose": { |
| "group": "data_layout", |
| "generator_args": [ |
| [ |
| "--filter", |
| "^transpose$", |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-rank", |
| "1", |
| "--target-rank", |
| "2", |
| "--num-rand-permutations", |
| "2" |
| ], |
| [ |
| "--filter", |
| "^transpose$", |
| "--target-dtype", |
| "fp32", |
| "--target-dtype", |
| "fp16", |
| "--target-dtype", |
| "bf16", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-rank", |
| "3", |
| "--target-rank", |
| "4", |
| "--num-rand-permutations", |
| "2" |
| ], |
| [ |
| "--filter", |
| "^transpose$", |
| "--target-dtype", |
| "fp32", |
| "--fp-values-range", |
| "-2.0,2.0", |
| "--target-shape", |
| "1,1,65539,2", |
| "--target-shape", |
| "1,1,1,65535", |
| "--num-rand-permutations", |
| "1" |
| ] |
| ], |
| "params": {}, |
| "permutes": [ |
| "shape", |
| "type", |
| "perm" |
| ], |
| "profile": [ |
| "tosa-mi" |
| ] |
| } |
| } |