@genai-fi/nanogpt 0.19.1 → 0.20.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (347) hide show
  1. package/dist/BaseTokeniser-DSg9zcYq.js +221 -0
  2. package/dist/DatasetBuilder-DgURD85T.js +712 -0
  3. package/dist/Generator.js +2 -11941
  4. package/dist/RealDiv-DBu0FQqT.js +362 -0
  5. package/dist/Reshape-CABOPB9d.js +94 -0
  6. package/dist/Reshape-DqO3r8BC.js +17 -0
  7. package/dist/TeachableLLM.d.ts +5 -5
  8. package/dist/TeachableLLM.js +2 -273
  9. package/dist/Trainer.js +2 -244
  10. package/dist/backend.js +12 -12
  11. package/dist/backend_util-Cg-roD1p.js +399 -0
  12. package/dist/binary_op_util-CrYk9LXL.js +103 -0
  13. package/dist/checks/appendCache.js +54 -21
  14. package/dist/checks/attentionMask.js +55 -36
  15. package/dist/checks/check.js +31 -19
  16. package/dist/checks/gelu.js +45 -17
  17. package/dist/checks/index.js +25 -25
  18. package/dist/checks/matMulGelu.js +83 -27
  19. package/dist/checks/normRMS.js +27 -15
  20. package/dist/checks/normRMSGrad.js +21 -11
  21. package/dist/checks/packUnpack.js +45 -17
  22. package/dist/checks/qkv.js +33 -33
  23. package/dist/checks/rope.js +29 -35
  24. package/dist/checks/weights.d.ts +1 -1
  25. package/dist/checks/weights.js +25 -29
  26. package/dist/chunk-BPntVaq0.js +23 -0
  27. package/dist/complex_util-CkazZsaH.js +60 -0
  28. package/dist/concat_util-CWDZCBlA.js +19 -0
  29. package/dist/data/docx.js +3044 -13
  30. package/dist/data/pdf.js +16 -13
  31. package/dist/data/textLoader.js +607 -112
  32. package/dist/dist-BewPQWjc.js +7572 -0
  33. package/dist/dist-DVmq73nz.js +8775 -0
  34. package/dist/dist-DXwIvKxl.js +896 -0
  35. package/dist/dist-VEU5mfO0.js +7545 -0
  36. package/dist/gelu-Bf1HW1RY.js +27 -0
  37. package/dist/gpgpu_math-DvLcCH6u.js +1612 -0
  38. package/dist/inference/types.js +0 -1
  39. package/dist/kernel_funcs_utils-HiXOOx3f.js +229 -0
  40. package/dist/layers/BaseLayer.d.ts +2 -2
  41. package/dist/layers/BaseLayer.js +75 -73
  42. package/dist/layers/CausalSelfAttention.d.ts +1 -1
  43. package/dist/layers/CausalSelfAttention.js +98 -85
  44. package/dist/layers/LoRA.js +47 -57
  45. package/dist/layers/MLP.d.ts +1 -1
  46. package/dist/layers/MLP.js +33 -43
  47. package/dist/layers/PositionEmbedding.d.ts +1 -1
  48. package/dist/layers/PositionEmbedding.js +26 -30
  49. package/dist/layers/RMSNorm.d.ts +1 -1
  50. package/dist/layers/RMSNorm.js +19 -21
  51. package/dist/layers/RoPECache.js +336 -49
  52. package/dist/layers/TiedEmbedding.d.ts +1 -1
  53. package/dist/layers/TiedEmbedding.js +30 -34
  54. package/dist/layers/TransformerBlock.d.ts +1 -1
  55. package/dist/layers/TransformerBlock.js +50 -39
  56. package/dist/layers/WeightStore.js +68 -75
  57. package/dist/loader/load.js +2 -68
  58. package/dist/loader/loadHF.d.ts +2 -2
  59. package/dist/loader/loadHF.js +2 -22
  60. package/dist/loader/loadTransformers.d.ts +1 -1
  61. package/dist/loader/loadTransformers.js +2 -44
  62. package/dist/loader/loadZipMeta.js +15 -15
  63. package/dist/loader/newZipLoad.js +2 -31
  64. package/dist/loader/oldZipLoad.d.ts +2 -2
  65. package/dist/loader/oldZipLoad.js +2 -80
  66. package/dist/loader/save.d.ts +5 -5
  67. package/dist/loader/save.js +2 -90
  68. package/dist/loader/types.d.ts +9 -8
  69. package/dist/loader/types.js +0 -1
  70. package/dist/main-CPjeMv0G.js +13500 -0
  71. package/dist/main.d.ts +1 -1
  72. package/dist/main.js +16 -109
  73. package/dist/matMul16-BNfZSnNM.js +81 -0
  74. package/dist/matMulGelu-CPTntosE.js +162 -0
  75. package/dist/models/NanoGPTV1.js +2 -99
  76. package/dist/models/NanoGPTV2.js +2 -90
  77. package/dist/models/config.js +34 -47
  78. package/dist/models/factory.d.ts +1 -1
  79. package/dist/models/factory.js +2 -16
  80. package/dist/models/model.d.ts +2 -2
  81. package/dist/models/model.js +2 -134
  82. package/dist/ops/adamAdjust.js +15 -6
  83. package/dist/ops/adamMoments.js +13 -6
  84. package/dist/ops/add16.js +9 -6
  85. package/dist/ops/appendCache.js +22 -19
  86. package/dist/ops/attentionMask.js +12 -6
  87. package/dist/ops/concat16.js +7 -8
  88. package/dist/ops/cpu/adamAdjust.js +15 -17
  89. package/dist/ops/cpu/adamMoments.js +15 -15
  90. package/dist/ops/cpu/appendCache.js +64 -22
  91. package/dist/ops/cpu/attentionMask.js +15 -21
  92. package/dist/ops/cpu/fusedSoftmax.js +21 -28
  93. package/dist/ops/cpu/gatherSub.js +11 -17
  94. package/dist/ops/cpu/gelu.js +34 -38
  95. package/dist/ops/cpu/matMul16.js +13 -14
  96. package/dist/ops/cpu/matMulGelu.js +39 -51
  97. package/dist/ops/cpu/matMulMul.js +19 -22
  98. package/dist/ops/cpu/mulDropout.js +19 -22
  99. package/dist/ops/cpu/normRMS.js +33 -37
  100. package/dist/ops/cpu/qkv.js +72 -40
  101. package/dist/ops/cpu/rope.js +79 -36
  102. package/dist/ops/cpu/scatterSub.js +11 -22
  103. package/dist/ops/dot16.js +28 -41
  104. package/dist/ops/dropout.js +10 -13
  105. package/dist/ops/dropout16.js +20 -23
  106. package/dist/ops/gatherSub.js +10 -6
  107. package/dist/ops/gelu.js +2 -8
  108. package/dist/ops/globalNorm.js +18 -12
  109. package/dist/ops/grads/add16.js +27 -26
  110. package/dist/ops/grads/attentionMask.js +26 -21
  111. package/dist/ops/grads/dropout16.js +0 -1
  112. package/dist/ops/grads/gelu.js +2 -5
  113. package/dist/ops/grads/matMul16.js +2 -9
  114. package/dist/ops/grads/matMulGelu.js +21 -16
  115. package/dist/ops/grads/mul16.js +0 -3
  116. package/dist/ops/grads/normRMS.js +34 -30
  117. package/dist/ops/grads/pack16.js +2 -6
  118. package/dist/ops/grads/qkv.js +44 -32
  119. package/dist/ops/grads/rope.js +2 -5
  120. package/dist/ops/grads/softmax16.js +21 -23
  121. package/dist/ops/grads/unpack16.js +2 -5
  122. package/dist/ops/grads/utils.js +9 -11
  123. package/dist/ops/matMul16.js +2 -13
  124. package/dist/ops/matMulGelu.js +16 -10
  125. package/dist/ops/matMulMul.js +13 -6
  126. package/dist/ops/mul16.js +42 -38
  127. package/dist/ops/mulDrop.js +12 -6
  128. package/dist/ops/normRMS.js +18 -15
  129. package/dist/ops/pack16.js +2 -5
  130. package/dist/ops/qkv.js +12 -6
  131. package/dist/ops/reshape16.js +31 -39
  132. package/dist/ops/rope.d.ts +1 -1
  133. package/dist/ops/rope.js +2 -7
  134. package/dist/ops/scatterSub.js +10 -6
  135. package/dist/ops/slice16.js +9 -7
  136. package/dist/ops/softmax16.js +7 -7
  137. package/dist/ops/sub16.js +9 -6
  138. package/dist/ops/sum16.js +12 -12
  139. package/dist/ops/transpose16.js +29 -37
  140. package/dist/ops/unpack16.js +2 -6
  141. package/dist/ops/webgl/adamAdjust.js +62 -29
  142. package/dist/ops/webgl/adamMoments.js +30 -26
  143. package/dist/ops/webgl/appendCache.js +30 -21
  144. package/dist/ops/webgl/attentionMask.js +43 -24
  145. package/dist/ops/webgl/dropout16.js +11 -10
  146. package/dist/ops/webgl/fusedSoftmax.js +69 -79
  147. package/dist/ops/webgl/gatherSub.js +27 -26
  148. package/dist/ops/webgl/gelu.js +32 -34
  149. package/dist/ops/webgl/log.js +14 -23
  150. package/dist/ops/webgl/matMul16.js +36 -44
  151. package/dist/ops/webgl/matMulGelu.js +2 -9
  152. package/dist/ops/webgl/matMulMul.js +23 -27
  153. package/dist/ops/webgl/mulDropout.js +31 -40
  154. package/dist/ops/webgl/normRMS.js +92 -71
  155. package/dist/ops/webgl/qkv.js +35 -27
  156. package/dist/ops/webgl/rope.js +37 -21
  157. package/dist/ops/webgl/scatterSub.js +27 -26
  158. package/dist/ops/webgpu/adamAdjust.js +59 -39
  159. package/dist/ops/webgpu/adamMoments.js +62 -46
  160. package/dist/ops/webgpu/add16.js +13 -12
  161. package/dist/ops/webgpu/appendCache.js +79 -54
  162. package/dist/ops/webgpu/attentionMask.js +41 -25
  163. package/dist/ops/webgpu/attentionMask32_program.js +34 -26
  164. package/dist/ops/webgpu/clipScale.js +44 -57
  165. package/dist/ops/webgpu/concat16.js +96 -111
  166. package/dist/ops/webgpu/dropout16.js +40 -32
  167. package/dist/ops/webgpu/gatherSub.js +43 -30
  168. package/dist/ops/webgpu/gelu.js +88 -82
  169. package/dist/ops/webgpu/index.js +16 -16
  170. package/dist/ops/webgpu/matMul16.js +69 -64
  171. package/dist/ops/webgpu/matMul16_program.js +152 -192
  172. package/dist/ops/webgpu/mul16.js +13 -12
  173. package/dist/ops/webgpu/norm2.js +45 -75
  174. package/dist/ops/webgpu/normRMS.js +25 -33
  175. package/dist/ops/webgpu/normRMS16_program.js +21 -18
  176. package/dist/ops/webgpu/normRMS32_program.js +21 -18
  177. package/dist/ops/webgpu/normRMSGrad.js +125 -184
  178. package/dist/ops/webgpu/pack16.js +20 -17
  179. package/dist/ops/webgpu/pack16_program.js +48 -47
  180. package/dist/ops/webgpu/qkv.js +63 -23
  181. package/dist/ops/webgpu/rope.js +85 -57
  182. package/dist/ops/webgpu/scatterSub.js +43 -30
  183. package/dist/ops/webgpu/slice16.js +66 -61
  184. package/dist/ops/webgpu/softmax16.js +17 -20
  185. package/dist/ops/webgpu/softmax16_program.js +34 -18
  186. package/dist/ops/webgpu/softmax16_subgroup_program.js +40 -45
  187. package/dist/ops/webgpu/softmax16grad.js +30 -36
  188. package/dist/ops/webgpu/sub16.js +13 -12
  189. package/dist/ops/webgpu/sum16.js +28 -37
  190. package/dist/ops/webgpu/transpose16.js +36 -33
  191. package/dist/ops/webgpu/transpose16_program.js +40 -39
  192. package/dist/ops/webgpu/transpose16_shared_program.js +53 -44
  193. package/dist/ops/webgpu/unpack16.js +49 -37
  194. package/dist/ops/webgpu/utils/binary_op.js +70 -68
  195. package/dist/ops/webgpu/utils/deviceInfo.d.ts +1 -1
  196. package/dist/ops/webgpu/utils/deviceInfo.js +10 -10
  197. package/dist/ops/webgpu/utils/reductions.js +136 -148
  198. package/dist/pack16-Ck-spx_F.js +39 -0
  199. package/dist/patches/webgpu_backend.d.ts +2 -2
  200. package/dist/patches/webgpu_backend.js +42 -55
  201. package/dist/patches/webgpu_base.js +21 -33
  202. package/dist/patches/webgpu_program.js +213 -320
  203. package/dist/pdf-UoDqCYzz.js +16726 -0
  204. package/dist/picomatch-3tUnMMbd.js +1063 -0
  205. package/dist/rope-CbeGlsV8.js +25 -0
  206. package/dist/selu_util-zkAx5doH.js +24 -0
  207. package/dist/shared-D1coEFea.js +1314 -0
  208. package/dist/shared-DOgWaqvL.js +5 -0
  209. package/dist/slice_util-Dgb3ANWI.js +208 -0
  210. package/dist/tfjs_backend-BjuQ5FqB.js +614 -0
  211. package/dist/tokeniser/BaseTokeniser.js +2 -124
  212. package/dist/tokeniser/CharTokeniser.js +91 -106
  213. package/dist/tokeniser/bpe.js +163 -166
  214. package/dist/tokeniser/messages.js +0 -1
  215. package/dist/tokeniser/type.js +0 -1
  216. package/dist/training/AdamW.js +127 -137
  217. package/dist/training/BasicTrainer.d.ts +1 -1
  218. package/dist/training/BasicTrainer.js +264 -264
  219. package/dist/training/DatasetBuilder.js +2 -86
  220. package/dist/training/Evaluator.d.ts +1 -1
  221. package/dist/training/Evaluator.js +47 -38
  222. package/dist/training/LRScheduler.js +37 -33
  223. package/dist/training/PreTrainer.d.ts +2 -2
  224. package/dist/training/PreTrainer.js +21 -19
  225. package/dist/training/SFTTrainer.d.ts +2 -2
  226. package/dist/training/SFTTrainer.js +23 -21
  227. package/dist/training/loss.js +17 -22
  228. package/dist/training/orthoGrad.js +9 -9
  229. package/dist/training/sparseCrossEntropy.js +45 -67
  230. package/dist/training/tasks/ConversationTask.d.ts +1 -1
  231. package/dist/training/tasks/ConversationTask.js +36 -38
  232. package/dist/training/tasks/PretrainingTask.d.ts +1 -1
  233. package/dist/training/tasks/PretrainingTask.js +41 -46
  234. package/dist/training/tasks/StartSentenceTask.d.ts +1 -1
  235. package/dist/training/tasks/StartSentenceTask.js +44 -48
  236. package/dist/training/tasks/Task.d.ts +1 -1
  237. package/dist/training/tasks/Task.js +53 -66
  238. package/dist/training/tasks/splitter.js +17 -20
  239. package/dist/training/types.d.ts +2 -2
  240. package/dist/training/types.js +0 -1
  241. package/dist/training/validation.d.ts +1 -1
  242. package/dist/training/validation.js +2 -84
  243. package/dist/utilities/arrayClose.js +15 -19
  244. package/dist/utilities/datasetID.js +17 -20
  245. package/dist/utilities/dummy.d.ts +1 -1
  246. package/dist/utilities/dummy.js +33 -40
  247. package/dist/utilities/multinomialCPU.js +8 -12
  248. package/dist/utilities/naming.js +0 -1
  249. package/dist/utilities/packed.js +10 -12
  250. package/dist/utilities/parameters.d.ts +1 -1
  251. package/dist/utilities/parameters.js +32 -51
  252. package/dist/utilities/performance.js +15 -15
  253. package/dist/utilities/profile.js +32 -37
  254. package/dist/utilities/safetensors.js +49 -79
  255. package/dist/utilities/sentences.d.ts +1 -1
  256. package/dist/utilities/sentences.js +29 -38
  257. package/dist/utilities/tokenParse.js +16 -20
  258. package/dist/utilities/topP.js +11 -12
  259. package/dist/utilities/waitForModel.d.ts +1 -1
  260. package/dist/utilities/waitForModel.js +11 -11
  261. package/dist/utilities/weights.js +37 -42
  262. package/dist/utilities/yielder.js +6 -6
  263. package/dist/webgpu-Dt7BMzWz.js +525 -0
  264. package/dist/webgpu_program-WOyIVMlZ.js +392 -0
  265. package/dist/webgpu_util-B_F3SShA.js +106 -0
  266. package/package.json +9 -10
  267. package/dist/RealDiv-CGwv0liw.js +0 -365
  268. package/dist/Reshape-BW__R4mZ.js +0 -79
  269. package/dist/Reshape-CPBkTIH2.js +0 -14
  270. package/dist/_commonjsHelpers-ByX85dGu.js +0 -33
  271. package/dist/axis_util-GTVlo58H.js +0 -55
  272. package/dist/backend_util-GaFarB78.js +0 -425
  273. package/dist/backend_webgpu-BqASlsbV.js +0 -545
  274. package/dist/binary_op_util-pKXltfxI.js +0 -192
  275. package/dist/broadcast_to-eS93CCN_.js +0 -28
  276. package/dist/clip_by_value-DDA7rrcT.js +0 -12
  277. package/dist/complex-DI35Q-gW.js +0 -11
  278. package/dist/complex_util-Yc1A_gV1.js +0 -55
  279. package/dist/concat-CAQpCret.js +0 -17
  280. package/dist/concat_util-D18dJ4fD.js +0 -22
  281. package/dist/data/parquet.d.ts +0 -2
  282. package/dist/data/parquet.js +0 -17
  283. package/dist/dataset-CGGp1z9P.js +0 -1124
  284. package/dist/dropout_util--NxWuYg2.js +0 -27
  285. package/dist/expand_dims-Bkd1YD5x.js +0 -11
  286. package/dist/exports_initializers-CYzKLjN7.js +0 -7
  287. package/dist/floor-BQtb-Azg.js +0 -9
  288. package/dist/gather-qIqEqaGn.js +0 -9
  289. package/dist/gelu-B220X1Go.js +0 -26
  290. package/dist/gpgpu_math-BwvV12df.js +0 -2022
  291. package/dist/index-CUXkjxiT.js +0 -3516
  292. package/dist/index-CieiGp4Y.js +0 -349
  293. package/dist/index-CjOWnMXP.js +0 -7308
  294. package/dist/index-Cp39cXWe.js +0 -1016
  295. package/dist/index-D5v913EJ.js +0 -4
  296. package/dist/index-DmeWGGmS.js +0 -1074
  297. package/dist/index-DvYrXKkX.js +0 -113
  298. package/dist/index-Ksja3su6.js +0 -151
  299. package/dist/index-xuotMAFm.js +0 -118
  300. package/dist/jszip.min-BZhlzntC.js +0 -2313
  301. package/dist/kernel_funcs_utils-pq0CK9co.js +0 -306
  302. package/dist/matMul16-BcVC_E62.js +0 -80
  303. package/dist/matMulGelu-JNLZqKQp.js +0 -163
  304. package/dist/mat_mul-DhG0Newp.js +0 -11
  305. package/dist/mod-CSdCpRjf.js +0 -11
  306. package/dist/non_max_suppression_impl-B2W7YjZB.js +0 -102
  307. package/dist/not_equal-hurPF26l.js +0 -64
  308. package/dist/ones-BytntneX.js +0 -14
  309. package/dist/ops-CsXeTq1P.js +0 -476
  310. package/dist/pack16-bqltoUlR.js +0 -39
  311. package/dist/papaparse.min-C0cScC2i.js +0 -418
  312. package/dist/parquet-Bqjmp2vo.js +0 -44231
  313. package/dist/pdf-NIhmP3sq.js +0 -19477
  314. package/dist/rand_util-CZ7yLoUm.js +0 -50
  315. package/dist/random_normal-IBRrha8a.js +0 -14
  316. package/dist/random_width-DN5ZtQkM.js +0 -9796
  317. package/dist/range-C-CjF-LI.js +0 -10
  318. package/dist/relu-J_X6MUzx.js +0 -9
  319. package/dist/reshape-BDOuCSNW.js +0 -9
  320. package/dist/resize_nearest_neighbor-BojqlfRe.js +0 -150
  321. package/dist/rope-DcrZM_e6.js +0 -24
  322. package/dist/scatter_nd_util-ByNJaL6I.js +0 -46
  323. package/dist/segment_util-Dasb2Zaf.js +0 -43
  324. package/dist/selu_util-BLhIqRkw.js +0 -44
  325. package/dist/shared-3agzAqQ_.js +0 -53
  326. package/dist/shared-CagdqkLh.js +0 -2143
  327. package/dist/slice-BzS11Qh0.js +0 -12
  328. package/dist/slice_util-CC35pLmT.js +0 -153
  329. package/dist/softmax-D4q1LJN7.js +0 -12
  330. package/dist/split-C2Sj255c.js +0 -9
  331. package/dist/squeeze-ho4wLUek.js +0 -10
  332. package/dist/stack-DudVrtmG.js +0 -11
  333. package/dist/step-BTxPtq1r.js +0 -261
  334. package/dist/sum-BpiwSWvg.js +0 -11
  335. package/dist/tensor-BWFldCso.js +0 -8
  336. package/dist/tensor1d-LMGMIUlr.js +0 -11
  337. package/dist/tensor2d-BnXMKScO.js +0 -14
  338. package/dist/tensor4d-C6UCG_u8.js +0 -14
  339. package/dist/tfjs_backend-BGnG-ppu.js +0 -654
  340. package/dist/tile-CFy-xTO6.js +0 -11
  341. package/dist/transpose-9kRxIXWR.js +0 -36
  342. package/dist/unsorted_segment_sum-DJvk5xnh.js +0 -277
  343. package/dist/variable-Ck482e3n.js +0 -7
  344. package/dist/webgpu_program-B4HmApL1.js +0 -525
  345. package/dist/webgpu_util-DYlGSwOJ.js +0 -64
  346. package/dist/zeros-DvZpK8s6.js +0 -13
  347. package/dist/zeros_like-CWjDdwr-.js +0 -721
