| { |
| "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" |
| ] |
| }, |
| "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" |
| ] |
| } |
| } |