{ "name": "ai.onnx.TopK", "id": "_ai_onnx_topk_webgpu_52be754", "version": 1, "license": "Apache-2.0", "backend": { "type": "webgpu" }, "digest": { "algorithm": "sha256", "files": { "bench.json": "PurEWvvW6gbYNkFFlfRDPuV5w56IJxPpwT0HV0cXd0A=", "manifest.json": "YqipHsTNrefVMyk2lSH1Vp9eXwjmXHibQ+FXiPhg3iw=", "test.json": "Fr5n36m9d6NTCaetFemmXK+rj/6dXPaw6INsNyX9jqs=", "topk-axis.wgsl.jinja": "Xl+bV55WupVkNPIof1xc0YfVFd0YlmTvWBoVKXuKH0s=", "topk-large-block.wgsl.jinja": "r0f8d02UPzDntSxjLxFsP7ix3osvWFRmzsyt6YQS5K8=", "topk-noop.wgsl.jinja": "rzwkvcZd8vJ3qLSosuI96CfUW+WUQyFxjc2Hgvq1bdo=", "topk-portable-rows-smallk.wgsl.jinja": "WLo2yQv5mq1yMAW6vNAFbCSzRwuQ8fvM3MtuYH/d5PM=", "topk-small-rows-batched.wgsl.jinja": "zaHFjRfth6UmWbFEYmjRbE2hpIdjuqyafituQUBTGvg=", "topk-strided-smallk.wgsl.jinja": "pMdoMVSn+bYSKpP/FHMFLL2hnGeMQyakRPPDTITae4I=", "topk-subgroup-rows.wgsl.jinja": "NPnHo98cYexGK/mRtrzxZvXOjmVF/1soDP/EaBRaApw=", "topk-top1-last-axis.wgsl.jinja": "AYmJPcushgFs3ixDK6sJtDQDNEGbBk/CAAPWRMF3g+Y=" } }, "provenance": { "kernel": { "sha": "91d990483a174128daf7673f3f37a7c890493ae1", "dirty": false } }, "webgpu": { "manifestSpec": "2.0", "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"] } } }