| { | |
| "op": "ai.onnx.Size", | |
| "cases": [ | |
| { | |
| "name": "rank4", | |
| "inputs": { "data": { "dtype": "float32", "shape": [1, 16, 128, 128] } }, | |
| "outputs": { "size": { "dtype": "uint32", "shape": [1] } }, | |
| "bench": { "metrics": [{ "type": "bandwidth", "name": "write-BW", "value": "4 * numel(shapes.size)" }] }, | |
| "preset": "smoke" | |
| } | |
| ] | |
| } | |