| { |
| "op": "ai.onnx.Scatter", |
| "cases": [ |
| { |
| "name": "axis1-1024x768-f32", |
| "preset": "smoke", |
| "attrs": { "axis": 1 }, |
| "inputs": { |
| "data": { "dtype": "float32", "shape": [1024, 768] }, |
| "indices": { "dtype": "int32", "shape": [1024, 128], "min": 0, "max": 768 }, |
| "updates": { "dtype": "float32", "shape": [1024, 128] } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1024, 768] } }, |
| "bench": { |
| "metrics": [ |
| { |
| "type": "bandwidth", |
| "value": "4 * (numel(shapes.data) + numel(shapes.output) + numel(shapes.indices) + 2 * numel(shapes.updates))" |
| } |
| ] |
| } |
| } |
| ] |
| } |
|
|
