| { |
| "name": "ai.onnx.TopK", |
| "id": "_ai_onnx_topk_webgpu_86e76bd", |
| "version": 1, |
| "license": "Apache-2.0", |
| "backend": { "type": "webgpu" }, |
| "digest": { |
| "algorithm": "sha256", |
| "files": { |
| "bench.json": "BstQ++nxAbgXP9WH+pnBB9P9NQgjR0tSlMSWpKyJ2/g=", |
| "manifest.json": "8xiJsa7sv4cFwwhLkkhdi885RdU06+N9dIKbPOuNQCA=", |
| "test.json": "cmbkkLOf0Y81WcqvDMn2iy5hoqB1Xv/6FGQV2qUgsd8=", |
| "topk-axis.wgsl.jinja": "tUaDaFWKpqQzThqgb9+44QmO68711FQ3vSVb0dTGsVs=", |
| "topk-large-block.wgsl.jinja": "mbFLg/zlRmGc8ERy+pGUE1iB45HecG/hojBA2WUUhYU=", |
| "topk-noop.wgsl.jinja": "rzwkvcZd8vJ3qLSosuI96CfUW+WUQyFxjc2Hgvq1bdo=", |
| "topk-portable-rows-smallk.wgsl.jinja": "4hylCNDZPtPJbDrY1EUyclDQb2bO1ghtcmlvITOp+7o=", |
| "topk-small-rows-batched.wgsl.jinja": "dnFSn35qHvZ2VcB6SfRmBLkBwJIwnYs5FqXbNSBWW8k=", |
| "topk-strided-smallk.wgsl.jinja": "n+03N8pyJQ1YbZIRO1Cytk1dckDjcq4rQWJrbpSyw+8=", |
| "topk-subgroup-rows.wgsl.jinja": "bRksgwaG+gsWNpW1sONw2BIJJcc75bk7K+vi863Yx2Y=", |
| "topk-top1-last-axis.wgsl.jinja": "083lZZtsapk+C0Hy4cSsrEvnX3y2N7WCOKyq3pISxUc=" |
| } |
| }, |
| "provenance": { "kernel": { "sha": "6fdf6301e2bbcc2f03bf1eaf493b7ad55ef33afc", "dirty": false } }, |
| "webgpu": { |
| "manifestSpec": "2.1", |
| "variants": { |
| "axis_smallk_tournament": ["topk-strided-smallk.wgsl.jinja"], |
| "last_axis_large_top1": ["topk-top1-last-axis.wgsl.jinja"], |
| "last_axis_top1": ["topk-top1-last-axis.wgsl.jinja"], |
| "axis_bitonic": ["topk-axis.wgsl.jinja"], |
| "top0_noop": ["topk-noop.wgsl.jinja"], |
| "subgroup_rows_smallk": ["topk-subgroup-rows.wgsl.jinja"], |
| "subgroup_min_rows_smallk": ["topk-subgroup-rows.wgsl.jinja"], |
| "portable_rows_smallk": ["topk-portable-rows-smallk.wgsl.jinja"], |
| "small_rows_batched": ["topk-small-rows-batched.wgsl.jinja"], |
| "last_axis_large_one_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_two_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_one_merge_int": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_two_merge_int": ["topk-large-block.wgsl.jinja"], |
| "axis_large_one_merge": ["topk-large-block.wgsl.jinja"], |
| "axis_large_two_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_three_merge": ["topk-large-block.wgsl.jinja"], |
| "last_axis_large_three_merge_int": ["topk-large-block.wgsl.jinja"] |
| } |
| } |
| } |
|
|
