Hugging Face's logo

Kernels:
webgpu-kernels
/
ai.onnx.LSTM

Kernel card Files
xet
 Community
ai.onnx.LSTM
File size: 4,976 Bytes
bc30b10
 
7e43ef2
bc30b10
 
 
 
 
 
7e43ef2
 
 
 
 
 
 
bc30b10
7e43ef2
 
 
 
 
 
bc30b10
 
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_4e0df38",
  "version": 1,
  "license": "Apache-2.0",
  "backend": { "type": "webgpu" },
  "digest": {
    "algorithm": "sha256",
    "files": {
      "bench.json": "WA/+IHAhAUjlUggZ6V3/WhmcI/PmyNdxrOkTOUvhcj8=",
      "datamove-transpose-2d-tiled.wgsl.jinja": "GNVRxhagDFVQ0DGjsUGD8BVzktDHsELwe0qw4xRHD60=",
      "lstm-general.wgsl.jinja": "dgYWK6Tew4CNvrrTU/Btmzz3iK/4Xi/+QP0OoY5iWPo=",
      "lstm-parallel-global.wgsl.jinja": "Svm4y3Mr/7fQU7THa74x4IObpiSZAA7YBazaw4HmTSA=",
      "lstm-recur-step.wgsl.jinja": "uC2kyXEj+k9N/GIMDXOGJAUpJp0MNTsmz7dlQmAhBTY=",
      "lstm-recur-subgroup-batched-step.wgsl.jinja": "QYlLH1YR+ZgkyKrCL2HtOp+rIsULvoyvWX5MylBdnPA=",
      "lstm-recur-subgroup-step.wgsl.jinja": "HBhdVEJF//yoqimtwOXqSRzlKtXRe0WSUN4HfZ9UPCQ=",
      "lstm-seed-state.wgsl.jinja": "aUVcYv1Sd+ys/whRCgpvjLJP2CvkP5Wk91Iwaze14nE=",
      "lstm-timestep-body.wgsl.jinja": "BhgNPYuCEEZJ13wfMQxampaMBZP3QgFEYyOgPKkKQ/I=",
      "manifest.json": "Ro8odeFMwqpVkX3Zy7d7OGB7psQ3wleaWmUbopwBtrM=",
      "matmul-notrans-vec4-tiled-reg.wgsl.jinja": "smOSOiIUTgO/IFKnCDq+UCf8qNDe44ccntiI6SSXOS8=",
      "rnn-input-proj.wgsl.jinja": "WIYuSwee6iExtELiYwRbaVn55LlRku0BdoGb0JuU7r0=",
      "rnn-sequence-mask.wgsl.jinja": "i2xKEOOTZo4nNR8CPcdWsfKa8QmSSLZPazYsvpmQ58U=",
      "test.json": "nSIO6fS07/Mwjj81Nl0DCKzKDr52Iq+JDo6Cq24Ix+s="
    }
  },
  "provenance": { "kernel": { "sha": "91d990483a174128daf7673f3f37a7c890493ae1", "dirty": false } },
  "webgpu": {
    "manifestSpec": "2.0",
    "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"],
      "general_peephole": ["lstm-general.wgsl.jinja", "rnn-input-proj.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_peephole_split": ["lstm-general.wgsl.jinja", "lstm-seed-state.wgsl.jinja", "rnn-input-proj.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_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_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_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"]
    }
  }
}