| { |
| "op": "ai.onnx.SpaceToDepth", |
| "cases": [ |
| { |
| "name": "nchw_1x64x128x128", |
| "attrs": { "blocksize": 2 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [1, 64, 128, 128] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 256, 64, 64] } } |
| }, |
| { |
| "name": "nchw_1x32x192x192_block3", |
| "attrs": { "blocksize": 3 }, |
| "inputs": { "input": { "dtype": "float32", "shape": [1, 32, 192, 192] } }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 288, 64, 64] } } |
| }, |
| { |
| "name": "nchw_f16_1x64x128x128", |
| "preset": "smoke", |
| "attrs": { "blocksize": 2 }, |
| "vars": { "dtype": "float16", "count": 1048576 }, |
| "inputs": { |
| "input": { "dtype": "float16", "shape": [1, 64, 128, 128], "dist": "normal", "seed": 4471, "scale": 1 } |
| }, |
| "outputs": { "output": { "dtype": "float16", "shape": [1, 256, 64, 64] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| }, |
| { |
| "name": "nchw_dispatch_cliff_1x4194241x2x2", |
| "attrs": { "blocksize": 2 }, |
| "vars": { "dtype": "float32", "count": 16776964 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 4194241, 2, 2], "dist": "normal", "seed": 4472, "scale": 1 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 16776964, 1, 1] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| }, |
| { |
| "name": "nchw-const-fallback-odd-width-block2-f32-1x128x2048x6", |
| "preset": "stress", |
| "attrs": { "blocksize": 2 }, |
| "vars": { "dtype": "float32", "count": 1572864 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [1, 128, 2048, 6], "dist": "normal", "seed": 5201, "scale": 1 } |
| }, |
| "outputs": { "output": { "dtype": "float32", "shape": [1, 512, 1024, 3] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| } |
| ] |
| } |
|
|
