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