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