File size: 2,390 Bytes
b6dbd82 2fc9420 b6dbd82 2fc9420 b6dbd82 2fc9420 b6dbd82 2fc9420 b4e7a99 2fc9420 b4e7a99 b6dbd82 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | {
"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"]
}
}
}
|