@@ -1,23 +1,27 @@
1
- import { c as m } from "../../index-CUXkjxiT.js";
2
- class i {
3
- variableNames = ["moments", "gradient"];
4
- outputShape;
5
- userCode;
6
- customUniforms = [
7
- { name: "beta1", type: "float" },
8
- { name: "beta2", type: "float" }
9
- ];
10
- constructor(e) {
11
- this.outputShape = e;
12
- const t = e.length, n = t === 1 ? "int" : `ivec${Math.min(t, 4)}`, a = t === 1 ? "coords" : `coords[${t - 1}]`, r = t === 1 ? "coords" : e.slice(0, -1).map((s, c) => `coords[${c}]`).join(", ");
13
- this.userCode = `
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ //#region lib/ops/webgl/adamMoments.ts
3
+ var t = class {
4
+ variableNames = ["moments", "gradient"];
5
+ outputShape;
6
+ userCode;
7
+ customUniforms = [{
8
+ name: "beta1",
9
+ type: "float"
10
+ }, {
11
+ name: "beta2",
12
+ type: "float"
13
+ }];
14
+ constructor(e) {
15
+ this.outputShape = e;
16
+ let t = e.length, n = t === 1 ? "int" : `ivec${Math.min(t, 4)}`, r = t === 1 ? "coords" : `coords[${t - 1}]`, i = t === 1 ? "coords" : e.slice(0, -1).map((e, t) => `coords[${t}]`).join(", ");
17
+ this.userCode = `
14
18
  void main() {
15
19
  float m = getMomentsAtOutCoords();
16
20
  ${n} coords = getOutputCoords();
17
21
 
18
22
  // Add gradient clipping here
19
- float g = clamp(getGradient(${r}), -1.0, 1.0);
20
- int which = ${a};
23
+ float g = clamp(getGradient(${i}), -1.0, 1.0);
24
+ int which = ${r};
21
25
 
22
26
  float beta = which == 0 ? beta1 : beta2;
23
27
  float gg = which == 0 ? g : g * g;
@@ -26,15 +30,15 @@ class i {
26
30
  setOutput(newM);
27
31
  }
28
32
  `;
29
- }
30
- }
31
- function d(o) {
32
- const { moments: e, gradient: t } = o.inputs, { beta1: n, beta2: a } = o.attrs, r = o.backend, s = new i(e.shape);
33
- return r.runWebGLProgram(s, [e, t], "float32", [[n], [a]]);
34
- }
35
- const g = {
36
- kernelName: "AdamMoments",
37
- backendName: "webgl",
38
- kernelFunc: d
33
+ }
39
34
  };
40
- m(g);
35
+ function n(e) {
36
+ let { moments: n, gradient: r } = e.inputs, { beta1: i, beta2: a } = e.attrs, o = e.backend, s = new t(n.shape);
37
+ return o.runWebGLProgram(s, [n, r], "float32", [[i], [a]]);
38
+ }
39
+ e({
40
+ kernelName: "AdamMoments",
41
+ backendName: "webgl",
42
+ kernelFunc: n
43
+ });
44
+ //#endregion
@@ -1,12 +1,21 @@
1
- import { c as p } from "../../index-CUXkjxiT.js";
2
- class m {
3
- variableNames = ["cache", "item"];
4
- outputShape;
5
- userCode;
6
- customUniforms = [{ name: "cacheT", type: "int" }];
7
- constructor(t, a, n, o, c) {
8
- const s = Math.min(n + 1, c);
9
- this.outputShape = [t, a, s, o], this.userCode = `
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ //#region lib/ops/webgl/appendCache.ts
3
+ var t = class {
4
+ variableNames = ["cache", "item"];
5
+ outputShape;
6
+ userCode;
7
+ customUniforms = [{
8
+ name: "cacheT",
9
+ type: "int"
10
+ }];
11
+ constructor(e, t, n, r, i) {
12
+ let a = Math.min(n + 1, i);
13
+ this.outputShape = [
14
+ e,
15
+ t,
16
+ a,
17
+ r
18
+ ], this.userCode = `
10
19
  void main() {
11
20
  ivec4 coords = getOutputCoords(); // [b, h, t, d]
12
21
  int b = coords.x;
@@ -15,7 +24,7 @@ class m {
15
24
  int d = coords.w;
16
25
 
17
26
  int itemT = 1;
18
- int maxSize = ${c};
27
+ int maxSize = ${i};
19
28
  int totalT = cacheT + itemT;
20
29
  int start = totalT >= maxSize ? 1 : 0;
21
30
 
@@ -30,15 +39,15 @@ class m {
30
39
  setOutput(val);
31
40
  }
32
41
  `;
33
- }
34
- }
35
- function d(e) {
36
- const { cache: t, item: a } = e.inputs, { maxSize: n, pastLen: o } = e.attrs, c = e.backend, s = t.shape[0], r = t.shape[2], i = t.shape[1], h = new m(s, i, r, a.shape[3], n);
37
- return c.runWebGLProgram(h, [t, a], "float32", [[o]]);
38
- }
39
- const l = {
40
- kernelName: "AppendCache",
41
- backendName: "webgl",
42
- kernelFunc: d
42
+ }
43
43
  };
44
- p(l);
44
+ function n(e) {
45
+ let { cache: n, item: r } = e.inputs, { maxSize: i, pastLen: a } = e.attrs, o = e.backend, s = n.shape[0], c = n.shape[2], l = n.shape[1], u = new t(s, l, c, r.shape[3], i);
46
+ return o.runWebGLProgram(u, [n, r], "float32", [[a]]);
47
+ }
48
+ e({
49
+ kernelName: "AppendCache",
50
+ backendName: "webgl",
51
+ kernelFunc: n
52
+ });
53
+ //#endregion
@@ -1,15 +1,30 @@
1
- import { c as m } from "../../index-CUXkjxiT.js";
2
- class h {
3
- variableNames = ["q", "k"];
4
- outputShape;
5
- userCode;
6
- customUniforms = [
7
- { name: "divisor", type: "float" },
8
- { name: "pastLen", type: "int" },
9
- { name: "inf", type: "float" }
10
- ];
11
- constructor(t, e, s, n, a) {
12
- this.outputShape = [t, e, s, n], this.userCode = `
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ //#region lib/ops/webgl/attentionMask.ts
3
+ var t = class {
4
+ variableNames = ["q", "k"];
5
+ outputShape;
6
+ userCode;
7
+ customUniforms = [
8
+ {
9
+ name: "divisor",
10
+ type: "float"
11
+ },
12
+ {
13
+ name: "pastLen",
14
+ type: "int"
15
+ },
16
+ {
17
+ name: "inf",
18
+ type: "float"
19
+ }
20
+ ];
21
+ constructor(e, t, n, r, i) {
22
+ this.outputShape = [
23
+ e,
24
+ t,
25
+ n,
26
+ r
27
+ ], this.userCode = `
13
28
  void main() {
14
29
  ivec4 coords = getOutputCoords(); // [batch, nh, t1, t2]
15
30
  int b = coords.x;
@@ -18,7 +33,7 @@ class h {
18
33
  int t2 = coords.w;
19
34
 
20
35
  float sum = 0.0;
21
- for (int i = 0; i < ${a}; ++i) {
36
+ for (int i = 0; i < ${i}; ++i) {
22
37
  float qv = getQ(b, h, t1, i);
23
38
  float kv = getK(b, h, t2, i);
24
39
  sum += qv * kv;
@@ -31,15 +46,19 @@ class h {
31
46
  setOutput((t2 > t1 + pastLen) ? inf : scaled);
32
47
  }
33
48
  `;
34
- }
35
- }
36
- function l(o) {
37
- const { q: t, k: e } = o.inputs, { divisor: s, pastLen: n } = o.attrs, a = o.backend, i = t.shape[0], r = t.shape[2], c = e.shape[2], u = t.shape[1], p = t.shape[3], d = new h(i, u, r, c, p);
38
- return a.runWebGLProgram(d, [t, e], "float32", [[s], [n], [Number.NEGATIVE_INFINITY]]);
39
- }
40
- const f = {
41
- kernelName: "AttentionMask",
42
- backendName: "webgl",
43
- kernelFunc: l
49
+ }
44
50
  };
45
- m(f);
51
+ function n(e) {
52
+ let { q: n, k: r } = e.inputs, { divisor: i, pastLen: a } = e.attrs, o = e.backend, s = n.shape[0], c = n.shape[2], l = r.shape[2], u = n.shape[1], d = n.shape[3], f = new t(s, u, c, l, d);
53
+ return o.runWebGLProgram(f, [n, r], "float32", [
54
+ [i],
55
+ [a],
56
+ [-Infinity]
57
+ ]);
58
+ }
59
+ e({
60
+ kernelName: "AttentionMask",
61
+ backendName: "webgl",
62
+ kernelFunc: n
63
+ });
64
+ //#endregion
@@ -1,11 +1,12 @@
1
- import { c as n } from "../../index-CUXkjxiT.js";
2
- function o(e) {
3
- const { x: r } = e.inputs;
4
- return r;
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ //#region lib/ops/webgl/dropout16.ts
3
+ function t(e) {
4
+ let { x: t } = e.inputs;
5
+ return t;
5
6
  }
6
- const t = {
7
- kernelName: "Dropout16",
8
- backendName: "webgl",
9
- kernelFunc: o
10
- };
11
- n(t);
7
+ e({
8
+ kernelName: "Dropout16",
9
+ backendName: "webgl",
10
+ kernelFunc: t
11
+ });
12
+ //#endregion
@@ -1,80 +1,70 @@
1
- import { m as b, s as I, r as k } from "../../RealDiv-CGwv0liw.js";
2
- import { r as v } from "../../Reshape-BW__R4mZ.js";
3
- import { c as w, ah as P } from "../../index-CUXkjxiT.js";
4
- import { e as S } from "../../axis_util-GTVlo58H.js";
5
- class T {
6
- variableNames = ["logits", "maxLogits"];
7
- outputShape;
8
- userCode;
9
- constructor(t) {
10
- this.outputShape = t, this.userCode = `
11
- void main() {
12
- ivec4 coords = getOutputCoords(); // [batch, nh, t1, t2]
13
- int b = coords.x;
14
- int h = coords.y;
15
- int t1 = coords.z;
16
- int t2 = coords.w;
17
- float x = getLogitsAtOutCoords();
18
- float maxLogit = getMaxLogits(b, h, t1);
19
- setOutput(exp(x - maxLogit));
20
- }
21
- `;
22
- }
23
- }
24
- class C {
25
- variableNames = ["exp", "sum"];
26
- outputShape;
27
- userCode;
28
- customUniforms = [
29
- { name: "dropoutRate", type: "float" },
30
- { name: "seed", type: "float" }
31
- ];
32
- constructor(t) {
33
- this.outputShape = t, this.userCode = `
34
- float random(ivec4 coords) {
35
- float x = float(coords.x * 4096 + coords.y * 256 + coords.z * 16 + coords.w);
36
- return fract(sin(seed + x) * 43758.5453123);
37
- }
38
- void main() {
39
- ivec4 coords = getOutputCoords();
40
- float numerator = getExp(coords.x, coords.y, coords.z, coords.w);
41
- float denominator = getSum(coords.x, coords.y, coords.z, coords.w);
42
- float val = numerator / denominator;
43
- float keepProb = 1.0 - dropoutRate;
44
- float rand = random(coords);
45
- float mask = step(rand, keepProb);
46
- setOutput(val * mask / keepProb);
47
- }
48
- `;
49
- }
50
- }
51
- function L(r) {
52
- const { inputs: t, attrs: m } = r, { logits: e } = t, { dim: c, dropoutRate: n, seed: u } = m, o = r.backend;
53
- if (!e)
54
- throw new Error("Error in softmax: input logits is null");
55
- const i = P([c], e.shape), d = b({
56
- inputs: { x: e },
57
- backend: o,
58
- attrs: { reductionIndices: i, keepDims: !1 }
59
- }), f = S(d.shape, i), l = new T(e.shape), s = o.runWebGLProgram(l, [e, d], "float32");
60
- o.disposeIntermediateTensorInfo(d);
61
- const p = I({ inputs: { x: s }, backend: o, attrs: { axis: i, keepDims: !1 } }), a = v({ inputs: { x: p }, backend: o, attrs: { shape: f } });
62
- if (n !== void 0 && n > 0) {
63
- const h = new C(e.shape), g = o.runWebGLProgram(h, [s, a], "float32", [
64
- [n],
65
- [u ?? Math.random() * 1e4]
66
- ]);
67
- return o.disposeIntermediateTensorInfo(s), o.disposeIntermediateTensorInfo(p), o.disposeIntermediateTensorInfo(a), g;
68
- }
69
- const x = k({ inputs: { a: s, b: a }, backend: o });
70
- return o.disposeIntermediateTensorInfo(s), o.disposeIntermediateTensorInfo(p), o.disposeIntermediateTensorInfo(a), x;
71
- }
72
- const E = {
73
- kernelName: "FusedSoftmax",
74
- backendName: "webgl",
75
- kernelFunc: L
76
- };
77
- w(E);
78
- export {
79
- L as softmax
1
+ import { Ii as e, Tn as t, nc as n } from "../../dist-BewPQWjc.js";
2
+ import { t as r } from "../../Reshape-CABOPB9d.js";
3
+ import { a as i, r as a, t as o } from "../../RealDiv-DBu0FQqT.js";
4
+ //#region lib/ops/webgl/fusedSoftmax.ts
5
+ var s = class {
6
+ variableNames = ["logits", "maxLogits"];
7
+ outputShape;
8
+ userCode;
9
+ constructor(e) {
10
+ this.outputShape = e, this.userCode = "\n void main() {\n ivec4 coords = getOutputCoords(); // [batch, nh, t1, t2]\n int b = coords.x;\n int h = coords.y;\n int t1 = coords.z;\n int t2 = coords.w;\n float x = getLogitsAtOutCoords();\n float maxLogit = getMaxLogits(b, h, t1);\n setOutput(exp(x - maxLogit));\n }\n ";
11
+ }
12
+ }, c = class {
13
+ variableNames = ["exp", "sum"];
14
+ outputShape;
15
+ userCode;
16
+ customUniforms = [{
17
+ name: "dropoutRate",
18
+ type: "float"
19
+ }, {
20
+ name: "seed",
21
+ type: "float"
22
+ }];
23
+ constructor(e) {
24
+ this.outputShape = e, this.userCode = "\n float random(ivec4 coords) {\n float x = float(coords.x * 4096 + coords.y * 256 + coords.z * 16 + coords.w);\n return fract(sin(seed + x) * 43758.5453123);\n }\n void main() {\n ivec4 coords = getOutputCoords();\n float numerator = getExp(coords.x, coords.y, coords.z, coords.w);\n float denominator = getSum(coords.x, coords.y, coords.z, coords.w);\n float val = numerator / denominator;\n float keepProb = 1.0 - dropoutRate;\n float rand = random(coords);\n float mask = step(rand, keepProb);\n setOutput(val * mask / keepProb);\n }\n ";
25
+ }
80
26
  };
27
+ function l(e) {
28
+ let { inputs: l, attrs: u } = e, { logits: d } = l, { dim: f, dropoutRate: p, seed: m } = u, h = e.backend;
29
+ if (!d) throw Error("Error in softmax: input logits is null");
30
+ let g = n([f], d.shape), _ = a({
31
+ inputs: { x: d },
32
+ backend: h,
33
+ attrs: {
34
+ reductionIndices: g,
35
+ keepDims: !1
36
+ }
37
+ }), v = t(_.shape, g), y = new s(d.shape), b = h.runWebGLProgram(y, [d, _], "float32");
38
+ h.disposeIntermediateTensorInfo(_);
39
+ let x = i({
40
+ inputs: { x: b },
41
+ backend: h,
42
+ attrs: {
43
+ axis: g,
44
+ keepDims: !1
45
+ }
46
+ }), S = r({
47
+ inputs: { x },
48
+ backend: h,
49
+ attrs: { shape: v }
50
+ });
51
+ if (p !== void 0 && p > 0) {
52
+ let e = new c(d.shape), t = h.runWebGLProgram(e, [b, S], "float32", [[p], [m ?? Math.random() * 1e4]]);
53
+ return h.disposeIntermediateTensorInfo(b), h.disposeIntermediateTensorInfo(x), h.disposeIntermediateTensorInfo(S), t;
54
+ }
55
+ let C = o({
56
+ inputs: {
57
+ a: b,
58
+ b: S
59
+ },
60
+ backend: h
61
+ });
62
+ return h.disposeIntermediateTensorInfo(b), h.disposeIntermediateTensorInfo(x), h.disposeIntermediateTensorInfo(S), C;
63
+ }
64
+ e({
65
+ kernelName: "FusedSoftmax",
66
+ backendName: "webgl",
67
+ kernelFunc: l
68
+ });
69
+ //#endregion
70
+ export { l as softmax };
@@ -1,27 +1,28 @@
1
- import { c as l } from "../../index-CUXkjxiT.js";
2
- class u {
3
- variableNames = ["labels", "logits", "values"];
4
- outputShape;
5
- userCode;
6
- constructor(e) {
7
- this.outputShape = [e], this.userCode = `
8
- void main() {
9
- int coords = getOutputCoords();
10
- int index = int(getLabelsAtOutCoords());
11
- float val = getValuesAtOutCoords();
12
- float logit = getLogits(coords, index);
13
- setOutput(val - logit);
14
- }
15
- `;
16
- }
17
- }
18
- function i(t) {
19
- const { logits: e, labels: o, values: s } = t.inputs, a = t.backend, r = o.shape[0], n = new u(r);
20
- return a.runWebGLProgram(n, [o, e, s], "float32");
21
- }
22
- const c = {
23
- kernelName: "EfficientGatherSub",
24
- backendName: "webgl",
25
- kernelFunc: i
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ //#region lib/ops/webgl/gatherSub.ts
3
+ var t = class {
4
+ variableNames = [
5
+ "labels",
6
+ "logits",
7
+ "values"
8
+ ];
9
+ outputShape;
10
+ userCode;
11
+ constructor(e) {
12
+ this.outputShape = [e], this.userCode = "\n void main() {\n int coords = getOutputCoords();\n int index = int(getLabelsAtOutCoords());\n float val = getValuesAtOutCoords();\n float logit = getLogits(coords, index);\n setOutput(val - logit);\n }\n ";
13
+ }
26
14
  };
27
- l(c);
15
+ function n(e) {
16
+ let { logits: n, labels: r, values: i } = e.inputs, a = e.backend, o = r.shape[0], s = new t(o);
17
+ return a.runWebGLProgram(s, [
18
+ r,
19
+ n,
20
+ i
21
+ ], "float32");
22
+ }
23
+ e({
24
+ kernelName: "EfficientGatherSub",
25
+ backendName: "webgl",
26
+ kernelFunc: n
27
+ });
28
+ //#endregion
@@ -1,50 +1,48 @@
1
- import { c as a } from "../../index-CUXkjxiT.js";
2
- import { u as s, C as i } from "../../kernel_funcs_utils-pq0CK9co.js";
3
- const t = 0.7978845608028654, r = 0.044715, c = i + `
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ import { _ as t, i as n } from "../../kernel_funcs_utils-HiXOOx3f.js";
3
+ //#region lib/ops/webgl/gelu.ts
4
+ var r = .7978845608028654, i = .044715, a = n({ opSnippet: t + `
4
5
  float x3 = x * x * x;
5
- float inner = x + ${r} * x3;
6
- inner = ${t} * inner;
6
+ float inner = x + ${i} * x3;
7
+ inner = ${r} * inner;
7
8
  inner = abs(inner) > 15.0 ? sign(inner) : tanh(inner);
8
9
  inner = 0.5 * (1.0 + inner);
9
10
  inner = x * inner;
10
11
  return inner;
11
- `, d = s({ opSnippet: c }), x = {
12
- kernelName: "Gelu",
13
- backendName: "webgl",
14
- kernelFunc: d
15
- };
16
- a(x);
17
- class f {
18
- // Inputs: dy, x
19
- variableNames = ["dy", "x"];
20
- outputShape;
21
- userCode;
22
- constructor(n) {
23
- this.outputShape = n, this.userCode = `
12
+ ` });
13
+ e({
14
+ kernelName: "Gelu",
15
+ backendName: "webgl",
16
+ kernelFunc: a
17
+ });
18
+ var o = class {
19
+ variableNames = ["dy", "x"];
20
+ outputShape;
21
+ userCode;
22
+ constructor(e) {
23
+ this.outputShape = e, this.userCode = `
24
24
  void main() {
25
25
  float dy = getDyAtOutCoords();
26
26
  float x = getXAtOutCoords();
27
27
  float x2 = x * x;
28
28
  float x3 = x2 * x;
29
- float u = ${t} * (x + ${r} * x3);
29
+ float u = ${r} * (x + ${i} * x3);
30
30
  float t = abs(u) > 15.0 ? sign(u) : tanh(u);
31
31
  float sech2 = 1.0 - t * t;
32
- float du_dx = ${t} * (1.0 + 3.0 * ${r} * x2);
32
+ float du_dx = ${r} * (1.0 + 3.0 * ${i} * x2);
33
33
  float dgelu = 0.5 * (1.0 + t) + 0.5 * x * sech2 * du_dx;
34
34
  setOutput(dy * dgelu);
35
35
  }`;
36
- }
37
- }
38
- function g(e) {
39
- const { dy: n, x: o } = e.inputs, u = e.backend, l = new f(o.shape);
40
- return u.runWebGLProgram(l, [n, o], "float32");
41
- }
42
- const p = {
43
- kernelName: "GeluGrad",
44
- backendName: "webgl",
45
- kernelFunc: g
46
- };
47
- a(p);
48
- export {
49
- d as gelu
36
+ }
50
37
  };
38
+ function s(e) {
39
+ let { dy: t, x: n } = e.inputs, r = e.backend, i = new o(n.shape);
40
+ return r.runWebGLProgram(i, [t, n], "float32");
41
+ }
42
+ e({
43
+ kernelName: "GeluGrad",
44
+ backendName: "webgl",
45
+ kernelFunc: s
46
+ });
47
+ //#endregion
48
+ export { a as gelu };
@@ -1,23 +1,14 @@
1
- import { c as e, aO as r } from "../../index-CUXkjxiT.js";
2
- import { u as s, l as N } from "../../kernel_funcs_utils-pq0CK9co.js";
3
- import { y as l } from "../../shared-3agzAqQ_.js";
4
- const a = N + `
5
- return x < 0.0 ? NAN : log(x);
6
- `, t = `
7
- vec4 result = log(x);
8
- bvec4 isNaN = isnan(x);
9
- result.r = isNaN.r ? x.r : (x.r < 0.0 ? NAN : result.r);
10
- result.g = isNaN.g ? x.g : (x.g < 0.0 ? NAN : result.g);
11
- result.b = isNaN.b ? x.b : (x.b < 0.0 ? NAN : result.b);
12
- result.a = isNaN.a ? x.a : (x.a < 0.0 ? NAN : result.a);
13
- return result;
14
- `, n = s({
15
- opSnippet: a,
16
- packedOpSnippet: t,
17
- cpuKernelImpl: l
18
- }), o = {
19
- kernelName: r,
20
- backendName: "webgl",
21
- kernelFunc: n
22
- };
23
- e(o);
1
+ import { Ii as e } from "../../dist-BewPQWjc.js";
2
+ import { y as t } from "../../shared-DOgWaqvL.js";
3
+ import { i as n, t as r } from "../../kernel_funcs_utils-HiXOOx3f.js";
4
+ //#region lib/ops/webgl/log.ts
5
+ e({
6
+ kernelName: "Log",
7
+ backendName: "webgl",
8
+ kernelFunc: n({
9
+ opSnippet: r + "\n return x < 0.0 ? NAN : log(x);\n",
10
+ packedOpSnippet: "\n vec4 result = log(x);\n bvec4 isNaN = isnan(x);\n result.r = isNaN.r ? x.r : (x.r < 0.0 ? NAN : result.r);\n result.g = isNaN.g ? x.g : (x.g < 0.0 ? NAN : result.g);\n result.b = isNaN.b ? x.b : (x.b < 0.0 ? NAN : result.b);\n result.a = isNaN.a ? x.a : (x.a < 0.0 ? NAN : result.a);\n return result;\n",
11
+ cpuKernelImpl: t
12
+ })
13
+ });
14
+ //#endregion
@@ -1,45 +1,37 @@
1
- import { isPackedTensor as h } from "../../utilities/packed.js";
2
- import { c as b, m as A, a as c, n as G } from "../../index-CUXkjxiT.js";
3
- import { matMulMul as P } from "../matMulMul.js";
4
- import { matMulGelu as R } from "../matMulGelu.js";
5
- import { m as f } from "../../mat_mul-DhG0Newp.js";
6
- import { r as B } from "../../relu-J_X6MUzx.js";
7
- import { r as v } from "../../reshape-BDOuCSNW.js";
8
- import { t as g } from "../../transpose-9kRxIXWR.js";
9
- function q(d) {
10
- const { A: i, B: m } = d.inputs, { transposeA: o, transposeB: a, scale: M, activation: u, scaleA: k, scaleB: w, forceOutputShape: n, perm: p } = d.attrs, N = !h(i), U = !h(m);
11
- if (N && U) {
12
- const t = k !== void 0 ? A(i, c(k)) : i, r = w !== void 0 ? A(m, c(w)) : m;
13
- let e;
14
- if (M !== void 0)
15
- e = P(t, r, c(M), o, a);
16
- else if (u === "gelu")
17
- e = R(t, r);
18
- else if (u === "relu2") {
19
- const s = f(t, r, o, a), l = B(s);
20
- s.dispose(), e = G(l), l.dispose();
21
- } else u === "relu" ? e = B(f(t, r, o, a)) : e = f(t, r, o, a);
22
- if (p)
23
- if (n) {
24
- const s = v(e, n);
25
- e.dispose();
26
- const l = g(s, p);
27
- return s.dispose(), l;
28
- } else {
29
- const s = g(e, p);
30
- return e.dispose(), s;
31
- }
32
- else if (n) {
33
- const s = v(e, n);
34
- return e.dispose(), s;
35
- } else
36
- return e;
37
- }
38
- throw new Error("Not implemented: matMul16 with packed inputs");
1
+ import { Ii as e, Wr as t, _n as n, br as r, gr as i, hn as a, lt as o, w as s } from "../../dist-BewPQWjc.js";
2
+ import { isPackedTensor as c } from "../../utilities/packed.js";
3
+ import { matMulMul as l } from "../matMulMul.js";
4
+ import { matMulGelu as u } from "../matMulGelu.js";
5
+ //#region lib/ops/webgl/matMul16.ts
6
+ function d(e) {
7
+ let { A: d, B: f } = e.inputs, { transposeA: p, transposeB: m, scale: h, activation: g, scaleA: _, scaleB: v, forceOutputShape: y, perm: b } = e.attrs, x = !c(d), S = !c(f);
8
+ if (x && S) {
9
+ let e = _ === void 0 ? d : t(d, n(_)), c = v === void 0 ? f : t(f, n(v)), x;
10
+ if (h !== void 0) x = l(e, c, n(h), p, m);
11
+ else if (g === "gelu") x = u(e, c);
12
+ else if (g === "relu2") {
13
+ let t = i(e, c, p, m), n = o(t);
14
+ t.dispose(), x = a(n), n.dispose();
15
+ } else x = g === "relu" ? o(i(e, c, p, m)) : i(e, c, p, m);
16
+ if (b) if (y) {
17
+ let e = r(x, y);
18
+ x.dispose();
19
+ let t = s(e, b);
20
+ return e.dispose(), t;
21
+ } else {
22
+ let e = s(x, b);
23
+ return x.dispose(), e;
24
+ }
25
+ else if (y) {
26
+ let e = r(x, y);
27
+ return x.dispose(), e;
28
+ } else return x;
29
+ }
30
+ throw Error("Not implemented: matMul16 with packed inputs");
39
31
  }
40
- const C = {
41
- kernelName: "MatMul16",
42
- backendName: "webgl",
43
- kernelFunc: q
44
- };
45
- b(C);
32
+ e({
33
+ kernelName: "MatMul16",
34
+ backendName: "webgl",
35
+ kernelFunc: d
36
+ });
37
+ //#endregion