File size: 613 Bytes
b1ec92f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | {
"op": "ai.onnx.If",
"cases": [
{
"name": "lowered_select_1m",
"preset": "smoke",
"inputs": {
"cond": { "dtype": "uint32", "shape": [1], "data": { "kind": "values", "values": [1] } },
"then_value": { "dtype": "float32", "shape": [1048576] },
"else_value": { "dtype": "float32", "shape": [1048576] }
},
"outputs": { "y": { "dtype": "float32", "shape": [1048576] } },
"bench": {
"metrics": [
{ "type": "bandwidth", "value": "4 * (numel(shapes.cond) + numel(shapes.then_value) + numel(shapes.y))" }
]
}
}
]
}
|