| { |
| "cases": [ |
| { |
| "name": "batch32_time128_hidden64", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [32, 128, 64] }, |
| "sequence_lens": { "dtype": "uint32", "shape": [32], "data": { "kind": "constant", "value": 128 } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [32, 128, 64] } } |
| }, |
| { |
| "name": "batch64_time512_hidden512", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [64, 512, 512] }, |
| "sequence_lens": { "dtype": "uint32", "shape": [64], "data": { "kind": "constant", "value": 512 } } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [64, 512, 512] } }, |
| "preset": "smoke" |
| }, |
| { |
| "name": "reverse-seq-scalar-fallback-inner-not-mul4-f32-4x2048x1023", |
| "preset": "stress", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "vars": { "dtype": "float32", "count": 8380416 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [4, 2048, 1023], "dist": "normal", "seed": 5101, "scale": 1 }, |
| "sequence_lens": { |
| "dtype": "uint32", |
| "shape": [4], |
| "data": { "kind": "values", "values": [2048, 1024, 2048, 512] } |
| } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 2048, 1023] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| }, |
| { |
| "name": "reverse-seq-scalar-fallback-inner-is-time-f32-4x524288", |
| "preset": "stress", |
| "attrs": { "batch_axis": 0, "time_axis": 1 }, |
| "vars": { "dtype": "float32", "count": 2097152 }, |
| "inputs": { |
| "input": { "dtype": "float32", "shape": [4, 524288] }, |
| "sequence_lens": { |
| "dtype": "uint32", |
| "shape": [4], |
| "data": { "kind": "values", "values": [524288, 262144, 524288, 131072] } |
| } |
| }, |
| "outputs": { "y": { "dtype": "float32", "shape": [4, 524288] } }, |
| "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] } |
| } |
| ] |
| } |
|
|
