File size: 4,976 Bytes
bc30b10 5974e09 bc30b10 5974e09 bc30b10 5974e09 7e43ef2 5974e09 7e43ef2 5974e09 7e43ef2 5974e09 7e43ef2 5974e09 7e43ef2 bc30b10 | 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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | {
"name": "ai.onnx.LSTM",
"id": "_ai_onnx_lstm_webgpu_db1746a",
"version": 1,
"license": "Apache-2.0",
"backend": { "type": "webgpu" },
"digest": {
"algorithm": "sha256",
"files": {
"bench.json": "R0YT6pRrkiftvosKKVrb1V16QrMAcrxnLCCc67VKKfg=",
"datamove-transpose-2d-tiled.wgsl.jinja": "DaZnArjagTZq8JXVP0GAm4tOHNYhu2AigaYZzR5rKzk=",
"lstm-general.wgsl.jinja": "CZStQGRy2OntVPfrmK/xqp1mQWcJJ22L6RLBWV8C/Xc=",
"lstm-parallel-global.wgsl.jinja": "7T+oZZYUTWCvclj5X3WLAOfB7YMBGU48T27JVybi4n8=",
"lstm-recur-step.wgsl.jinja": "smcdIdJzgUINCuUHRGZqgVQYjo1MkRAP0g/KAsbqjuM=",
"lstm-recur-subgroup-batched-step.wgsl.jinja": "YZ0bCtxm2CQJ85idBH59rVRw32gnR7kO6FvDg+EW/6w=",
"lstm-recur-subgroup-step.wgsl.jinja": "gXLJ5OfrLE1ohxKvzhC1kwiRDWeuxZGaOrQwk5pmMYE=",
"lstm-seed-state.wgsl.jinja": "pr8JDMbjl4DdJLnjcy38U5lte1xMDvxvx5j14emkafw=",
"lstm-timestep-body.wgsl.jinja": "MV7hyEOPI4EyKfdWBi3Qec5JJgHlWRVsczJfF2t2ff0=",
"manifest.json": "pkzYoKXemllh79MMy6EfnVuGCo7OUmlkYs8zWjRHjPs=",
"matmul-notrans-vec4-tiled-reg.wgsl.jinja": "Qq2g2ygF+bH6NhVLNHBHAS5OHC1YPP2w8txZ8noBsnY=",
"rnn-input-proj.wgsl.jinja": "Ei5DhmJ9OI8S0rAikhRM7s/XCX/vqcAjrSovFQbtYdo=",
"rnn-sequence-mask.wgsl.jinja": "TYvv9fk6+kgxRyUFYZLZLZJssgXUdOTqx5fUHeQVSqE=",
"test.json": "hVVa45HDwETqbYCTah6cyRX+l7THpnHmHv8pBLItils="
}
},
"provenance": { "kernel": { "sha": "6fdf6301e2bbcc2f03bf1eaf493b7ad55ef33afc", "dirty": false } },
"webgpu": {
"manifestSpec": "2.1",
"variants": {
"parallel_input_proj_recurrent_subgroup_gemv_seqlens": ["lstm-recur-subgroup-step.wgsl.jinja", "rnn-input-proj.wgsl.jinja", "rnn-sequence-mask.wgsl.jinja"],
"parallel_tiled_input_proj_recurrent_subgroup_batched": ["lstm-recur-subgroup-batched-step.wgsl.jinja", "matmul-notrans-vec4-tiled-reg.wgsl.jinja"],
"parallel_input_proj_recurrent_subgroup_gemv": ["lstm-recur-subgroup-batched-step.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"parallel_input_proj_recurrent_packed_multiwg": ["datamove-transpose-2d-tiled.wgsl.jinja", "lstm-recur-step.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"parallel_input_proj_recurrent_packed": ["datamove-transpose-2d-tiled.wgsl.jinja", "lstm-timestep-body.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_attr_input_proj": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"forward_layout0": ["lstm-parallel-global.wgsl.jinja"],
"general_attr": ["lstm-general.wgsl.jinja"],
"parallel_batch_gates_vec4": ["lstm-timestep-body.wgsl.jinja"],
"parallel_batch_gates_vec2_recur": ["lstm-timestep-body.wgsl.jinja"],
"parallel_batch_gates_vec4tail": ["lstm-timestep-body.wgsl.jinja"],
"general_seqlens_inith_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja"],
"general_seqlens_inithc_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja"],
"general_seqlens_initc_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja"],
"general_inithc_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja"],
"general_peephole": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_no_state": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_seqlens": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_inith": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_initc": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_inithc": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_seqlens_inith": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_seqlens_initc": ["lstm-general.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_seqlens_inith_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_seqlens_initc_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_peephole_inithc_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"general_seqlens": ["lstm-general.wgsl.jinja"],
"general_seqlens_inith": ["lstm-general.wgsl.jinja"],
"general_seqlens_inithc": ["lstm-general.wgsl.jinja"],
"general_inith": ["lstm-general.wgsl.jinja"],
"general_initc": ["lstm-general.wgsl.jinja"],
"general_inithc": ["lstm-general.wgsl.jinja"],
"general_seqlens_initc": ["lstm-general.wgsl.jinja"],
"parallel_input_proj_vec4": ["lstm-timestep-body.wgsl.jinja", "rnn-input-proj.wgsl.jinja"],
"parallel_input_proj_vec2_recur": ["lstm-timestep-body.wgsl.jinja", "rnn-input-proj.wgsl.jinja"]
}
}
}
|