@genai-fi/nanogpt 0.11.0 → 0.12.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 (236) hide show
  1. package/dist/Generator.js +29 -29
  2. package/dist/{RealDiv-Ds-jvL09.js → RealDiv-C8neBwFi.js} +17 -17
  3. package/dist/{Reshape-Cd6e-Otn.js → Reshape-Bd4V_4X7.js} +1 -1
  4. package/dist/{Reshape-Ct266DEk.js → Reshape-Ck29jQSY.js} +7 -7
  5. package/dist/TeachableLLM.d.ts +2 -1
  6. package/dist/TeachableLLM.js +9 -9
  7. package/dist/Trainer.d.ts +4 -2
  8. package/dist/Trainer.js +12 -9
  9. package/dist/{axis_util-DofAuy0p.js → axis_util-DGqbT-FX.js} +1 -1
  10. package/dist/backend.js +2 -2
  11. package/dist/{backend_util-C7NWHpv7.js → backend_util-DC3rBo_H.js} +18 -18
  12. package/dist/{backend_webgpu-B0Vls736.js → backend_webgpu-mbhNnlx9.js} +10 -10
  13. package/dist/{broadcast_to-DDaNMbX7.js → broadcast_to-D1Dmg2Oz.js} +2 -2
  14. package/dist/checks/appendCache.js +2 -2
  15. package/dist/checks/attentionMask.js +3 -3
  16. package/dist/checks/gelu.js +2 -2
  17. package/dist/checks/matMulGelu.js +2 -2
  18. package/dist/checks/normRMS.js +4 -4
  19. package/dist/checks/normRMSGrad.js +3 -3
  20. package/dist/checks/packUnpack.js +2 -2
  21. package/dist/checks/qkv.js +2 -2
  22. package/dist/checks/rope.js +2 -2
  23. package/dist/clip_by_value-fg2aKzUy.js +12 -0
  24. package/dist/{complex-DClmWqJt.js → complex-Cyg-eQeZ.js} +1 -1
  25. package/dist/concat-CSm2rMwe.js +17 -0
  26. package/dist/{concat_util-CHsJFZJJ.js → concat_util-D0je5Ppu.js} +1 -1
  27. package/dist/{dataset-DcjWqUVQ.js → dataset-CVIJu7Xa.js} +3 -3
  28. package/dist/{dropout-OxuaJz6z.js → dropout-DLhSMNTZ.js} +14 -14
  29. package/dist/expand_dims-ChkuOp6I.js +11 -0
  30. package/dist/{exports_initializers-eS9QJ6ut.js → exports_initializers-1KWPiStI.js} +1 -1
  31. package/dist/{floor-DIb-lN_u.js → floor-BRMPgeIs.js} +1 -1
  32. package/dist/gather-BSULDalH.js +9 -0
  33. package/dist/{gelu-DqTbCx5x.js → gelu-BK1k-n1i.js} +1 -1
  34. package/dist/{gpgpu_math-CJcbnKPC.js → gpgpu_math-BJSTk_mW.js} +25 -25
  35. package/dist/{index-Dj5TkmPY.js → index-BBVLAXZD.js} +14 -14
  36. package/dist/{index-D0RBWjq8.js → index-Duu1Lvvv.js} +45 -45
  37. package/dist/{kernel_funcs_utils-CSaumNDs.js → kernel_funcs_utils-BtYrPoJu.js} +8 -8
  38. package/dist/layers/BaseLayer.js +2 -2
  39. package/dist/layers/CausalSelfAttention.js +6 -6
  40. package/dist/layers/MLP.js +4 -4
  41. package/dist/layers/PositionEmbedding.js +5 -5
  42. package/dist/layers/RMSNorm.js +3 -3
  43. package/dist/layers/RoPECache.js +4 -4
  44. package/dist/layers/TiedEmbedding.js +6 -6
  45. package/dist/layers/TransformerBlock.js +1 -1
  46. package/dist/loader/loadTransformers.js +1 -1
  47. package/dist/loader/oldZipLoad.js +17 -17
  48. package/dist/{log_sum_exp-VLZgbFAH.js → log_sum_exp-CVqLsVLl.js} +4 -4
  49. package/dist/main.d.ts +9 -0
  50. package/dist/main.js +69 -58
  51. package/dist/{matMul16-cDxwemKj.js → matMul16-xswmhSuF.js} +7 -7
  52. package/dist/{matMulGelu-B2s_80-H.js → matMulGelu-BpvgnYG8.js} +26 -26
  53. package/dist/mat_mul-Bn2BDpT4.js +11 -0
  54. package/dist/{mod-PrOKlFxH.js → mod-B4AUd1Np.js} +1 -1
  55. package/dist/models/NanoGPTV1.js +2 -2
  56. package/dist/models/model.js +9 -9
  57. package/dist/{ones-BX_wEgzB.js → ones-CBI1AQjb.js} +3 -3
  58. package/dist/ops/adamAdjust.js +1 -1
  59. package/dist/ops/adamMoments.js +1 -1
  60. package/dist/ops/add16.js +1 -1
  61. package/dist/ops/appendCache.js +3 -3
  62. package/dist/ops/attentionMask.js +1 -1
  63. package/dist/ops/concat16.js +2 -2
  64. package/dist/ops/cpu/adamAdjust.js +7 -7
  65. package/dist/ops/cpu/adamMoments.js +5 -5
  66. package/dist/ops/cpu/appendCache.js +6 -6
  67. package/dist/ops/cpu/attentionMask.js +6 -6
  68. package/dist/ops/cpu/fusedSoftmax.js +5 -5
  69. package/dist/ops/cpu/gatherSub.js +7 -7
  70. package/dist/ops/cpu/gelu.js +5 -5
  71. package/dist/ops/cpu/matMul16.js +2 -2
  72. package/dist/ops/cpu/matMulGelu.js +3 -3
  73. package/dist/ops/cpu/matMulMul.js +5 -5
  74. package/dist/ops/cpu/mulDropout.js +1 -1
  75. package/dist/ops/cpu/normRMS.js +5 -5
  76. package/dist/ops/cpu/qkv.js +3 -3
  77. package/dist/ops/cpu/rope.js +9 -9
  78. package/dist/ops/cpu/scatterSub.js +5 -5
  79. package/dist/ops/dot16.js +2 -2
  80. package/dist/ops/gatherSub.js +1 -1
  81. package/dist/ops/gelu.js +2 -2
  82. package/dist/ops/grads/add16.js +1 -1
  83. package/dist/ops/grads/attentionMask.js +2 -2
  84. package/dist/ops/grads/gelu.js +2 -2
  85. package/dist/ops/grads/matMul16.js +3 -3
  86. package/dist/ops/grads/matMulGelu.js +5 -5
  87. package/dist/ops/grads/normRMS.js +6 -6
  88. package/dist/ops/grads/pack16.js +3 -3
  89. package/dist/ops/grads/qkv.js +9 -9
  90. package/dist/ops/grads/rope.js +2 -2
  91. package/dist/ops/grads/softmax16.js +1 -1
  92. package/dist/ops/grads/unpack16.js +2 -2
  93. package/dist/ops/matMul16.js +3 -3
  94. package/dist/ops/matMulGelu.js +2 -2
  95. package/dist/ops/matMulMul.js +1 -1
  96. package/dist/ops/mul16.js +1 -1
  97. package/dist/ops/mulDrop.js +1 -1
  98. package/dist/ops/normRMS.js +1 -1
  99. package/dist/ops/pack16.js +2 -2
  100. package/dist/ops/qkv.js +1 -1
  101. package/dist/ops/reshape16.js +6 -6
  102. package/dist/ops/rope.js +2 -2
  103. package/dist/ops/scatterSub.js +1 -1
  104. package/dist/ops/slice16.js +2 -2
  105. package/dist/ops/softmax16.js +1 -1
  106. package/dist/ops/sub16.js +1 -1
  107. package/dist/ops/sum16.js +2 -2
  108. package/dist/ops/transpose16.js +6 -6
  109. package/dist/ops/unpack16.js +2 -2
  110. package/dist/ops/webgl/adamAdjust.js +2 -2
  111. package/dist/ops/webgl/adamMoments.js +1 -1
  112. package/dist/ops/webgl/appendCache.js +1 -1
  113. package/dist/ops/webgl/attentionMask.js +4 -4
  114. package/dist/ops/webgl/fusedSoftmax.js +6 -6
  115. package/dist/ops/webgl/gatherSub.js +1 -1
  116. package/dist/ops/webgl/gelu.js +2 -2
  117. package/dist/ops/webgl/log.js +3 -3
  118. package/dist/ops/webgl/matMul16.js +10 -10
  119. package/dist/ops/webgl/matMulGelu.js +4 -4
  120. package/dist/ops/webgl/matMulMul.js +2 -2
  121. package/dist/ops/webgl/mulDropout.js +1 -1
  122. package/dist/ops/webgl/normRMS.js +2 -2
  123. package/dist/ops/webgl/qkv.js +1 -1
  124. package/dist/ops/webgl/rope.js +4 -4
  125. package/dist/ops/webgl/scatterSub.js +1 -1
  126. package/dist/ops/webgpu/adamAdjust.js +3 -3
  127. package/dist/ops/webgpu/adamMoments.js +5 -5
  128. package/dist/ops/webgpu/add16.js +1 -1
  129. package/dist/ops/webgpu/appendCache.js +3 -3
  130. package/dist/ops/webgpu/attentionMask.js +5 -5
  131. package/dist/ops/webgpu/attentionMask32_program.js +2 -2
  132. package/dist/ops/webgpu/concat16.js +5 -5
  133. package/dist/ops/webgpu/gatherSub.js +3 -3
  134. package/dist/ops/webgpu/gelu.js +3 -3
  135. package/dist/ops/webgpu/matMul16.js +19 -19
  136. package/dist/ops/webgpu/matMul16_program.js +2 -2
  137. package/dist/ops/webgpu/mul16.js +1 -1
  138. package/dist/ops/webgpu/normRMS.js +2 -2
  139. package/dist/ops/webgpu/normRMSGrad.js +4 -4
  140. package/dist/ops/webgpu/pack16.js +3 -3
  141. package/dist/ops/webgpu/pack16_program.js +2 -2
  142. package/dist/ops/webgpu/qkv.js +4 -4
  143. package/dist/ops/webgpu/rope.js +3 -3
  144. package/dist/ops/webgpu/scatterSub.js +3 -3
  145. package/dist/ops/webgpu/slice16.js +4 -4
  146. package/dist/ops/webgpu/softmax16.js +4 -4
  147. package/dist/ops/webgpu/softmax16_program.js +2 -2
  148. package/dist/ops/webgpu/softmax16_subgroup_program.js +2 -2
  149. package/dist/ops/webgpu/softmax16grad.js +1 -1
  150. package/dist/ops/webgpu/sub16.js +1 -1
  151. package/dist/ops/webgpu/sum16.js +5 -5
  152. package/dist/ops/webgpu/transpose16.js +2 -2
  153. package/dist/ops/webgpu/transpose16_program.js +2 -2
  154. package/dist/ops/webgpu/transpose16_shared_program.js +3 -3
  155. package/dist/ops/webgpu/unpack16.js +5 -5
  156. package/dist/ops/webgpu/utils/binary_op.js +3 -3
  157. package/dist/ops/webgpu/utils/reductions.js +4 -4
  158. package/dist/{ops-FJapAPfm.js → ops-C2_OXuZ4.js} +35 -35
  159. package/dist/{pack16-k4jq6aMX.js → pack16-atD0eYRm.js} +6 -6
  160. package/dist/patches/webgpu_backend.js +8 -8
  161. package/dist/patches/webgpu_base.js +1 -1
  162. package/dist/patches/webgpu_program.js +2 -2
  163. package/dist/{random_width-UGQn4OWb.js → random_width-BN4wGJaW.js} +33 -33
  164. package/dist/{range-CuGvVN2c.js → range-DKmP1-OQ.js} +1 -1
  165. package/dist/relu-BsXmGzzu.js +9 -0
  166. package/dist/{reshape-CkjKPPqB.js → reshape-BI0yzp1T.js} +1 -1
  167. package/dist/{resize_nearest_neighbor-DB8k9KN_.js → resize_nearest_neighbor-BA_BX-ub.js} +25 -25
  168. package/dist/{rope-BmZmp9uP.js → rope-DJ7Y7c-u.js} +1 -1
  169. package/dist/{scatter_nd_util-BY22Cc-C.js → scatter_nd_util-k9MUVUkn.js} +1 -1
  170. package/dist/{selu_util-BuLbmbrl.js → selu_util-DyW0X1WG.js} +5 -5
  171. package/dist/{shared-B7USJZgw.js → shared-Q3BS6T03.js} +1 -1
  172. package/dist/{shared-BQboIImQ.js → shared-nnSWpC3u.js} +6 -6
  173. package/dist/{slice-Aqy7KbJh.js → slice-wBNvzVyz.js} +3 -3
  174. package/dist/{slice_util-D8CQRenR.js → slice_util-zN8KFC5I.js} +7 -7
  175. package/dist/{softmax-faLoUZVT.js → softmax-DfuYyjMh.js} +1 -1
  176. package/dist/split-BYrLboMq.js +9 -0
  177. package/dist/squeeze-Bk8Brcct.js +10 -0
  178. package/dist/{stack-WJK22CFn.js → stack-CDWShFHF.js} +1 -1
  179. package/dist/{step-dXR33iOg.js → step-BS5JXRR6.js} +14 -14
  180. package/dist/sum-BPUfDB2X.js +11 -0
  181. package/dist/{tensor-BQqrDvpx.js → tensor-CEt9Nm2s.js} +1 -1
  182. package/dist/{tensor1d-LxP9asMm.js → tensor1d-Cc_KCIDg.js} +1 -1
  183. package/dist/{tensor2d-BN1sSfQO.js → tensor2d-BN97fF71.js} +1 -1
  184. package/dist/{tensor4d-DVwr7pLF.js → tensor4d-vuDDgdUI.js} +1 -1
  185. package/dist/{tfjs_backend-Vi4JfLzT.js → tfjs_backend-806hyYve.js} +36 -36
  186. package/dist/tile-OWUvpIVt.js +11 -0
  187. package/dist/tokeniser/BaseTokeniser.d.ts +6 -8
  188. package/dist/tokeniser/BaseTokeniser.js +6 -6
  189. package/dist/tokeniser/CharTokeniser.d.ts +6 -6
  190. package/dist/tokeniser/CharTokeniser.js +26 -26
  191. package/dist/tokeniser/bpe.d.ts +6 -6
  192. package/dist/tokeniser/bpe.js +9 -9
  193. package/dist/tokeniser/type.d.ts +6 -8
  194. package/dist/training/Adam.js +2 -2
  195. package/dist/training/AdamExt.js +1 -1
  196. package/dist/training/DatasetBuilder.d.ts +1 -1
  197. package/dist/training/DatasetBuilder.js +29 -29
  198. package/dist/training/FullTrainer.js +1 -1
  199. package/dist/training/Trainer.d.ts +5 -4
  200. package/dist/training/Trainer.js +37 -40
  201. package/dist/training/sparseCrossEntropy.js +3 -3
  202. package/dist/training/tasks/ConversationTask.d.ts +11 -0
  203. package/dist/training/tasks/ConversationTask.js +26 -0
  204. package/dist/training/tasks/PretrainingTask.d.ts +11 -0
  205. package/dist/training/tasks/PretrainingTask.js +34 -0
  206. package/dist/training/tasks/StartSentenceTask.d.ts +12 -0
  207. package/dist/training/tasks/StartSentenceTask.js +42 -0
  208. package/dist/training/tasks/Task.d.ts +8 -0
  209. package/dist/training/tasks/Task.js +44 -0
  210. package/dist/{transpose-JawVKyZy.js → transpose-BUkQCJp9.js} +7 -7
  211. package/dist/{unsorted_segment_sum-LAbmE9G4.js → unsorted_segment_sum-BljxHhCY.js} +78 -78
  212. package/dist/utilities/dummy.js +3 -3
  213. package/dist/utilities/multinomialCPU.js +2 -2
  214. package/dist/utilities/packed.js +1 -1
  215. package/dist/utilities/performance.js +1 -1
  216. package/dist/utilities/profile.js +1 -1
  217. package/dist/utilities/safetensors.js +2 -2
  218. package/dist/utilities/sentences.d.ts +1 -1
  219. package/dist/utilities/sentences.js +11 -11
  220. package/dist/utilities/weights.js +2 -2
  221. package/dist/{variable-DQ9yYgEU.js → variable-DPt_Iuog.js} +1 -1
  222. package/dist/{webgpu_program-CAE4RICo.js → webgpu_program-BpWRlghH.js} +1 -1
  223. package/dist/{webgpu_util-BdovYhXr.js → webgpu_util-DMiKzzQM.js} +7 -7
  224. package/dist/{zeros-DeiE2zTa.js → zeros-5YROwwUH.js} +2 -2
  225. package/dist/{zeros_like-BAz3iKru.js → zeros_like-De4n1C3m.js} +57 -57
  226. package/package.json +1 -1
  227. package/dist/clip_by_value-Dn5tzexi.js +0 -12
  228. package/dist/concat-C6X3AAlQ.js +0 -17
  229. package/dist/expand_dims-BzfJK2uc.js +0 -11
  230. package/dist/gather-BcO5UQNJ.js +0 -9
  231. package/dist/mat_mul-DxpNTCRz.js +0 -11
  232. package/dist/relu-Cf80uA2p.js +0 -9
  233. package/dist/split-BNz5jcGc.js +0 -9
  234. package/dist/squeeze--YMgaAAf.js +0 -10
  235. package/dist/sum-BdplSvq_.js +0 -11
  236. package/dist/tile-CvN_LyVr.js +0 -11
