@genai-fi/nanogpt 0.7.0 → 0.7.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 (152) hide show
  1. package/dist/Generator.js +13 -9
  2. package/dist/NanoGPTModel.js +10 -10
  3. package/dist/{RealDiv-C4hOvYOZ.js → RealDiv-CVYNbZxu.js} +11 -11
  4. package/dist/{Reshape-BLijOA8h.js → Reshape-CEsEp0AI.js} +2 -2
  5. package/dist/Reshape-Do18N3gO.js +30 -0
  6. package/dist/TeachableLLM.js +9 -5
  7. package/dist/{TiedEmbedding-BLltddza.js → TiedEmbedding-ccLBFiZi.js} +4 -4
  8. package/dist/{axis_util-DaAl5MER.js → axis_util-5DTW2tFV.js} +1 -1
  9. package/dist/backend.js +2 -2
  10. package/dist/{backend_util-DWiwsi2N.js → backend_util-C9Ut8n0Q.js} +40 -40
  11. package/dist/{broadcast_to-C4v-j9yA.js → broadcast_to-Ba9h_8DO.js} +2 -2
  12. package/dist/{concat-CsHeR4zV.js → concat-CbXTetof.js} +1 -1
  13. package/dist/{dataset-JDyjG3QR.js → dataset-U3PrjwgU.js} +7 -7
  14. package/dist/{dropout-hpDwECTe.js → dropout-DPfPgWWe.js} +11 -11
  15. package/dist/{gather-D0_gPiBz.js → gather-Bbh8DHhM.js} +4 -4
  16. package/dist/{gelu-uyHP1x1f.js → gelu-BFwVnd1r.js} +1 -1
  17. package/dist/{gpgpu_math-DJm3ZTAf.js → gpgpu_math-DffelNS-.js} +2 -2
  18. package/dist/{index-BPPzKVdR.js → index-DYD_yPa-.js} +1083 -1106
  19. package/dist/{index-C0dhsYom.js → index-UdZhlibC.js} +126 -126
  20. package/dist/{kernel_funcs_utils-CwRTFqrc.js → kernel_funcs_utils-CXDy3EN7.js} +3 -3
  21. package/dist/layers/BaseLayer.js +2 -2
  22. package/dist/layers/CausalSelfAttention.js +8 -8
  23. package/dist/layers/MLP.js +5 -5
  24. package/dist/layers/RMSNorm.js +3 -3
  25. package/dist/layers/RoPECache.js +4 -4
  26. package/dist/layers/TiedEmbedding.js +5 -5
  27. package/dist/layers/TransformerBlock.js +1 -1
  28. package/dist/loader/loadTransformers.js +1 -1
  29. package/dist/loader/oldZipLoad.js +11 -7
  30. package/dist/{log_sum_exp-D086OgZJ.js → log_sum_exp-BnmCkHWl.js} +8 -8
  31. package/dist/main.d.ts +11 -0
  32. package/dist/main.js +44 -27
  33. package/dist/{mat_mul-1nwdPkQ_.js → mat_mul-dwmZz69e.js} +1 -1
  34. package/dist/{max-BQc2Aj-I.js → max-ByjEGoFx.js} +3 -3
  35. package/dist/{mulmat_packed_gpu-Gzf3I9UV.js → mulmat_packed_gpu-IGPBp6h9.js} +1 -1
  36. package/dist/{ones-D63HpSF_.js → ones-C8Mfln6-.js} +2 -2
  37. package/dist/ops/adamAdjust.d.ts +2 -0
  38. package/dist/ops/adamAdjust.js +9 -0
  39. package/dist/ops/adamMoments.d.ts +2 -0
  40. package/dist/ops/adamMoments.js +9 -0
  41. package/dist/ops/appendCache.js +3 -3
  42. package/dist/ops/attentionMask.js +1 -1
  43. package/dist/ops/cpu/adamAdjust.d.ts +1 -0
  44. package/dist/ops/cpu/adamAdjust.js +18 -0
  45. package/dist/ops/cpu/adamMoments.d.ts +1 -0
  46. package/dist/ops/cpu/adamMoments.js +16 -0
  47. package/dist/ops/cpu/appendCache.js +2 -2
  48. package/dist/ops/cpu/attentionMask.js +5 -5
  49. package/dist/ops/cpu/fusedSoftmax.js +2 -2
  50. package/dist/ops/cpu/gatherSub.js +3 -3
  51. package/dist/ops/cpu/gelu.js +1 -1
  52. package/dist/ops/cpu/matMulGelu.js +2 -2
  53. package/dist/ops/cpu/matMulMul.js +1 -1
  54. package/dist/ops/cpu/mulDropout.js +1 -1
  55. package/dist/ops/cpu/normRMS.js +1 -1
  56. package/dist/ops/cpu/qkv.js +3 -3
  57. package/dist/ops/cpu/rope.js +5 -5
  58. package/dist/ops/cpu/scatterSub.js +11 -11
  59. package/dist/ops/fusedSoftmax.js +1 -1
  60. package/dist/ops/gatherSub.js +1 -1
  61. package/dist/ops/gelu.js +2 -2
  62. package/dist/ops/grads/attentionMask.js +1 -1
  63. package/dist/ops/grads/fusedSoftmax.js +2 -2
  64. package/dist/ops/grads/gelu.js +2 -2
  65. package/dist/ops/grads/matMulGelu.js +1 -1
  66. package/dist/ops/grads/normRMS.js +1 -1
  67. package/dist/ops/grads/qkv.js +1 -1
  68. package/dist/ops/grads/rope.js +1 -1
  69. package/dist/ops/matMulGelu.js +1 -1
  70. package/dist/ops/matMulMul.js +1 -1
  71. package/dist/ops/mulDrop.js +1 -1
  72. package/dist/ops/normRMS.js +1 -1
  73. package/dist/ops/qkv.js +1 -1
  74. package/dist/ops/rope.js +4 -4
  75. package/dist/ops/scatterSub.js +1 -1
  76. package/dist/ops/webgl/adamAdjust.d.ts +1 -0
  77. package/dist/ops/webgl/adamAdjust.js +50 -0
  78. package/dist/ops/webgl/adamMoments.d.ts +1 -0
  79. package/dist/ops/webgl/adamMoments.js +38 -0
  80. package/dist/ops/webgl/appendCache.js +1 -1
  81. package/dist/ops/webgl/attentionMask.js +1 -1
  82. package/dist/ops/webgl/fusedSoftmax.js +4 -4
  83. package/dist/ops/webgl/gatherSub.js +8 -8
  84. package/dist/ops/webgl/gelu.js +2 -2
  85. package/dist/ops/webgl/log.js +3 -3
  86. package/dist/ops/webgl/matMulGelu.js +4 -4
  87. package/dist/ops/webgl/matMulMul.js +1 -1
  88. package/dist/ops/webgl/mulDropout.js +1 -1
  89. package/dist/ops/webgl/normRMS.js +2 -2
  90. package/dist/ops/webgl/qkv.js +1 -1
  91. package/dist/ops/webgl/rope.js +1 -1
  92. package/dist/ops/webgl/scatterSub.js +1 -1
  93. package/dist/ops/webgpu/adamAdjust.d.ts +1 -0
  94. package/dist/ops/webgpu/adamAdjust.js +52 -0
  95. package/dist/ops/webgpu/adamMoments.d.ts +1 -0
  96. package/dist/ops/webgpu/adamMoments.js +51 -0
  97. package/dist/ops/webgpu/appendCache.js +13 -12
  98. package/dist/ops/webgpu/attentionMask.js +11 -10
  99. package/dist/ops/webgpu/gatherSub.js +26 -11
  100. package/dist/ops/webgpu/gelu.js +7 -6
  101. package/dist/ops/webgpu/index.js +3 -0
  102. package/dist/ops/webgpu/normRMS.js +27 -101
  103. package/dist/ops/webgpu/normRMSGrad.d.ts +1 -0
  104. package/dist/ops/webgpu/normRMSGrad.js +128 -0
  105. package/dist/ops/webgpu/qkv.js +9 -8
  106. package/dist/ops/webgpu/rope.js +8 -7
  107. package/dist/ops/webgpu/scatterSub.js +8 -7
  108. package/dist/ops/webgpu/utils/reductions.d.ts +9 -0
  109. package/dist/ops/webgpu/utils/reductions.js +68 -0
  110. package/dist/{ops-CIQLNshk.js → ops-aRTXR2Sr.js} +195 -219
  111. package/dist/{random_width-DkYP8W8N.js → random_width-DbSpgl4o.js} +22 -21
  112. package/dist/{range-CYzpQY53.js → range-D9CZhVlR.js} +1 -1
  113. package/dist/{reciprocal-_A9yv27J.js → reciprocal-CGB48wZB.js} +1 -1
  114. package/dist/{register_all_kernels-guvSxp7M.js → register_all_kernels-DnbAyBXt.js} +30 -29
  115. package/dist/{reshape-BMUzc1UY.js → reshape-BR0eoLYN.js} +3 -3
  116. package/dist/{scatter_nd_util-IRBqKz_b.js → scatter_nd_util-OjyAxku2.js} +1 -1
  117. package/dist/{selu_util-Dt_iuXaq.js → selu_util-Ce6pu9IM.js} +41 -41
  118. package/dist/{shared-CDu9S76h.js → shared-Czipaeb6.js} +6 -6
  119. package/dist/{shared-BNa2q6jD.js → shared-DS5waSIY.js} +1 -1
  120. package/dist/{sin-Cocju-BY.js → sin-CiBxrDqX.js} +6 -6
  121. package/dist/slice-BHbDHObE.js +28 -0
  122. package/dist/{softmax-GPNK3o-U.js → softmax-JMEIUo2J.js} +3 -3
  123. package/dist/{split-CHzJjxDv.js → split-CRU0PjVV.js} +1 -1
  124. package/dist/{stack-Dpgg_1W1.js → stack-ikk2Y8_P.js} +1 -1
  125. package/dist/{sum-B8wEpKsg.js → sum-NLYbiDag.js} +3 -3
  126. package/dist/{tensor-RvZVNmg0.js → tensor-Do9PKbIE.js} +1 -1
  127. package/dist/{tensor2d-B_kyod7_.js → tensor2d-CWHxHpLh.js} +1 -1
  128. package/dist/training/Adam.d.ts +22 -0
  129. package/dist/training/Adam.js +93 -0
  130. package/dist/training/AdamExt.d.ts +1 -1
  131. package/dist/training/AdamExt.js +13 -12
  132. package/dist/training/DatasetBuilder.js +2 -2
  133. package/dist/training/FullTrainer.js +22 -22
  134. package/dist/training/Trainer.d.ts +1 -1
  135. package/dist/training/Trainer.js +32 -32
  136. package/dist/training/sparseCrossEntropy.d.ts +0 -4
  137. package/dist/training/sparseCrossEntropy.js +7 -7
  138. package/dist/utilities/arrayClose.d.ts +1 -0
  139. package/dist/utilities/arrayClose.js +11 -0
  140. package/dist/utilities/dummy.js +2 -2
  141. package/dist/utilities/generate.js +3 -3
  142. package/dist/utilities/multinomialCPU.js +2 -2
  143. package/dist/utilities/performance.d.ts +1 -1
  144. package/dist/utilities/performance.js +11 -11
  145. package/dist/utilities/profile.js +1 -1
  146. package/dist/utilities/safetensors.js +2 -2
  147. package/dist/utilities/weights.js +2 -2
  148. package/dist/{variable-DXEUOwew.js → variable-BTBkayv_.js} +1 -1
  149. package/dist/{webgpu_util-g13LvDIv.js → webgpu_program-WaoMq-WD.js} +138 -215
  150. package/dist/webgpu_util-DhSeP4b6.js +80 -0
  151. package/dist/{zeros-DCPCdFGq.js → zeros-DnPT2nD4.js} +4 -4
  152. package/package.json +1 -1
