| { |
| "op": "ai.onnx.SplitToSequence", |
| "cases": [ |
| { |
| "name": "split3-1536x256-f32", |
| "attrs": { "axis": 0 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1536, 256] }, |
| "split": { "dtype": "uint32", "shape": [3], "data": { "kind": "values", "values": [512, 512, 512] } } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [512, 256] }, |
| "y1": { "dtype": "float32", "shape": [512, 256] }, |
| "y2": { "dtype": "float32", "shape": [512, 256] } |
| }, |
| "preset": "smoke" |
| }, |
| { |
| "name": "rank6_squeeze_six_outputs_addr_arith_f32_stress", |
| "preset": "stress", |
| "attrs": { "axis": 1, "keepdims": 0 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [8, 6, 4, 4, 16, 64], "dist": "normal", "seed": 1234, "scale": 2 }, |
| "split": { "dtype": "uint32", "shape": [], "dist": "constant", "value": 1 } |
| }, |
| "outputs": { |
| "y0": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" }, |
| "y1": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" }, |
| "y2": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" }, |
| "y3": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" }, |
| "y4": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" }, |
| "y5": { "dtype": "float32", "shape": [8, 4, 4, 16, 64], "dist": "empty" } |
| }, |
| "bench": { |
| "primary": true, |
| "metrics": [ |
| { "type": "elements", "value": "numel(shapes.input)" }, |
| { "type": "bandwidth", "value": "numel(shapes.input)*4*2", "unit": "GB/s" } |
| ] |
| } |
| } |
| ] |
| } |
|
|
