{
  "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" }
        ]
      }
    }
  ]
}