| { |
| "op": "ai.onnx.LRN", |
| "cases": [ |
| { |
| "name": "nchw_8_64_32_32", |
| "attrs": { "alpha": 0.0001, "beta": 0.75, "bias": 1, "size": 5 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [8, 64, 32, 32] } }, |
| "outputs": { "y": { "dtype": "float32", "shape": [8, 64, 32, 32] } } |
| }, |
| { |
| "name": "nchw_8_96_64_64_vec4w", |
| "preset": "smoke", |
| "attrs": { "alpha": 0.0001, "beta": 0.75, "bias": 1, "size": 5 }, |
| "vars": { "numel": 3145728 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [8, 96, 64, 64], "dist": "normal", "seed": 301 } }, |
| "outputs": { "y": { "dtype": "float32", "shape": [8, 96, 64, 64], "dist": "empty" } }, |
| "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "args.numel * 4 * 2" }] } |
| }, |
| { |
| "name": "nchw_scalar_path_unaligned_width_w6_stress", |
| "preset": "stress", |
| "attrs": { "alpha": 0.0001, "beta": 0.75, "bias": 1, "size": 5 }, |
| "vars": { "numel": 1179648 }, |
| "inputs": { "x": { "dtype": "float32", "shape": [8, 96, 256, 6], "dist": "normal", "seed": 4201, "scale": 1 } }, |
| "outputs": { "y": { "dtype": "float32", "shape": [8, 96, 256, 6], "dist": "empty" } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.numel * 4 * 2" }] } |
| } |
| ], |
| "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] } |
| } |
|
|
