File size: 683 Bytes
0dcee9d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | {
"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)" }] }
}
]
}
|