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