@@ -1,24 +1,25 @@
1
- import { x as T, y as I, E as O, bE as Oa, bF as Ma, bG as Ci, l as b, bH as Ii, Q as L, bI as Di, bJ as $i, bK as Ti, bL as zi, h as Ei, bM as Li, bN as Fi, bO as Oi, bP as Mi, bQ as _a, bR as _i, bS as Ra, bT as Ri, bU as Ba, bV as Bi, F as Ge, k as kt, bq as Wa, bW as Wi, bX as Gi, bY as Pi, q as Pe, c as V, a as w, bZ as Ga, b_ as Ui, b$ as Vi, c0 as ji, p as ce, aU as pt, bv as Pa, c1 as Ki, c2 as Hi, c3 as qi, c4 as Ji, c5 as Zi, bx as Xi, c6 as Yi, c7 as Qi, C as Ua, ai as Va, ao as ja, c8 as tr, c9 as Ka, a6 as z, ca as Ps, cb as Ha, cc as qa, j as pn, cd as Us, ce as Ja, cf as Za, cg as Xa, A as Ya, ch as Qa, ci as tl, cj as el, bf as nl, ck as sl, aT as he, b as et, aj as U, cl as il, bm as rl, au as ht, cm as ol, z as Q, cn as al, co as ll, cp as ul, cq as cl, cr as hl, cs as pl, ct as dl, cu as fl, H as ml, cv as gl, bj as bl, bp as yl, cw as wl, M as kl, cx as xl, a3 as Nl, cy as vl, cz as Al, cA as Sl, aq as Cl, cB as Il, a7 as Dl, aV as $l, br as Tl, ah as zl, bs as El, G as Ll, aX as Fl, ak as Ol, cC as Ml, cD as _l, cE as Rl, al as Bl, a8 as Wl, cF as Gl, cG as Pl, cH as Ul, N as Vl, bt as jl, cI as Kl, cJ as Hl, aR as ql, b1 as Jl, cK as Zl, cL as Xl, bu as Yl, a$ as Ql, P as tu, cM as eu, n as rs, an as nu, bw as su, ax as iu, O as ru, as as ou, ar as au, I as lu, bc as uu, cN as cu, bd as hu, cO as pu, b3 as du, aP as fu, ap as mu, cP as gu, a4 as bu, aN as yu, S as wu, J as ku, bz as xu, cQ as Nu, bA as vu, at as Au, bC as Su, U as Cu, cR as Iu, L as Du, b5 as $u, b4 as Tu, cS as Oe, cT as zu, i as Eu, ag as Vs, cU as Lu, t as x, aS as $e, cV as S, cW as He, cX as qe, ab as Vt, d as J, ac as Fu, cY as js, o as Jt, K as Ou, T as Te, cZ as Mu, c_ as _u, m as Ks, c$ as Ru, d0 as Hs, d1 as Bu } from "./index-C0dhsYom.js";
2
- import { k as ke, c as Nt, o as ze, s as er, b as nr, d as Wu, m as sr, t as In, l as ir, v as os, a as Gu, S as Pu, p as Uu, w as as, x as rr, y as Je, z as Vu, A as ju } from "./selu_util-Dt_iuXaq.js";
3
- import { s as Dt, n as wt, w as ne, a as Ze, g as Xe, b as ls, t as K, c as Ce, d as Xt, e as Ku, u as dn, f as ye, h as fn, i as Hu, l as qu, j as us, m as or, k as qt, o as Ju } from "./ops-CIQLNshk.js";
4
- import { r as N } from "./reshape-BMUzc1UY.js";
5
- import { s as W } from "./sum-B8wEpKsg.js";
6
- import { m as ct } from "./mat_mul-1nwdPkQ_.js";
7
- import { s as Qt } from "./split-CHzJjxDv.js";
8
- import { s as Zu, c as ar } from "./sin-Cocju-BY.js";
9
- import { e as Hn, g as lr, h as cs, c as Xu } from "./axis_util-DaAl5MER.js";
10
- import { a as se, e as ie, l as Yu } from "./log_sum_exp-D086OgZJ.js";
11
- import { s as Dn } from "./stack-Dpgg_1W1.js";
12
- import { o as xe } from "./ones-D63HpSF_.js";
13
- import { M as Qu, f as ur, r as tc, d as ec, a as $n } from "./dropout-hpDwECTe.js";
14
- import { z as vt } from "./zeros-DCPCdFGq.js";
15
- import { c as pe } from "./concat-CsHeR4zV.js";
16
- import { g as cr } from "./gather-D0_gPiBz.js";
17
- import { s as hr } from "./softmax-GPNK3o-U.js";
18
- import { m as Ee } from "./max-BQc2Aj-I.js";
19
- import { t as nc } from "./tensor-RvZVNmg0.js";
20
- import { r as sc } from "./range-CYzpQY53.js";
21
- import { v as ic } from "./variable-DXEUOwew.js";
1
+ import { B as T, C as I, E as O, bE as Oa, bF as Ma, bG as Ci, l as b, bH as Ii, X as L, bI as Di, bJ as $i, bK as Ti, bL as zi, h as Ei, bM as Li, bN as Fi, bO as Oi, bP as Mi, bQ as _a, bR as _i, bS as Ra, bT as Ri, bU as Ba, bV as Bi, I as Ge, k as kt, bq as Wa, bW as Wi, bX as Gi, bY as Pi, y as Pe, c as V, a as w, bZ as Ga, b_ as Ui, b$ as Vi, c0 as ji, p as ce, aU as pt, bv as Pa, c1 as Ki, c2 as Hi, c3 as qi, c4 as Ji, c5 as Zi, bx as Xi, c6 as Yi, c7 as Qi, F as Ua, al as Va, aq as ja, c8 as tr, c9 as Ka, o as z, ca as Ps, cb as Ha, cc as qa, j as pn, cd as Us, ce as Ja, cf as Za, cg as Xa, ch as Ya, ci as Qa, cj as tl, ck as el, bf as nl, cl as sl, q as he, b as et, n as U, cm as il, bm as rl, aw as ht, cn as ol, z as Q, co as al, cp as ll, cq as ul, cr as cl, cs as hl, ct as pl, cu as dl, cv as fl, J as ml, cw as gl, bj as bl, bp as yl, cx as wl, Q as kl, cy as xl, a8 as Nl, cz as vl, cA as Al, cB as Sl, as as Cl, cC as Il, ae as Dl, aV as $l, br as Tl, ak as zl, bs as El, G as Ll, aX as Fl, am as Ol, cD as Ml, cE as _l, cF as Rl, an as Bl, af as Wl, cG as Gl, cH as Pl, cI as Ul, U as Vl, bt as jl, cJ as Kl, cK as Hl, aS as ql, b1 as Jl, cL as Zl, cM as Xl, bu as Yl, a$ as Ql, P as tu, cN as eu, x as rs, ap as nu, bw as su, az as iu, V as ru, au as ou, at as au, K as lu, bc as uu, cO as cu, bd as hu, cP as pu, b3 as du, aQ as fu, ar as mu, cQ as gu, a9 as bu, W as yu, S as wu, L as ku, bz as xu, cR as Nu, bA as vu, av as Au, bC as Su, Y as Cu, cS as Iu, N as Du, b5 as $u, b4 as Tu, cT as Oe, cU as zu, i as Eu, aj as Vs, cV as Lu, t as x, aT as $e, cW as S, cX as He, cY as qe, ab as Vt, d as J, ac as Fu, cZ as js, w as Jt, M as Ou, T as Te, O as Mu, c_ as _u, m as Ks, c$ as Ru, d0 as Hs, d1 as Bu } from "./index-UdZhlibC.js";
2
+ import { k as ke, c as Nt, o as ze, s as er, b as nr, d as Wu, m as sr, t as In, l as ir, v as os, a as Gu, S as Pu, p as Uu, w as as, x as rr, y as Je, z as Vu, A as ju } from "./selu_util-Ce6pu9IM.js";
3
+ import { n as wt, w as ne, a as Ze, g as Xe, b as ls, t as K, c as Ce, d as Xt, e as Ku, u as dn, f as ye, h as fn, i as Hu, l as qu, s as us, m as or, j as qt, k as Ju } from "./ops-aRTXR2Sr.js";
4
+ import { r as N } from "./reshape-BR0eoLYN.js";
5
+ import { s as W } from "./sum-NLYbiDag.js";
6
+ import { m as ct } from "./mat_mul-dwmZz69e.js";
7
+ import { s as Qt } from "./split-CRU0PjVV.js";
8
+ import { s as Zu, c as ar } from "./sin-CiBxrDqX.js";
9
+ import { e as Hn, g as lr, h as cs, c as Xu } from "./axis_util-5DTW2tFV.js";
10
+ import { a as se, e as ie, l as Yu } from "./log_sum_exp-BnmCkHWl.js";
11
+ import { s as Dn } from "./stack-ikk2Y8_P.js";
12
+ import { o as xe } from "./ones-C8Mfln6-.js";
13
+ import { s as Dt } from "./slice-BHbDHObE.js";
14
+ import { M as Qu, f as ur, r as tc, d as ec, a as $n } from "./dropout-DPfPgWWe.js";
15
+ import { z as vt } from "./zeros-DnPT2nD4.js";
16
+ import { c as pe } from "./concat-CbXTetof.js";
17
+ import { g as cr } from "./gather-Bbh8DHhM.js";
18
+ import { s as hr } from "./softmax-JMEIUo2J.js";
19
+ import { m as Ee } from "./max-ByjEGoFx.js";
20
+ import { t as nc } from "./tensor-Do9PKbIE.js";
21
+ import { r as sc } from "./range-D9CZhVlR.js";
22
+ import { v as ic } from "./variable-BTBkayv_.js";
22
23
  /**
23
24
  * @license
24
25
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { E as e, R as f } from "./index-C0dhsYom.js";
1
+ import { E as e, R as f } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { x as c, y as n, E as e, O as s } from "./index-C0dhsYom.js";
1
+ import { B as c, C as n, E as e, V as s } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -1,34 +1,35 @@
1
- import { x as oe, y as se, E as le, ck as ps, cl as hs, cn as en, co as fs, cp as ms, cr as gs, F as xs, cq as Cs, cs as bs, bp as To, l as Y, dd as tn, cy as sn, aj as Do, z as nn, br as Eo, cC as Is, cD as ys, cE as $s, ao as Nc, cH as on, q as vc, a as an, k as Ne, dl as ks, cK as rn, cL as ws, Q as It, bw as Fo, cP as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cR as vs, dv as dn, dw as pn, ai as xe, dx as S, aS as Sc, a6 as Rc, T as Ao, dy as Tc, cS as Dc, n as Ec, aT as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aM as Vt, d7 as Vo, aK as Wo, d4 as Ac, aJ as Wt, ag as Ke, d8 as Mo, aq as hn, al as _c, j as z, bg as Ge, an as Lc, ar as fn, as as mn, I as Vc, a9 as Wc, H as Bo, Z as ne, cd as Go, cm as zo, bE as Uo, p as me, a2 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, ct as jo, cu as Yo, bL as Qo, bK as Zo, aO as Jo, dc as ea, bM as gn, cw as ta, aY as sa, M as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cx as ca, bQ as la, a3 as xn, bR as Cn, b6 as ua, bS as da, u as nt, D as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cz as Kn, dz as mt, dA as Ca, cA as qn, df as Bc, U as Gc, aQ as ba, cB as Ia, bW as In, aV as ya, ax as $a, am as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, $ as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cG as Ea, N as zc, b_ as Fa, b$ as Pa, cI as Oa, cJ as Aa, dm as _a, c0 as La, aR as Va, bh as Wa, S as Ma, a5 as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cT as Xa, c2 as Ka, P as qa, a0 as ja, c3 as Ya, cM as wn, dB as Qa, dC as Za, dD as Ja, R as er, O as Nn, bc as tr, cN as sr, bd as nr, cO as or, c5 as ar, b8 as rr, b3 as vn, _ as ir, dn as cr, aP as lr, c6 as Sn, a4 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, J as gr, cQ as xr, at as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, L as $r, be as kr, av as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, aN as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aG as Rr, bq as ol, a7 as al, ah as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, a8 as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, aw as Tr, bx as Cl, ap as bl, bz as Il, bB as yl } from "./index-C0dhsYom.js";
2
- import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-DkYP8W8N.js";
3
- import { r as re } from "./reshape-BMUzc1UY.js";
4
- import { b as jl } from "./broadcast_to-C4v-j9yA.js";
5
- import { c as os } from "./concat-CsHeR4zV.js";
6
- import { c as Yl, s as Ql } from "./sin-Cocju-BY.js";
7
- import { w as Lr, p as Vr, n as ss, a as zs, r as Us, q as Wr, s as as, f as Zl, g as Jl, d as eu, b as tu, e as su, o as nu, m as ou, v as au, x as ru, y as iu, j as cu, c as lu, t as uu, u as du } from "./ops-CIQLNshk.js";
8
- import { m as Tt } from "./mat_mul-1nwdPkQ_.js";
9
- import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-Dt_iuXaq.js";
10
- import { e as $u, l as ku, a as wu } from "./log_sum_exp-D086OgZJ.js";
11
- import { f as Nu } from "./dropout-hpDwECTe.js";
12
- import { g as vu } from "./gather-D0_gPiBz.js";
13
- import { c as Hs, z as Hr } from "./zeros-DCPCdFGq.js";
14
- import { m as Su } from "./max-BQc2Aj-I.js";
15
- import { r as Ru } from "./reciprocal-_A9yv27J.js";
16
- import { s as Xs } from "./split-CHzJjxDv.js";
17
- import { s as Tu } from "./softmax-GPNK3o-U.js";
18
- import { s as Du } from "./stack-Dpgg_1W1.js";
19
- import { s as Eu } from "./sum-B8wEpKsg.js";
20
- import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-CDu9S76h.js";
21
- import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-DWiwsi2N.js";
22
- import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-DaAl5MER.js";
1
+ import { B as oe, C as se, E as le, cl as ps, cm as hs, co as en, cp as fs, cq as ms, cs as gs, I as xs, cr as Cs, ct as bs, bp as To, l as Y, dd as tn, cz as sn, n as Do, z as nn, br as Eo, cD as Is, cE as ys, cF as $s, aq as Nc, cI as on, y as vc, a as an, k as Ne, dl as ks, cL as rn, cM as ws, X as It, bw as Fo, cQ as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cS as vs, dv as dn, dw as pn, al as xe, dx as S, aT as Sc, o as Rc, T as Ao, dy as Tc, cT as Dc, x as Ec, q as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aO as Vt, d7 as Vo, aM as Wo, d4 as Ac, aL as Wt, aj as Ke, d8 as Mo, as as hn, an as _c, j as z, bg as Ge, ap as Lc, at as fn, au as mn, K as Vc, a2 as Wc, J as Bo, a1 as ne, cd as Go, cn as zo, bE as Uo, p as me, a7 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, cu as jo, cv as Yo, bL as Qo, bK as Zo, aP as Jo, dc as ea, bM as gn, cx as ta, aY as sa, Q as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cy as ca, bQ as la, a8 as xn, bR as Cn, b6 as ua, bS as da, u as nt, H as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cA as Kn, dz as mt, dA as Ca, cB as qn, df as Bc, Y as Gc, aR as ba, cC as Ia, bW as In, aV as ya, az as $a, ao as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, a4 as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cH as Ea, U as zc, b_ as Fa, b$ as Pa, cJ as Oa, cK as Aa, dm as _a, c0 as La, aS as Va, bh as Wa, S as Ma, ad as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cU as Xa, c2 as Ka, P as qa, a5 as ja, c3 as Ya, cN as wn, dB as Qa, dC as Za, dD as Ja, R as er, V as Nn, bc as tr, cO as sr, bd as nr, cP as or, c5 as ar, b8 as rr, b3 as vn, a3 as ir, dn as cr, aQ as lr, c6 as Sn, a9 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, L as gr, cR as xr, av as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, N as $r, be as kr, ax as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, W as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aI as Rr, bq as ol, ae as al, ak as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, af as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, ay as Tr, bx as Cl, ar as bl, bz as Il, bB as yl } from "./index-UdZhlibC.js";
2
+ import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-DbSpgl4o.js";
3
+ import { r as re } from "./reshape-BR0eoLYN.js";
4
+ import { b as jl } from "./broadcast_to-Ba9h_8DO.js";
5
+ import { c as os } from "./concat-CbXTetof.js";
6
+ import { c as Yl, s as Ql } from "./sin-CiBxrDqX.js";
7
+ import { w as Lr, o as Vr, n as ss, a as zs, r as Us, p as Wr, f as Zl, g as Jl, d as eu, b as tu, e as su, k as nu, m as ou, q as au, v as ru, x as iu, s as cu, c as lu, t as uu, u as du } from "./ops-aRTXR2Sr.js";
8
+ import { m as Tt } from "./mat_mul-dwmZz69e.js";
9
+ import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-Ce6pu9IM.js";
10
+ import { e as $u, l as ku, a as wu } from "./log_sum_exp-BnmCkHWl.js";
11
+ import { f as Nu } from "./dropout-DPfPgWWe.js";
12
+ import { g as vu } from "./gather-Bbh8DHhM.js";
13
+ import { c as Hs, z as Hr } from "./zeros-DnPT2nD4.js";
14
+ import { s as as } from "./slice-BHbDHObE.js";
15
+ import { m as Su } from "./max-ByjEGoFx.js";
16
+ import { r as Ru } from "./reciprocal-CGB48wZB.js";
17
+ import { s as Xs } from "./split-CRU0PjVV.js";
18
+ import { s as Tu } from "./softmax-JMEIUo2J.js";
19
+ import { s as Du } from "./stack-ikk2Y8_P.js";
20
+ import { s as Eu } from "./sum-NLYbiDag.js";
21
+ import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-Czipaeb6.js";
22
+ import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-C9Ut8n0Q.js";
23
+ import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-5DTW2tFV.js";
23
24
  import { s as zd } from "./index-C4L8Cm77.js";
24
25
  import { n as mi, b as gi, a as xi } from "./non_max_suppression_impl-CsEgBuMA.js";
25
- import { c as wt } from "./scatter_nd_util-IRBqKz_b.js";
26
- import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DJm3ZTAf.js";
27
- import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-BNa2q6jD.js";
28
- import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-CwRTFqrc.js";
29
- import { R as hh, r as H, a as fh } from "./Reshape-BLijOA8h.js";
30
- import { M as Ri } from "./mulmat_packed_gpu-Gzf3I9UV.js";
31
- import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-C4hOvYOZ.js";
26
+ import { c as wt } from "./scatter_nd_util-OjyAxku2.js";
27
+ import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DffelNS-.js";
28
+ import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-DS5waSIY.js";
29
+ import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-CXDy3EN7.js";
30
+ import { R as hh, r as H, a as fh } from "./Reshape-CEsEp0AI.js";
31
+ import { M as Ri } from "./mulmat_packed_gpu-IGPBp6h9.js";
32
+ import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-CVYNbZxu.js";
32
33
  /**
33
34
  * @license
34
35
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { x as o, y as t, E as a, I as p } from "./index-C0dhsYom.js";
1
+ import { B as o, C as t, E as a, K as p } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -19,7 +19,7 @@ function c(r, s) {
19
19
  const e = { x: t(r, "x", "reshape", "string_or_numeric") }, n = { shape: s };
20
20
  return a.runKernel(p, e, n);
21
21
  }
22
- const u = /* @__PURE__ */ o({ reshape_: c });
22
+ const x = /* @__PURE__ */ o({ reshape_: c });
23
23
  export {
24
- u as r
24
+ x as r
25
25
  };