@@ -1,12 +1,12 @@
1
- import { h, a7 as l } from "../../index-D0RBWjq8.js";
2
- import { b as f } from "../../matMul16-cDxwemKj.js";
1
+ import { f as h, a6 as f } from "../../index-Duu1Lvvv.js";
2
+ import { b as l } from "../../matMul16-xswmhSuF.js";
3
3
  import { slice16 as a } from "../slice16.js";
4
4
  import { isPackedTensor as u } from "../../utilities/packed.js";
5
5
  function k(i) {
6
6
  const { x: n, kernel: c } = i.inputs, { heads: e } = i.attrs, r = n.shape[0], t = n.shape[1], s = n.shape[2], p = u(n);
7
- if (l(c.shape, [p ? s * 2 : s, 3 * s], "Error in QKV: "), s % e !== 0)
7
+ if (f(c.shape, [p ? s * 2 : s, 3 * s], "Error in QKV: "), s % e !== 0)
8
8
  throw new Error(`Channel dimension ${s} must be divisible by number of heads ${e} in QKV.`);
9
- const o = f(n, c, !1, !1, {
9
+ const o = l(n, c, !1, !1, {
10
10
  forceOutputShape: [r, t, 3 * e, s / e],
11
11
  perm: [0, 2, 1, 3]
12
12
  }), m = [
@@ -1,7 +1,7 @@
1
1
  import { isPackedTensor as y } from "../../utilities/packed.js";
2
- import { e as c } from "../../webgpu_program-CAE4RICo.js";
3
- import { f as x, c as l } from "../../webgpu_util-BdovYhXr.js";
4
- import { h as w, a7 as b } from "../../index-D0RBWjq8.js";
2
+ import { e as c } from "../../webgpu_program-BpWRlghH.js";
3
+ import { f as x, c as l } from "../../webgpu_util-DMiKzzQM.js";
4
+ import { f as w, a6 as b } from "../../index-Duu1Lvvv.js";
5
5
  class v {
6
6
  variableNames = ["x", "sin", "cos"];
7
7
  outputShape;
@@ -1,6 +1,6 @@
1
- import { e as p } from "../../webgpu_program-CAE4RICo.js";
2
- import { f as u, c as d } from "../../webgpu_util-BdovYhXr.js";
3
- import { h, a7 as o } from "../../index-D0RBWjq8.js";
1
+ import { e as p } from "../../webgpu_program-BpWRlghH.js";
2
+ import { f as u, c as d } from "../../webgpu_util-DMiKzzQM.js";
3
+ import { f as h, a6 as o } from "../../index-Duu1Lvvv.js";
4
4
  class b {
5
5
  variableNames = ["labels", "softmaxProbs", "dy"];
6
6
  outputShape;
@@ -1,7 +1,7 @@
1
- import { b as u, c as m, e as l } from "../../webgpu_program-CAE4RICo.js";
2
- import { f, c as g } from "../../webgpu_util-BdovYhXr.js";
3
- import { h as S, V as k } from "../../index-D0RBWjq8.js";
4
- import { p as y, a as $ } from "../../slice_util-D8CQRenR.js";
1
+ import { b as u, c as m, e as l } from "../../webgpu_program-BpWRlghH.js";
2
+ import { f, c as g } from "../../webgpu_util-DMiKzzQM.js";
3
+ import { f as S, U as k } from "../../index-Duu1Lvvv.js";
4
+ import { p as y, a as $ } from "../../slice_util-zN8KFC5I.js";
5
5
  function x(o) {
6
6
  switch (o) {
7
7
  case 1:
@@ -1,12 +1,12 @@
1
- import { h, V as S, e as b } from "../../index-D0RBWjq8.js";
1
+ import { f as S, U as h, e as b } from "../../index-Duu1Lvvv.js";
2
2
  import { reshape16 as d } from "../reshape16.js";
3
3
  import x from "./softmax16_program.js";
4
4
  import k from "./softmax16_subgroup_program.js";
5
5
  import l from "./utils/deviceInfo.js";
6
- import { r as z } from "../../reshape-CkjKPPqB.js";
6
+ import { r as z } from "../../reshape-BI0yzp1T.js";
7
7
  function F(a) {
8
8
  const { inputs: t, backend: o, attrs: p } = a, { logits: e } = t, { dim: r } = p, m = o.subgroupMinSize, i = o.subgroupMaxSize, c = l(o).subgroupsSupported, s = z(e, [
9
- S(e.shape) / e.shape[r],
9
+ h(e.shape) / e.shape[r],
10
10
  e.shape[r]
11
11
  ]), u = c ? new k(s.shape, m, i) : new x(s.shape), f = o.runWebGPUProgram(u, [s], "packedF16");
12
12
  s.dispose();
@@ -18,4 +18,4 @@ const I = {
18
18
  backendName: "webgpu",
19
19
  kernelFunc: F
20
20
  };
21
- h(I);
21
+ S(I);
@@ -1,5 +1,5 @@
1
- import { e } from "../../webgpu_program-CAE4RICo.js";
2
- import { f as o } from "../../webgpu_util-BdovYhXr.js";
1
+ import { e } from "../../webgpu_program-BpWRlghH.js";
2
+ import { f as o } from "../../webgpu_util-DMiKzzQM.js";
3
3
  class i {
4
4
  variableNames = ["logits"];
5
5
  outputShape;
@@ -1,5 +1,5 @@
1
- import { e as o } from "../../webgpu_program-CAE4RICo.js";
2
- import { f as u } from "../../webgpu_util-BdovYhXr.js";
1
+ import { e as o } from "../../webgpu_program-BpWRlghH.js";
2
+ import { f as u } from "../../webgpu_util-DMiKzzQM.js";
3
3
  class i {
4
4
  variableNames = ["logits"];
5
5
  outputShape;
@@ -1,4 +1,4 @@
1
- import { h as p } from "../../index-D0RBWjq8.js";
1
+ import { f as p } from "../../index-Duu1Lvvv.js";
2
2
  import { createReduceInfo as m, reduce as l, ReduceProgram as i } from "./utils/reductions.js";
3
3
  import { isPackedTensor as n } from "../../utilities/packed.js";
4
4
  import k from "./utils/deviceInfo.js";
@@ -1,4 +1,4 @@
1
- import { h as t } from "../../index-D0RBWjq8.js";
1
+ import { f as t } from "../../index-Duu1Lvvv.js";
2
2
  import { BinaryOpProgram as s } from "./utils/binary_op.js";
3
3
  import { B as p } from "../../binary_op_util-pKXltfxI.js";
4
4
  function c(e) {
@@ -1,10 +1,10 @@
1
- import { createReduceInfo as h, reduce as g, ReduceProgram as x } from "./utils/reductions.js";
2
- import { h as k, af as A } from "../../index-D0RBWjq8.js";
1
+ import { createReduceInfo as g, reduce as h, ReduceProgram as x } from "./utils/reductions.js";
2
+ import { f as k, af as A } from "../../index-Duu1Lvvv.js";
3
3
  import { isPackedTensor as P } from "../../utilities/packed.js";
4
4
  import { transpose16 as b } from "../transpose16.js";
5
5
  import I from "./utils/deviceInfo.js";
6
- import { s as w } from "../../sum-BdplSvq_.js";
7
- import { a as D, b as K } from "../../axis_util-DofAuy0p.js";
6
+ import { s as w } from "../../sum-BPUfDB2X.js";
7
+ import { a as D, b as K } from "../../axis_util-DGqbT-FX.js";
8
8
  class v extends x {
9
9
  shaderKey = "sum16";
10
10
  constructor(e, t, o) {
@@ -27,7 +27,7 @@ function y(r) {
27
27
  const i = D(n, e.shape.length);
28
28
  let s = e;
29
29
  i != null && (s = b(e, i), n = K(n.length, s.shape.length), c.push(s));
30
- const p = h([s], -1), f = new v(u, p, m), d = g(f, [s], a);
30
+ const p = g([s], -1), f = new v(u, p, m), d = h(f, [s], a);
31
31
  return c.forEach((l) => l.dispose()), d;
32
32
  }
33
33
  const N = {
@@ -1,9 +1,9 @@
1
1
  import { isPackedTensor as d } from "../../utilities/packed.js";
2
- import { h as k, e as l } from "../../index-D0RBWjq8.js";
2
+ import { f as k, e as l } from "../../index-Duu1Lvvv.js";
3
3
  import { reshape16 as u } from "../reshape16.js";
4
4
  import P from "./transpose16_shared_program.js";
5
5
  import b from "./transpose16_program.js";
6
- import { t as T } from "../../transpose-JawVKyZy.js";
6
+ import { t as T } from "../../transpose-BUkQCJp9.js";
7
7
  function w(p) {
8
8
  const { inputs: h, attrs: i } = p, { x: e } = h, { perm: r } = i, a = p.backend, m = d(e);
9
9
  if (m && r[r.length - 1] !== e.shape.length - 1) {
@@ -1,5 +1,5 @@
1
- import { f as a, c as i } from "../../webgpu_util-BdovYhXr.js";
2
- import { b as h, e as d, c as n } from "../../webgpu_program-CAE4RICo.js";
1
+ import { f as a, c as i } from "../../webgpu_util-DMiKzzQM.js";
2
+ import { b as h, e as d, c as n } from "../../webgpu_program-BpWRlghH.js";
3
3
  function p(r) {
4
4
  const e = r.length;
5
5
  if (e > 6)
@@ -1,6 +1,6 @@
1
- import { c as a } from "../../webgpu_util-BdovYhXr.js";
2
- import { e as l } from "../../webgpu_program-CAE4RICo.js";
3
- import { y as p } from "../../index-D0RBWjq8.js";
1
+ import { c as a } from "../../webgpu_util-DMiKzzQM.js";
2
+ import { e as l } from "../../webgpu_program-BpWRlghH.js";
3
+ import { x as p } from "../../index-Duu1Lvvv.js";
4
4
  class n {
5
5
  variableNames = ["A"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { f as c, c as r } from "../../webgpu_util-BdovYhXr.js";
2
- import { e as u } from "../../webgpu_program-CAE4RICo.js";
3
- import { h as p } from "../../index-D0RBWjq8.js";
1
+ import { f as c, c as r } from "../../webgpu_util-DMiKzzQM.js";
2
+ import { e as u } from "../../webgpu_program-BpWRlghH.js";
3
+ import { f as p } from "../../index-Duu1Lvvv.js";
4
4
  class l {
5
5
  outputShape;
6
6
  shaderKey = "Unpack16";
@@ -40,9 +40,9 @@ function h(e) {
40
40
  const o = [{ type: "float32", data: [1 / a] }];
41
41
  return i.runWebGPUProgram(n, [t], "float32", s ? o : void 0);
42
42
  }
43
- const d = {
43
+ const f = {
44
44
  kernelName: "Unpack16",
45
45
  backendName: "webgpu",
46
46
  kernelFunc: h
47
47
  };
48
- p(d);
48
+ p(f);
@@ -1,8 +1,8 @@
1
- import { f as s, c as n } from "../../../webgpu_util-BdovYhXr.js";
1
+ import { f as s, c as n } from "../../../webgpu_util-DMiKzzQM.js";
2
2
  import { g as p } from "../../../binary_op_util-pKXltfxI.js";
3
3
  import { B as b } from "../../../binary_op_util-pKXltfxI.js";
4
- import { $ as u, ag as r } from "../../../index-D0RBWjq8.js";
5
- import { e as l } from "../../../webgpu_program-CAE4RICo.js";
4
+ import { _ as u, ag as r } from "../../../index-Duu1Lvvv.js";
5
+ import { e as l } from "../../../webgpu_program-BpWRlghH.js";
6
6
  class x {
7
7
  dispatch;
8
8
  dispatchLayout;
@@ -1,8 +1,8 @@
1
- import { af as l, V as c, e as S } from "../../../index-D0RBWjq8.js";
2
- import { e as a } from "../../../webgpu_program-CAE4RICo.js";
1
+ import { af as l, U as c, e as S } from "../../../index-Duu1Lvvv.js";
2
+ import { e as a } from "../../../webgpu_program-BpWRlghH.js";
3
3
  import { reshape16 as b } from "../../reshape16.js";
4
- import { f } from "../../../webgpu_util-BdovYhXr.js";
5
- import { c as h } from "../../../axis_util-DofAuy0p.js";
4
+ import { f } from "../../../webgpu_util-DMiKzzQM.js";
5
+ import { c as h } from "../../../axis_util-DGqbT-FX.js";
6
6
  function d(e, u, t, i) {
7
7
  return e && !u ? `
8
8
  bestValue = subgroupAdd(bestValue);
@@ -1,26 +1,26 @@
1
- import { q as b, u, y as p, E as k, bf as hs, bg as bs, at as ds, bh as gs, bi as Es, N as W, bj as $s, bk as ys, bl as _s, bm as Ns, b as $, m as h, n as G, l as S, i as Ss, c as d, bn as ks, x as K, d as Is, a7 as L, aZ as ns, b0 as Ts, a2 as Ms } from "./index-D0RBWjq8.js";
2
- import { t as j } from "./tile-CvN_LyVr.js";
3
- import { e as R } from "./expand_dims-BzfJK2uc.js";
4
- import { t as x } from "./tensor1d-LxP9asMm.js";
1
+ import { o as b, q as u, x as p, E as k, bf as hs, bg as bs, at as ds, bh as gs, bi as Es, L as W, bj as $s, bk as ys, bl as _s, bm as Ns, b as $, m as h, l as G, j as S, h as Ss, c as d, bn as ks, w as K, d as Is, a6 as L, aZ as ns, b0 as Ts, a1 as Ms } from "./index-Duu1Lvvv.js";
2
+ import { t as v } from "./tile-OWUvpIVt.js";
3
+ import { e as R } from "./expand_dims-ChkuOp6I.js";
4
+ import { t as x } from "./tensor1d-Cc_KCIDg.js";
5
5
  import { n as xs, a as As, b as ws } from "./non_max_suppression_impl-B2W7YjZB.js";
6
- import { r as Os, l as ls, g as ts, w as q, a as rs, m as Q, b as Gs, c as Ls, u as us, n as ps, d as Bs, e as Ds, s as Ws, f as Rs, h as qs, i as zs } from "./resize_nearest_neighbor-DB8k9KN_.js";
7
- import { s as ms } from "./split-BNz5jcGc.js";
8
- import { s as _ } from "./sum-BdplSvq_.js";
9
- import { s as T } from "./slice-Aqy7KbJh.js";
10
- import { r as F } from "./range-CuGvVN2c.js";
11
- import { t as Vs } from "./tensor-BQqrDvpx.js";
12
- import { n as C, t as as } from "./transpose-JawVKyZy.js";
13
- import { r as O } from "./reshape-CkjKPPqB.js";
14
- import { s as Z } from "./stack-WJK22CFn.js";
15
- import { z as Cs } from "./zeros-DeiE2zTa.js";
16
- import { s as Ps } from "./squeeze--YMgaAAf.js";
17
- import { c as X } from "./concat-C6X3AAlQ.js";
18
- import { m as w } from "./mat_mul-DxpNTCRz.js";
19
- import { t as J } from "./tensor2d-BN1sSfQO.js";
20
- import { o as Ys } from "./ones-BX_wEgzB.js";
21
- import { r as fs } from "./relu-Cf80uA2p.js";
22
- import { a as is, e as ss, l as vs } from "./log_sum_exp-VLZgbFAH.js";
23
- import { e as js } from "./axis_util-DofAuy0p.js";
6
+ import { r as Os, l as ls, g as ts, w as q, a as rs, m as Q, b as Gs, c as Ls, u as us, n as ps, d as Bs, e as Ds, s as Ws, f as Rs, h as qs, i as zs } from "./resize_nearest_neighbor-BA_BX-ub.js";
7
+ import { s as ms } from "./split-BYrLboMq.js";
8
+ import { s as _ } from "./sum-BPUfDB2X.js";
9
+ import { s as T } from "./slice-wBNvzVyz.js";
10
+ import { r as F } from "./range-DKmP1-OQ.js";
11
+ import { t as Vs } from "./tensor-CEt9Nm2s.js";
12
+ import { n as C, t as as } from "./transpose-BUkQCJp9.js";
13
+ import { r as O } from "./reshape-BI0yzp1T.js";
14
+ import { s as Z } from "./stack-CDWShFHF.js";
15
+ import { z as Cs } from "./zeros-5YROwwUH.js";
16
+ import { s as Ps } from "./squeeze-Bk8Brcct.js";
17
+ import { c as X } from "./concat-CSm2rMwe.js";
18
+ import { m as w } from "./mat_mul-Bn2BDpT4.js";
19
+ import { t as J } from "./tensor2d-BN97fF71.js";
20
+ import { o as Ys } from "./ones-CBI1AQjb.js";
21
+ import { r as fs } from "./relu-BsXmGzzu.js";
22
+ import { a as is, e as ss, l as js } from "./log_sum_exp-CVqLsVLl.js";
23
+ import { e as vs } from "./axis_util-DGqbT-FX.js";
24
24
  function Fs(e, a, n) {
25
25
  const r = u(e, "x", "bincount"), t = u(a, "weights", "bincount");
26
26
  p(r.dtype === "int32", () => `Error in bincount: input dtype must be int32, but got ${r.dtype}`), p(n >= 0, () => `size must be non-negative, but got ${n}.`), p(t.size === r.size || t.size === 0, () => `Error in bincount: weights must have the same size as input or0-length, but got input shape: ${r.shape}, weights shape: ${t.shape}.`);
@@ -42,11 +42,11 @@ function Hs(e, a, n, r = "float32") {
42
42
  if (n == null)
43
43
  return o;
44
44
  if (n.length === 1)
45
- return j(R(o, 0), [n[0], 1, 1]);
45
+ return v(R(o, 0), [n[0], 1, 1]);
46
46
  if (n.length === 2)
47
- return j(R(R(o, 0), 0), [n[0], n[1], 1, 1]);
47
+ return v(R(R(o, 0), 0), [n[0], n[1], 1, 1]);
48
48
  if (n.length === 3)
49
- return j(R(R(R(o, 0), 0), 0), [
49
+ return v(R(R(R(o, 0), 0), 0), [
50
50
  n[0],
51
51
  n[1],
52
52
  n[2],
@@ -74,7 +74,7 @@ function nn(e) {
74
74
  const a = u(e, "image", "grayscaleToRGB"), n = a.rank - 1, r = a.shape[n];
75
75
  p(a.rank >= 2, () => `Error in grayscaleToRGB: images must be at least rank 2, but got rank ${a.rank}.`), p(r === 1, () => `Error in grayscaleToRGB: last dimension of a grayscale image should be size 1, but got size ${r}.`);
76
76
  const t = new Array(a.rank);
77
- return t.fill(1, 0, n), t[n] = 3, j(a, t);
77
+ return t.fill(1, 0, n), t[n] = 3, v(a, t);
78
78
  }
79
79
  const tn = /* @__PURE__ */ b({ grayscaleToRGB_: nn });
80
80
  function en(e) {
@@ -294,11 +294,11 @@ function cs(e, a = !1) {
294
294
  const H = d(Y, w(U, w(es, Y)));
295
295
  s = X([T(s, [0, 0], [c, r]), H], 0);
296
296
  }
297
- const os = as(U), v = T(t, [0, c], [n, t.shape[1] - c]);
297
+ const os = as(U), j = T(t, [0, c], [n, t.shape[1] - c]);
298
298
  if (c === 0)
299
- t = d(v, w(w(v, i), os));
299
+ t = d(j, w(w(j, i), os));
300
300
  else {
301
- const H = d(v, w(w(v, i), os));
301
+ const H = d(j, w(w(j, i), os));
302
302
  t = X([T(t, [0, 0], [n, c]), H], 1);
303
303
  }
304
304
  return [i, s, t];
@@ -391,13 +391,13 @@ function Pn(e, a, n, r = N.SUM_BY_NONZERO_WEIGHTS) {
391
391
  return B(i, o, r);
392
392
  }
393
393
  const Yn = /* @__PURE__ */ b({ meanSquaredError_: Pn });
394
- function vn(e, a) {
394
+ function jn(e, a) {
395
395
  const n = u(e, "labels", "sigmoidCrossEntropyWithLogits"), r = u(a, "logits", "sigmoidCrossEntropyWithLogits");
396
396
  L(n.shape, r.shape, "Error in sigmoidCrossEntropyWithLogits: ");
397
397
  const t = fs(r), s = h(r, n), o = Rs(ss(C(ns(r))));
398
398
  return G(d(t, s), o);
399
399
  }
400
- function jn(e, a, n, r = 0, t = N.SUM_BY_NONZERO_WEIGHTS) {
400
+ function vn(e, a, n, r = 0, t = N.SUM_BY_NONZERO_WEIGHTS) {
401
401
  let s = u(e, "multiClassLabels", "sigmoidCrossEntropy");
402
402
  const o = u(a, "logits", "sigmoidCrossEntropy");
403
403
  let i = null;
@@ -405,19 +405,19 @@ function jn(e, a, n, r = 0, t = N.SUM_BY_NONZERO_WEIGHTS) {
405
405
  const c = $(r), f = $(1), m = $(0.5);
406
406
  s = G(h(s, d(f, c)), h(m, c));
407
407
  }
408
- const l = vn(s, o);
408
+ const l = jn(s, o);
409
409
  return B(l, i, t);
410
410
  }
411
- const Fn = /* @__PURE__ */ b({ sigmoidCrossEntropy_: jn });
411
+ const Fn = /* @__PURE__ */ b({ sigmoidCrossEntropy_: vn });
412
412
  function Zn(e, a, n = -1) {
413
413
  if (n === -1 && (n = a.rank - 1), n !== a.rank - 1)
414
414
  throw Error(`Softmax cross entropy along a non-last dimension is not yet supported. Labels / logits was rank ${a.rank} and dim was ${n}`);
415
415
  return Ms((t, s, o) => {
416
- const l = vs(s, [n], !0), c = d(W(s, "float32"), l);
416
+ const l = js(s, [n], !0), c = d(W(s, "float32"), l);
417
417
  o([t, c]);
418
418
  const f = C(h(c, t));
419
419
  return { value: _(f, [n]), gradFunc: (g, y) => {
420
- const [M, I] = y, A = js(g.shape, [n]);
420
+ const [M, I] = y, A = vs(g.shape, [n]);
421
421
  return [
422
422
  h(O(g, A), d(W(M, "float32"), ss(I))),
423
423
  h(O(g, A), d(ss(I), W(M, "float32")))
@@ -1,6 +1,6 @@
1
- import { j as o, e as i } from "./index-D0RBWjq8.js";
1
+ import { i, e as o } from "./index-Duu1Lvvv.js";
2
2
  import { isPackedTensor as t } from "./utilities/packed.js";
3
- import { s as u } from "./slice-Aqy7KbJh.js";
3
+ import { s as u } from "./slice-wBNvzVyz.js";
4
4
  const s = {
5
5
  kernelName: "Unpack16",
6
6
  inputsToSave: [],
@@ -9,11 +9,11 @@ const s = {
9
9
  x: () => d(n)
10
10
  })
11
11
  };
12
- o(s);
12
+ i(s);
13
13
  function p(n, a = 1, e = !1) {
14
14
  if (!t(n))
15
15
  return n;
16
- const r = i().runKernel("Unpack16", { x: n }, { scaling: a });
16
+ const r = o().runKernel("Unpack16", { x: n }, { scaling: a });
17
17
  return e && n.dispose(), r;
18
18
  }
19
19
  const c = {
@@ -27,9 +27,9 @@ const c = {
27
27
  }
28
28
  })
29
29
  };
30
- o(c);
30
+ i(c);
31
31
  function d(n, a = 1, e = 0) {
32
- return i().runKernel("Pack16", { x: n }, { scaling: a, padding: e });
32
+ return o().runKernel("Pack16", { x: n }, { scaling: a, padding: e });
33
33
  }
34
34
  export {
35
35
  s as a,
@@ -1,8 +1,8 @@
1
- import { V as c, ae as l, ab as m, y as n } from "../index-D0RBWjq8.js";
2
- import "../webgpu_util-BdovYhXr.js";
3
- import { W as y } from "../backend_webgpu-B0Vls736.js";
4
- import { compileProgram as P } from "./webgpu_program.js";
5
- import { m as M } from "../webgpu_program-CAE4RICo.js";
1
+ import { U as c, ae as l, aa as m, x as n } from "../index-Duu1Lvvv.js";
2
+ import "../webgpu_util-DMiKzzQM.js";
3
+ import { W as P } from "../backend_webgpu-mbhNnlx9.js";
4
+ import { compileProgram as y } from "./webgpu_program.js";
5
+ import { m as M } from "../webgpu_program-BpWRlghH.js";
6
6
  const b = (o, s) => {
7
7
  const i = o.limits.maxComputeWorkgroupsPerDimension, t = s.dispatchLayout, a = s.dispatch;
8
8
  if (a.every((r) => r <= i))
@@ -17,7 +17,7 @@ const b = (o, s) => {
17
17
  () => "Total dispatch size exceeds WebGPU maximum."
18
18
  ), [e, e, e]) : [e, e, 1];
19
19
  };
20
- class z extends y {
20
+ class U extends P {
21
21
  subgroupMaxSize;
22
22
  subgroupMinSize;
23
23
  constructor(s, i) {
@@ -42,7 +42,7 @@ class z extends y {
42
42
  });
43
43
  s.shaderKey = M(s, r, e);
44
44
  const d = m().getBool("WEBGPU_ENGINE_COMPILE_ONLY");
45
- return s.shaderKey in this.pipelineCache || (this.pipelineCache[s.shaderKey] = P(
45
+ return s.shaderKey in this.pipelineCache || (this.pipelineCache[s.shaderKey] = y(
46
46
  this.device,
47
47
  s,
48
48
  r,
@@ -52,5 +52,5 @@ class z extends y {
52
52
  }
53
53
  }
54
54
  export {
55
- z as default
55
+ U as default
56
56
  };
@@ -1,4 +1,4 @@
1
- import { f as n } from "../index-D0RBWjq8.js";
1
+ import { ad as n } from "../index-Duu1Lvvv.js";
2
2
  import p from "./webgpu_backend.js";
3
3
  function c(a) {
4
4
  n(
@@ -1,5 +1,5 @@
1
- import { ab as z, y as F, ac as O, ad as _ } from "../index-D0RBWjq8.js";
2
- import { g as A, d as b, a as x, b as g, s as L, t as l, c as y } from "../webgpu_program-CAE4RICo.js";
1
+ import { aa as z, x as F, ab as O, ac as _ } from "../index-Duu1Lvvv.js";
2
+ import { g as A, d as b, a as x, b as g, s as L, t as l, c as y } from "../webgpu_program-BpWRlghH.js";
3
3
  var N = /* @__PURE__ */ ((s) => (s[s.FROM_PIXELS = 0] = "FROM_PIXELS", s[s.DRAW = 1] = "DRAW", s))(N || {});
4
4
  const H = (s, t, e, o, u) => {
5
5
  const a = { dtype: o.dtype, shape: o.shape }, n = D(e, a, t), r = s.createShaderModule({ code: n, label: t.constructor.name });
@@ -1,36 +1,36 @@
1
- import { q as tt, u as R, y as S, E as X, cA as li, N as L, cB as ui, cC as Oa, cD as Ra, cE as ci, af as Yt, b0 as ut, c as U, w as _a, at as Ba, cF as Wa, n as z, D as Ga, $ as Lt, a2 as An, cG as In, cH as Pa, cI as Ua, cJ as Va, cK as ja, cL as Ha, cM as qa, cN as Ka, cO as Za, cP as Ja, bR as Xa, m as f, c9 as Ya, o as Qt, b as Q, l as W, ca as Qa, bX as to, a0 as lt, cQ as eo, bq as so, a3 as Y, cb as no, cc as io, cd as ro, cf as ao, ce as oo, cg as lo, cR as uo, cS as co, br as ho, H as po, bt as fo, cT as mo, bU as go, b_ as bo, C as yo, cU as wo, A as ko, bv as No, bw as xo, cV as vo, bx as So, by as Ao, bA as Io, bB as Co, ci as Do, cW as zo, cX as To, aI as Eo, cY as Lo, bD as $o, an as Fo, B as Mo, b$ as Oo, F as Ro, c0 as _o, bu as Bo, G as Wo, b2 as Go, aQ as Po, cj as Uo, ck as Vo, cl as jo, aJ as Ho, b5 as qo, ao as Ko, cZ as Zo, c_ as Jo, cm as Xo, am as Yo, c1 as Qo, c$ as tl, d0 as el, bG as sl, b7 as nl, V as hi, a$ as il, b8 as rl, co as al, M as ol, c2 as ll, aq as ul, bH as cl, bI as hl, P as pl, bJ as dl, d1 as fl, p as Ws, aK as ml, c3 as gl, aT as bl, cp as yl, aL as wl, I as kl, R as Nl, bd as xl, d2 as vl, be as Sl, d3 as Al, bL as Il, ba as Cl, bM as Dl, a_ as zl, bN as Tl, aH as El, cq as Ll, bO as $l, bP as Fl, S as Ml, J as Ol, bE as Rl, bK as _l, K as Bl, c5 as Wl, d4 as Gl, bb as Pl, aM as Ul, c7 as Vl, Q as jl, cu as Hl, bs as ql, T as Kl, as as Zl, bc as Jl, bQ as Xl, cz as Ne, d5 as Yl, j as Ql, d6 as v, t as y, d7 as Me, d8 as Oe, ac as $t, d as q, ab as tu, d9 as Cn, k as _t, aZ as ze, i as eu, x as su, a4 as we, O as nu, da as iu, a as Dn, db as ru, dc as zn, dd as au } from "./index-D0RBWjq8.js";
2
- import { c as te, k as Gs, m as Tn, b as ou, t as hs, a as Ps, s as Us, l as lu, p as uu, e as cu } from "./step-dXR33iOg.js";
3
- import { n as pt, t as P } from "./transpose-JawVKyZy.js";
4
- import { r as C } from "./reshape-CkjKPPqB.js";
5
- import { s as _ } from "./sum-BdplSvq_.js";
6
- import { m as Mt } from "./mat_mul-DxpNTCRz.js";
7
- import { j as Vs, o as Xe, G as pi, b as hu, s as pu, D as du, t as fu, A as mu, F as gu, e as Ht, q as En, r as Ye, l as bu, c as di, p as yu, z as fi, f as wu, x as ku, a as js, i as Qe, y as vt, B as Nu, E as Hs, w as xu, v as vu, n as Su, C as Au, k as Iu, u as Cu, h as Ln, g as Du, m as zu, d as Tu } from "./unsorted_segment_sum-LAbmE9G4.js";
8
- import { d as st, w as qt, b as Re, c as _e, l as qs, g as Wt, a as Eu, u as ts, f as Lu, m as mi, e as Ts, j as $u } from "./resize_nearest_neighbor-DB8k9KN_.js";
9
- import { s as Vt } from "./split-BNz5jcGc.js";
10
- import { e as Es, a as gi, g as Ks, c as Fu } from "./axis_util-DofAuy0p.js";
11
- import { e as xe, a as le, m as Te, l as Mu } from "./log_sum_exp-VLZgbFAH.js";
12
- import { t as Ce } from "./tile-CvN_LyVr.js";
13
- import { s as ps } from "./stack-WJK22CFn.js";
14
- import { o as he } from "./ones-BX_wEgzB.js";
15
- import { s as me } from "./slice-Aqy7KbJh.js";
16
- import { f as bi } from "./floor-DIb-lN_u.js";
17
- import { z as ft } from "./zeros-DeiE2zTa.js";
18
- import { s as Ou, b as Ru, c as _u, g as Bu, a as Wu, S as Gu } from "./selu_util-BuLbmbrl.js";
19
- import { p as Pu } from "./slice_util-D8CQRenR.js";
20
- import { c as Zs } from "./concat-C6X3AAlQ.js";
21
- import { e as ue } from "./expand_dims-BzfJK2uc.js";
22
- import { g as Uu } from "./gather-BcO5UQNJ.js";
23
- import { V as d, N as B, r as ds, c as Js, a as et, b as ge, e as Be, s as Xs, g as yi, f as wi, t as Ot, R as zt, h as ht, A as Gt, i as V, n as oe, j as nt, k as Vu, l as ju, m as We, o as Pt, p as Tt, q as ae, u as jt, v as qe, w as ce, x as Hu, y as es, z as fs, B as ki, C as St, D as $n, E as qu, F as Ku, G as Zu, H as Kt, I as Ju, J as Ys, K as It, L as Ke, M as Xu, O as ot, P as Ni, Q as mt, S as ee, T as Fn, U as ke, d as Et, W as Mn, X as Ge, Y as xi, Z as Qs, _ as Yu, $ as Qu, a0 as vi, a1 as tc } from "./tfjs_backend-Vi4JfLzT.js";
24
- import { s as tn } from "./squeeze--YMgaAAf.js";
25
- import { t as Ls } from "./tensor1d-LxP9asMm.js";
26
- import { r as Pe } from "./relu-Cf80uA2p.js";
27
- import { c as At } from "./clip_by_value-Dn5tzexi.js";
28
- import { s as Si } from "./softmax-faLoUZVT.js";
29
- import { M as ec, a as ms } from "./dropout-OxuaJz6z.js";
30
- import { e as sc, l as nc, i as Rt } from "./ops-FJapAPfm.js";
31
- import { t as ic } from "./tensor-BQqrDvpx.js";
32
- import { r as rc } from "./range-CuGvVN2c.js";
33
- import { v as ac } from "./variable-DQ9yYgEU.js";
1
+ import { o as tt, q as R, x as S, E as X, cA as li, L, cB as ui, cC as Oa, cD as Ra, cE as ci, af as Yt, b0 as ut, c as U, u as _a, at as Ba, cF as Wa, l as z, B as Ga, _ as Lt, a1 as An, cG as In, cH as Pa, cI as Ua, cJ as Va, cK as ja, cL as Ha, cM as qa, cN as Ka, cO as Za, cP as Ja, bR as Xa, m as f, c9 as Ya, n as Qt, b as Q, j as W, ca as Qa, bX as to, $ as lt, cQ as eo, bq as so, a2 as Y, cb as no, cc as io, cd as ro, cf as ao, ce as oo, cg as lo, cR as uo, cS as co, br as ho, D as po, bt as fo, cT as mo, bU as go, b_ as bo, C as yo, cU as wo, z as ko, bv as No, bw as xo, cV as vo, bx as So, by as Ao, bA as Io, bB as Co, ci as Do, cW as zo, cX as To, aI as Eo, cY as Lo, bD as $o, an as Fo, A as Mo, b$ as Oo, F as Ro, c0 as _o, bu as Bo, G as Wo, b2 as Go, aQ as Po, cj as Uo, ck as Vo, cl as jo, aJ as Ho, b5 as qo, ao as Ko, cZ as Zo, c_ as Jo, cm as Xo, am as Yo, c1 as Qo, c$ as tl, d0 as el, bG as sl, b7 as nl, U as hi, a$ as il, b8 as rl, co as al, M as ol, c2 as ll, aq as ul, bH as cl, bI as hl, P as pl, bJ as dl, d1 as fl, p as Ws, aK as ml, c3 as gl, aT as bl, cp as yl, aL as wl, H as kl, R as Nl, bd as xl, d2 as vl, be as Sl, d3 as Al, bL as Il, ba as Cl, bM as Dl, a_ as zl, bN as Tl, aH as El, cq as Ll, bO as $l, bP as Fl, S as Ml, I as Ol, bE as Rl, bK as _l, J as Bl, c5 as Wl, d4 as Gl, bb as Pl, aM as Ul, c7 as Vl, N as jl, cu as Hl, bs as ql, T as Kl, as as Zl, bc as Jl, bQ as Xl, cz as Ne, d5 as Yl, i as Ql, d6 as v, t as y, d7 as Me, d8 as Oe, ab as $t, d as q, aa as tu, d9 as Cn, k as _t, aZ as ze, h as eu, w as su, a3 as we, O as nu, da as iu, a as Dn, db as ru, dc as zn, dd as au } from "./index-Duu1Lvvv.js";
2
+ import { c as te, k as Gs, m as Tn, b as ou, t as hs, a as Ps, s as Us, l as lu, p as uu, e as cu } from "./step-BS5JXRR6.js";
3
+ import { n as pt, t as P } from "./transpose-BUkQCJp9.js";
4
+ import { r as C } from "./reshape-BI0yzp1T.js";
5
+ import { s as _ } from "./sum-BPUfDB2X.js";
6
+ import { m as Mt } from "./mat_mul-Bn2BDpT4.js";
7
+ import { j as Vs, o as Xe, G as pi, b as hu, s as pu, D as du, t as fu, A as mu, F as gu, e as Ht, q as En, r as Ye, l as bu, c as di, p as yu, z as fi, f as wu, x as ku, a as js, i as Qe, y as vt, B as Nu, E as Hs, w as xu, v as vu, n as Su, C as Au, k as Iu, u as Cu, h as Ln, g as Du, m as zu, d as Tu } from "./unsorted_segment_sum-BljxHhCY.js";
8
+ import { d as st, w as qt, b as Re, c as _e, l as qs, g as Wt, a as Eu, u as ts, f as Lu, m as mi, e as Ts, j as $u } from "./resize_nearest_neighbor-BA_BX-ub.js";
9
+ import { s as Vt } from "./split-BYrLboMq.js";
10
+ import { e as Es, a as gi, g as Ks, c as Fu } from "./axis_util-DGqbT-FX.js";
11
+ import { e as xe, a as le, m as Te, l as Mu } from "./log_sum_exp-CVqLsVLl.js";
12
+ import { t as Ce } from "./tile-OWUvpIVt.js";
13
+ import { s as ps } from "./stack-CDWShFHF.js";
14
+ import { o as he } from "./ones-CBI1AQjb.js";
15
+ import { s as me } from "./slice-wBNvzVyz.js";
16
+ import { f as bi } from "./floor-BRMPgeIs.js";
17
+ import { z as ft } from "./zeros-5YROwwUH.js";
18
+ import { s as Ou, b as Ru, c as _u, g as Bu, a as Wu, S as Gu } from "./selu_util-DyW0X1WG.js";
19
+ import { p as Pu } from "./slice_util-zN8KFC5I.js";
20
+ import { c as Zs } from "./concat-CSm2rMwe.js";
21
+ import { e as ue } from "./expand_dims-ChkuOp6I.js";
22
+ import { g as Uu } from "./gather-BSULDalH.js";
23
+ import { V as d, N as B, r as ds, c as Js, a as et, b as ge, e as Be, s as Xs, g as yi, f as wi, t as Ot, R as zt, h as ht, A as Gt, i as V, n as oe, j as nt, k as Vu, l as ju, m as We, o as Pt, p as Tt, q as ae, u as jt, v as qe, w as ce, x as Hu, y as es, z as fs, B as ki, C as St, D as $n, E as qu, F as Ku, G as Zu, H as Kt, I as Ju, J as Ys, K as It, L as Ke, M as Xu, O as ot, P as Ni, Q as mt, S as ee, T as Fn, U as ke, d as Et, W as Mn, X as Ge, Y as xi, Z as Qs, _ as Yu, $ as Qu, a0 as vi, a1 as tc } from "./tfjs_backend-806hyYve.js";
24
+ import { s as tn } from "./squeeze-Bk8Brcct.js";
25
+ import { t as Ls } from "./tensor1d-Cc_KCIDg.js";
26
+ import { r as Pe } from "./relu-BsXmGzzu.js";
27
+ import { c as At } from "./clip_by_value-fg2aKzUy.js";
28
+ import { s as Si } from "./softmax-DfuYyjMh.js";
29
+ import { M as ec, a as ms } from "./dropout-DLhSMNTZ.js";
30
+ import { e as sc, l as nc, i as Rt } from "./ops-C2_OXuZ4.js";
31
+ import { t as ic } from "./tensor-CEt9Nm2s.js";
32
+ import { r as rc } from "./range-DKmP1-OQ.js";
33
+ import { v as ac } from "./variable-DPt_Iuog.js";
34
34
  function oc(n, t, e, s, i, r = "NDHWC") {
35
35
  const a = R(n, "x", "avgPool3d", "float32");
36
36
  let o = a, l = !1;
@@ -1,4 +1,4 @@
1
- import { E as e, _ as f } from "./index-D0RBWjq8.js";
1
+ import { E as e, Z as f } from "./index-Duu1Lvvv.js";
2
2
  function E(n, o, r = 1, a = "float32") {
3
3
  if (r === 0)
4
4
  throw new Error("Cannot have a step of zero");
@@ -0,0 +1,9 @@
1
+ import { o as n, q as e, E as s, H as t } from "./index-Duu1Lvvv.js";
2
+ function u(r) {
3
+ const o = { x: e(r, "x", "relu") };
4
+ return s.runKernel(t, o);
5
+ }
6
+ const l = /* @__PURE__ */ n({ relu_: u });
7
+ export {
8
+ l as r
9
+ };
@@ -1,4 +1,4 @@
1
- import { q as o, u as t, E as a, R as p } from "./index-D0RBWjq8.js";
1
+ import { o, q as t, E as a, R as p } from "./index-Duu1Lvvv.js";
2
2
  function c(r, s) {
3
3
  const e = { x: t(r, "x", "reshape", "string_or_numeric") }, n = { shape: s };
4
4
  return a.runKernel(p, e, n);