@@ -1,4 +1,4 @@
1
- import { j as p, Z as w } from "./index-C0dhsYom.js";
1
+ import { j as p, a1 as w } from "./index-UdZhlibC.js";
2
2
  function k(o, t, r) {
3
3
  const n = t.rank > 1 ? t.shape[t.rank - 1] : 1, e = t.rank > 1 ? t.rank - 1 : 1, h = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${r.shape}, indices.shape: ${t.shape}, shape: ${o}, sliceDim: ${n}, and batchDim: ${e}.`;
4
4
  if (r.rank < e)
@@ -1,6 +1,6 @@
1
- import { l as N, ao as H, x as _, y as S, E as L, ap as te, aq as ne, al as se, an as re, ar as ie, as as oe, at as le, a as ue, au as fe, av as O } from "./index-C0dhsYom.js";
2
- import { r as ce } from "./reshape-BMUzc1UY.js";
3
- import { s as ae } from "./sum-B8wEpKsg.js";
1
+ import { l as N, aq as H, B as _, C as S, E as L, ar as te, as as ne, an as se, ap as re, at as ie, au as oe, av as le, a as ue, aw as fe, ax as O } from "./index-UdZhlibC.js";
2
+ import { r as ce } from "./reshape-BR0eoLYN.js";
3
+ import { s as ae } from "./sum-NLYbiDag.js";
4
4
  /**
5
5
  * @license
6
6
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -17,11 +17,11 @@ import { s as ae } from "./sum-B8wEpKsg.js";
17
17
  * limitations under the License.
18
18
  * =============================================================================
19
19
  */
20
- function Xe(e, t, n, s, r = "NHWC", u) {
20
+ function Be(e, t, n, s, r = "NHWC", u) {
21
21
  const i = e[3], o = [...t, i], c = Ae(r);
22
- return X(e, o, n, u, s, null, null, c);
22
+ return B(e, o, n, u, s, null, null, c);
23
23
  }
24
- function Be(e, t, n, s, r, u, i = "channelsLast") {
24
+ function Xe(e, t, n, s, r, u, i = "channelsLast") {
25
25
  const [o, c] = T(t);
26
26
  let a;
27
27
  if (i === "channelsLast")
@@ -30,7 +30,7 @@ function Be(e, t, n, s, r, u, i = "channelsLast") {
30
30
  a = [o, c, e[1], e[1]];
31
31
  else
32
32
  throw new Error(`Unknown dataFormat ${i}`);
33
- return X(e, a, n, s, r, u, !1, i);
33
+ return B(e, a, n, s, r, u, !1, i);
34
34
  }
35
35
  function je(e, t, n, s, r, u, i = "NDHWC") {
36
36
  const [o, c, a] = W(t);
@@ -43,7 +43,7 @@ function je(e, t, n, s, r, u, i = "NDHWC") {
43
43
  throw new Error(`Unknown dataFormat ${i}`);
44
44
  return he(e, p, n, s, r, !1, g, u);
45
45
  }
46
- function X(e, t, n, s, r, u, i = !1, o = "channelsLast") {
46
+ function B(e, t, n, s, r, u, i = !1, o = "channelsLast") {
47
47
  let [c, a, p, g] = [-1, -1, -1, -1];
48
48
  if (o === "channelsLast")
49
49
  [c, a, p, g] = e;
@@ -51,7 +51,7 @@ function X(e, t, n, s, r, u, i = !1, o = "channelsLast") {
51
51
  [c, g, a, p] = e;
52
52
  else
53
53
  throw new Error(`Unknown dataFormat ${o}`);
54
- const [l, h, , d] = t, [A, m] = T(n), [b, k] = T(s), f = G(l, b), E = G(h, k), { padInfo: w, outHeight: I, outWidth: x } = de(r, a, p, A, m, f, E, u, o), y = i ? d * g : d;
54
+ const [l, h, , d] = t, [A, m] = T(n), [b, k] = T(s), f = G(l, b), w = G(h, k), { padInfo: E, outHeight: I, outWidth: x } = de(r, a, p, A, m, f, w, u, o), y = i ? d * g : d;
55
55
  let $;
56
56
  return o === "channelsFirst" ? $ = [c, y, I, x] : o === "channelsLast" && ($ = [c, I, x, y]), {
57
57
  batchSize: c,
@@ -62,13 +62,13 @@ function X(e, t, n, s, r, u, i = !1, o = "channelsLast") {
62
62
  outHeight: I,
63
63
  outWidth: x,
64
64
  outChannels: y,
65
- padInfo: w,
65
+ padInfo: E,
66
66
  strideHeight: A,
67
67
  strideWidth: m,
68
68
  filterHeight: l,
69
69
  filterWidth: h,
70
70
  effectiveFilterHeight: f,
71
- effectiveFilterWidth: E,
71
+ effectiveFilterWidth: w,
72
72
  dilationHeight: b,
73
73
  dilationWidth: k,
74
74
  inShape: e,
@@ -84,7 +84,7 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
84
84
  [c, l, a, p, g] = e;
85
85
  else
86
86
  throw new Error(`Unknown dataFormat ${i}`);
87
- const [h, d, A, , m] = t, [b, k, f] = W(n), [E, w, I] = W(s), x = G(h, E), y = G(d, w), $ = G(A, I), { padInfo: C, outDepth: M, outHeight: D, outWidth: F } = me(r, a, p, g, b, k, f, x, y, $, o), V = u ? m * l : m;
87
+ const [h, d, A, , m] = t, [b, k, f] = W(n), [w, E, I] = W(s), x = G(h, w), y = G(d, E), $ = G(A, I), { padInfo: C, outDepth: M, outHeight: D, outWidth: F } = me(r, a, p, g, b, k, f, x, y, $, o), V = u ? m * l : m;
88
88
  let U;
89
89
  return i === "channelsFirst" ? U = [c, V, M, D, F] : i === "channelsLast" && (U = [c, M, D, F, V]), {
90
90
  batchSize: c,
@@ -107,8 +107,8 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
107
107
  effectiveFilterDepth: x,
108
108
  effectiveFilterHeight: y,
109
109
  effectiveFilterWidth: $,
110
- dilationDepth: E,
111
- dilationHeight: w,
110
+ dilationDepth: w,
111
+ dilationHeight: E,
112
112
  dilationWidth: I,
113
113
  inShape: e,
114
114
  outShape: U,
@@ -116,18 +116,18 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
116
116
  };
117
117
  }
118
118
  function ge(e, t, n, s, r) {
119
- s == null && (s = B(e, t, n));
119
+ s == null && (s = X(e, t, n));
120
120
  const u = e[0], i = e[1], o = R((u - t + 2 * s) / n + 1, r), c = R((i - t + 2 * s) / n + 1, r);
121
121
  return [o, c];
122
122
  }
123
123
  function pe(e, t, n, s, r, u) {
124
- r == null && (r = B(e, t[0], s[0]));
124
+ r == null && (r = X(e, t[0], s[0]));
125
125
  const i = [0, 0, 0, n];
126
126
  for (let o = 0; o < 3; o++)
127
127
  e[o] + 2 * r >= t[o] && (i[o] = R((e[o] - t[o] + 2 * r) / s[o] + 1, u));
128
128
  return i;
129
129
  }
130
- function B(e, t, n, s = 1) {
130
+ function X(e, t, n, s = 1) {
131
131
  const r = G(t, s);
132
132
  return Math.floor((e[0] * (n - 1) - n + r) / 2);
133
133
  }
@@ -175,8 +175,8 @@ function me(e, t, n, s, r, u, i, o, c, a, p) {
175
175
  l = m[0], h = m[1], d = m[2];
176
176
  } else if (e === "same") {
177
177
  l = Math.ceil(t / r), h = Math.ceil(n / u), d = Math.ceil(s / i);
178
- const A = (l - 1) * r + o - t, m = (h - 1) * u + c - n, b = (d - 1) * i + a - s, k = Math.floor(A / 2), f = A - k, E = Math.floor(m / 2), w = m - E, I = Math.floor(b / 2), x = b - I;
179
- g = { top: E, bottom: w, left: I, right: x, front: k, back: f, type: "SAME" };
178
+ const A = (l - 1) * r + o - t, m = (h - 1) * u + c - n, b = (d - 1) * i + a - s, k = Math.floor(A / 2), f = A - k, w = Math.floor(m / 2), E = m - w, I = Math.floor(b / 2), x = b - I;
179
+ g = { top: w, bottom: E, left: I, right: x, front: k, back: f, type: "SAME" };
180
180
  } else
181
181
  throw Error(`Unknown padding parameter: ${e}`);
182
182
  return { padInfo: g, outDepth: l, outHeight: h, outWidth: d };
@@ -244,11 +244,11 @@ function Qe(e, t, n) {
244
244
  * limitations under the License.
245
245
  * =============================================================================
246
246
  */
247
- function Ee(e) {
247
+ function we(e) {
248
248
  const n = { x: S(e, "x", "sigmoid", "float32") };
249
249
  return L.runKernel(te, n);
250
250
  }
251
- const we = /* @__PURE__ */ _({ sigmoid_: Ee });
251
+ const Ee = /* @__PURE__ */ _({ sigmoid_: we });
252
252
  /**
253
253
  * @license
254
254
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -417,7 +417,7 @@ function ze(e, t, n, s) {
417
417
  if (t === "leakyrelu")
418
418
  return xe(e, s);
419
419
  if (t === "sigmoid")
420
- return we(e);
420
+ return Ee(e);
421
421
  throw new Error(`Unknown fused activation ${t}.`);
422
422
  }
423
423
  const et = (e, t) => !(e > 0) || t === "linear";
@@ -600,43 +600,43 @@ function We(e, t, n, s, r, u, i, o, c) {
600
600
  for (let f = 0; f < e.length; ++f) {
601
601
  if (l.strides[f] === 0)
602
602
  throw Error(`strides[${f}] must be non-zero`);
603
- const E = !!(l.shrinkAxisMask & 1 << f), w = e[f];
604
- if (w === -1) {
605
- m.push(E ? 1 : -1);
603
+ const w = !!(l.shrinkAxisMask & 1 << f), E = e[f];
604
+ if (E === -1) {
605
+ m.push(w ? 1 : -1);
606
606
  continue;
607
607
  }
608
608
  const I = [l.beginMask & 1 << f, l.endMask & 1 << f], x = [
609
609
  l.strides[f] > 0 ? 0 : -1,
610
- l.strides[f] > 0 ? w : w - 1
610
+ l.strides[f] > 0 ? E : E - 1
611
611
  ];
612
- if (E && l.strides[f] <= 0)
612
+ if (w && l.strides[f] <= 0)
613
613
  throw Error("only stride 1 allowed on non-range indexing.");
614
614
  A = A && l.strides[f] === 1;
615
615
  const y = !!(l.beginMask & 1 << f && l.endMask & 1 << f);
616
616
  if (l.beginValid && l.endValid) {
617
- if (E) {
618
- const D = l.begin[f] < 0 ? w + l.begin[f] : l.begin[f];
619
- if (l.begin[f] = D, l.end[f] = l.begin[f] + 1, D < 0 || D >= w)
617
+ if (w) {
618
+ const D = l.begin[f] < 0 ? E + l.begin[f] : l.begin[f];
619
+ if (l.begin[f] = D, l.end[f] = l.begin[f] + 1, D < 0 || D >= E)
620
620
  throw Error(`slice index ${l.begin[f]} of dimension ${f} out of bounds.`);
621
621
  } else
622
- l.begin[f] = K(l.begin[f], 0, l.strides[f], w, I, x), l.end[f] = K(l.end[f], 1, l.strides[f], w, I, x);
623
- const M = l.strides[f] === 1 && l.begin[f] === 0 && l.end[f] === w;
622
+ l.begin[f] = K(l.begin[f], 0, l.strides[f], E, I, x), l.end[f] = K(l.end[f], 1, l.strides[f], E, I, x);
623
+ const M = l.strides[f] === 1 && l.begin[f] === 0 && l.end[f] === E;
624
624
  h = h && M, d = d && (f === 0 && l.strides[f] === 1 || M);
625
625
  } else
626
626
  h = h && l.strides[f] === 1 && y, d = d && (f === 0 && l.strides[f] === 1 || y);
627
627
  let $, C = !1;
628
- if (l.beginValid && l.endValid ? ($ = l.end[f] - l.begin[f], C = !0) : E ? ($ = 1, C = !0) : y && w >= 0 && (l.strides[f] < 0 ? $ = -w : $ = w, C = !0), C) {
628
+ if (l.beginValid && l.endValid ? ($ = l.end[f] - l.begin[f], C = !0) : w ? ($ = 1, C = !0) : y && E >= 0 && (l.strides[f] < 0 ? $ = -E : $ = E, C = !0), C) {
629
629
  let M;
630
630
  $ === 0 || $ < 0 != l.strides[f] < 0 ? M = 0 : M = Math.trunc($ / l.strides[f]) + ($ % l.strides[f] !== 0 ? 1 : 0), m.push(M);
631
631
  } else
632
632
  m.push(-1);
633
633
  }
634
634
  for (let f = 0; f < l.finalShapeGatherIndices.length; ++f) {
635
- const E = l.finalShapeGatherIndices[f];
636
- E >= 0 ? b.push(m[E]) : E === v && b.push(1);
635
+ const w = l.finalShapeGatherIndices[f];
636
+ w >= 0 ? b.push(m[w]) : w === v && b.push(1);
637
637
  }
638
638
  return {
639
- finalShapeSparse: b.filter((f, E) => l.finalShapeGatherIndices[E] !== v),
639
+ finalShapeSparse: b.filter((f, w) => l.finalShapeGatherIndices[w] !== v),
640
640
  finalShape: b,
641
641
  isIdentity: h,
642
642
  sliceDim0: d,
@@ -714,11 +714,11 @@ export {
714
714
  nt as a,
715
715
  ze as b,
716
716
  Qe as c,
717
- X as d,
717
+ B as d,
718
718
  he as e,
719
- B as f,
720
- Xe as g,
721
- Be as h,
719
+ X as f,
720
+ Be as g,
721
+ Xe as h,
722
722
  je as i,
723
723
  Ae as j,
724
724
  qe as k,
@@ -733,7 +733,7 @@ export {
733
733
  P as t,
734
734
  Ve as u,
735
735
  Le as v,
736
- we as w,
736
+ Ee as w,
737
737
  $e as x,
738
738
  Se as y,
739
739
  xe as z
@@ -1,9 +1,9 @@
1
- import { l as Lt, bf as Ce, j as V, k as Pt, Z as H, bg as et, aa as ht, bh as dt, bi as at, a1 as De, a2 as st, ak as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, ai as K, bn as At, bo as W, bp as _t, bq as Vt, a7 as Ct, br as Dt, ah as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, a8 as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, am as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ap as Yt, aN as Je, bz as te, bA as ee, bB as ne, aK as pt, bC as se, bD as ot } from "./index-C0dhsYom.js";
2
- import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-DWiwsi2N.js";
3
- import { g as pn, a as wn, e as In, c as bn } from "./axis_util-DaAl5MER.js";
4
- import { b as xn } from "./broadcast_to-C4v-j9yA.js";
5
- import { r as En } from "./reshape-BMUzc1UY.js";
6
- import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-Dt_iuXaq.js";
1
+ import { l as Lt, bf as Ce, j as V, k as Pt, a1 as H, bg as et, aa as ht, bh as dt, bi as at, a6 as De, a7 as st, am as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, al as K, bn as At, bo as W, bp as _t, bq as Vt, ae as Ct, br as Dt, ak as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, af as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, ao as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ar as Yt, W as Je, bz as te, bA as ee, bB as ne, aM as pt, bC as se, bD as ot } from "./index-UdZhlibC.js";
2
+ import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-C9Ut8n0Q.js";
3
+ import { g as pn, a as wn, e as In, c as bn } from "./axis_util-5DTW2tFV.js";
4
+ import { b as xn } from "./broadcast_to-Ba9h_8DO.js";
5
+ import { r as En } from "./reshape-BR0eoLYN.js";
6
+ import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-Ce6pu9IM.js";
7
7
  import { g as vn } from "./_commonjsHelpers-ByX85dGu.js";
8
8
  function Sn(e, t) {
9
9
  for (var n = 0; n < t.length; n++) {
@@ -1,4 +1,4 @@
1
- import { s as l } from "./shared-CDu9S76h.js";
1
+ import { s as l } from "./shared-Czipaeb6.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { x as o, y as t, E as c, a3 as a, a4 as x } from "./index-C0dhsYom.js";
1
+ import { B as o, C as t, E as c, a8 as a, a9 as e } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -15,11 +15,11 @@ import { x as o, y as t, E as c, a3 as a, a4 as x } from "./index-C0dhsYom.js";
15
15
  * limitations under the License.
16
16
  * =============================================================================
17
17
  */
18
- function e(n) {
18
+ function i(n) {
19
19
  const s = { x: t(n, "x", "cos", "float32") };
20
20
  return c.runKernel(a, s);
21
21
  }
22
- const f = /* @__PURE__ */ o({ cos_: e });
22
+ const f = /* @__PURE__ */ o({ cos_: i });
23
23
  /**
24
24
  * @license
25
25
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -36,11 +36,11 @@ const f = /* @__PURE__ */ o({ cos_: e });
36
36
  * limitations under the License.
37
37
  * =============================================================================
38
38
  */
39
- function i(n) {
39
+ function x(n) {
40
40
  const s = { x: t(n, "x", "sin", "float32") };
41
- return c.runKernel(x, s);
41
+ return c.runKernel(e, s);
42
42
  }
43
- const p = /* @__PURE__ */ o({ sin_: i });
43
+ const p = /* @__PURE__ */ o({ sin_: x });
44
44
  export {
45
45
  f as c,
46
46
  p as s
@@ -0,0 +1,28 @@
1
+ import { B as c, C as e, E as a, W as l } from "./index-UdZhlibC.js";
2
+ /**
3
+ * @license
4
+ * Copyright 2018 Google LLC. All Rights Reserved.
5
+ * Licensed under the Apache License, Version 2.0 (the "License");
6
+ * you may not use this file except in compliance with the License.
7
+ * You may obtain a copy of the License at
8
+ *
9
+ * http://www.apache.org/licenses/LICENSE-2.0
10
+ *
11
+ * Unless required by applicable law or agreed to in writing, software
12
+ * distributed under the License is distributed on an "AS IS" BASIS,
13
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
+ * See the License for the specific language governing permissions and
15
+ * limitations under the License.
16
+ * =============================================================================
17
+ */
18
+ function p(r, n, o) {
19
+ const s = e(r, "x", "slice", "string_or_numeric");
20
+ if (s.rank === 0)
21
+ throw new Error("Slicing scalar is not possible");
22
+ const t = { x: s }, i = { begin: n, size: o };
23
+ return a.runKernel(l, t, i);
24
+ }
25
+ const x = /* @__PURE__ */ c({ slice_: p });
26
+ export {
27
+ x as s
28
+ };
@@ -1,4 +1,4 @@
1
- import { x as r, y as f, E as e, S as i } from "./index-C0dhsYom.js";
1
+ import { B as r, C as f, E as e, S as i } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -22,7 +22,7 @@ function l(s, o = -1) {
22
22
  const n = { logits: t }, a = { dim: o };
23
23
  return e.runKernel(i, n, a);
24
24
  }
25
- const c = /* @__PURE__ */ r({ softmax_: l });
25
+ const p = /* @__PURE__ */ r({ softmax_: l });
26
26
  export {
27
- c as s
27
+ p as s
28
28
  };
@@ -1,4 +1,4 @@
1
- import { x as p, y as i, E as a, J as c } from "./index-C0dhsYom.js";
1
+ import { B as p, C as i, E as a, L as c } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { x as e, B as c, l as n, E as k, P as i } from "./index-C0dhsYom.js";
1
+ import { B as e, D as c, l as n, E as k, P as i } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { x as e, y as u, Q as c, E as l, U as m } from "./index-C0dhsYom.js";
1
+ import { B as e, C as u, X as c, E as l, Y as m } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -21,7 +21,7 @@ function i(t, n = null, o = !1) {
21
21
  const r = { x: s }, a = { axis: n, keepDims: o };
22
22
  return l.runKernel(m, r, a);
23
23
  }
24
- const x = /* @__PURE__ */ e({ sum_: i });
24
+ const f = /* @__PURE__ */ e({ sum_: i });
25
25
  export {
26
- x as s
26
+ f as s
27
27
  };
@@ -1,4 +1,4 @@
1
- import { V as t, W as a } from "./index-C0dhsYom.js";
1
+ import { Z as t, _ as a } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -1,4 +1,4 @@
1
- import { X as t, V as s, W as a } from "./index-C0dhsYom.js";
1
+ import { $ as t, Z as s, _ as a } from "./index-UdZhlibC.js";
2
2
  /**
3
3
  * @license
4
4
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -0,0 +1,22 @@
1
+ import { Optimizer } from '@tensorflow/tfjs-core';
2
+ import { ConfigDict, Serializable, SerializableConstructor } from '@tensorflow/tfjs-core/dist/serialization';
3
+ import { NamedTensor, NamedVariableMap } from '@tensorflow/tfjs-core/dist/tensor_types';
4
+ export declare class AdamOptimizer extends Optimizer {
5
+ protected learningRate: number;
6
+ protected beta1: number;
7
+ protected beta2: number;
8
+ protected epsilon: number | null;
9
+ /** @nocollapse */
10
+ static get className(): string;
11
+ private accBeta1;
12
+ private accBeta2;
13
+ private accumulatedMoments;
14
+ constructor(learningRate: number, beta1: number, beta2: number, epsilon?: number | null);
15
+ applyGradients(variableGradients: NamedVariableMap | NamedTensor[]): void;
16
+ dispose(): void;
17
+ getWeights(): Promise<NamedTensor[]>;
18
+ setWeights(weightValues: NamedTensor[]): Promise<void>;
19
+ getConfig(): ConfigDict;
20
+ /** @nocollapse */
21
+ static fromConfig<T extends Serializable>(cls: SerializableConstructor<T>, config: ConfigDict): T;
22
+ }