@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.
- package/dist/Generator.js +13 -9
- package/dist/NanoGPTModel.js +10 -10
- package/dist/{RealDiv-C4hOvYOZ.js → RealDiv-CVYNbZxu.js} +11 -11
- package/dist/{Reshape-BLijOA8h.js → Reshape-CEsEp0AI.js} +2 -2
- package/dist/Reshape-Do18N3gO.js +30 -0
- package/dist/TeachableLLM.js +9 -5
- package/dist/{TiedEmbedding-BLltddza.js → TiedEmbedding-ccLBFiZi.js} +4 -4
- package/dist/{axis_util-DaAl5MER.js → axis_util-5DTW2tFV.js} +1 -1
- package/dist/backend.js +2 -2
- package/dist/{backend_util-DWiwsi2N.js → backend_util-C9Ut8n0Q.js} +40 -40
- package/dist/{broadcast_to-C4v-j9yA.js → broadcast_to-Ba9h_8DO.js} +2 -2
- package/dist/{concat-CsHeR4zV.js → concat-CbXTetof.js} +1 -1
- package/dist/{dataset-JDyjG3QR.js → dataset-U3PrjwgU.js} +7 -7
- package/dist/{dropout-hpDwECTe.js → dropout-DPfPgWWe.js} +11 -11
- package/dist/{gather-D0_gPiBz.js → gather-Bbh8DHhM.js} +4 -4
- package/dist/{gelu-uyHP1x1f.js → gelu-BFwVnd1r.js} +1 -1
- package/dist/{gpgpu_math-DJm3ZTAf.js → gpgpu_math-DffelNS-.js} +2 -2
- package/dist/{index-BPPzKVdR.js → index-DYD_yPa-.js} +1083 -1106
- package/dist/{index-C0dhsYom.js → index-UdZhlibC.js} +126 -126
- package/dist/{kernel_funcs_utils-CwRTFqrc.js → kernel_funcs_utils-CXDy3EN7.js} +3 -3
- package/dist/layers/BaseLayer.js +2 -2
- package/dist/layers/CausalSelfAttention.js +8 -8
- package/dist/layers/MLP.js +5 -5
- package/dist/layers/RMSNorm.js +3 -3
- package/dist/layers/RoPECache.js +4 -4
- package/dist/layers/TiedEmbedding.js +5 -5
- package/dist/layers/TransformerBlock.js +1 -1
- package/dist/loader/loadTransformers.js +1 -1
- package/dist/loader/oldZipLoad.js +11 -7
- package/dist/{log_sum_exp-D086OgZJ.js → log_sum_exp-BnmCkHWl.js} +8 -8
- package/dist/main.d.ts +11 -0
- package/dist/main.js +44 -27
- package/dist/{mat_mul-1nwdPkQ_.js → mat_mul-dwmZz69e.js} +1 -1
- package/dist/{max-BQc2Aj-I.js → max-ByjEGoFx.js} +3 -3
- package/dist/{mulmat_packed_gpu-Gzf3I9UV.js → mulmat_packed_gpu-IGPBp6h9.js} +1 -1
- package/dist/{ones-D63HpSF_.js → ones-C8Mfln6-.js} +2 -2
- package/dist/ops/adamAdjust.d.ts +2 -0
- package/dist/ops/adamAdjust.js +9 -0
- package/dist/ops/adamMoments.d.ts +2 -0
- package/dist/ops/adamMoments.js +9 -0
- package/dist/ops/appendCache.js +3 -3
- package/dist/ops/attentionMask.js +1 -1
- package/dist/ops/cpu/adamAdjust.d.ts +1 -0
- package/dist/ops/cpu/adamAdjust.js +18 -0
- package/dist/ops/cpu/adamMoments.d.ts +1 -0
- package/dist/ops/cpu/adamMoments.js +16 -0
- package/dist/ops/cpu/appendCache.js +2 -2
- package/dist/ops/cpu/attentionMask.js +5 -5
- package/dist/ops/cpu/fusedSoftmax.js +2 -2
- package/dist/ops/cpu/gatherSub.js +3 -3
- package/dist/ops/cpu/gelu.js +1 -1
- package/dist/ops/cpu/matMulGelu.js +2 -2
- package/dist/ops/cpu/matMulMul.js +1 -1
- package/dist/ops/cpu/mulDropout.js +1 -1
- package/dist/ops/cpu/normRMS.js +1 -1
- package/dist/ops/cpu/qkv.js +3 -3
- package/dist/ops/cpu/rope.js +5 -5
- package/dist/ops/cpu/scatterSub.js +11 -11
- package/dist/ops/fusedSoftmax.js +1 -1
- package/dist/ops/gatherSub.js +1 -1
- package/dist/ops/gelu.js +2 -2
- package/dist/ops/grads/attentionMask.js +1 -1
- package/dist/ops/grads/fusedSoftmax.js +2 -2
- package/dist/ops/grads/gelu.js +2 -2
- package/dist/ops/grads/matMulGelu.js +1 -1
- package/dist/ops/grads/normRMS.js +1 -1
- package/dist/ops/grads/qkv.js +1 -1
- package/dist/ops/grads/rope.js +1 -1
- package/dist/ops/matMulGelu.js +1 -1
- package/dist/ops/matMulMul.js +1 -1
- package/dist/ops/mulDrop.js +1 -1
- package/dist/ops/normRMS.js +1 -1
- package/dist/ops/qkv.js +1 -1
- package/dist/ops/rope.js +4 -4
- package/dist/ops/scatterSub.js +1 -1
- package/dist/ops/webgl/adamAdjust.d.ts +1 -0
- package/dist/ops/webgl/adamAdjust.js +50 -0
- package/dist/ops/webgl/adamMoments.d.ts +1 -0
- package/dist/ops/webgl/adamMoments.js +38 -0
- package/dist/ops/webgl/appendCache.js +1 -1
- package/dist/ops/webgl/attentionMask.js +1 -1
- package/dist/ops/webgl/fusedSoftmax.js +4 -4
- package/dist/ops/webgl/gatherSub.js +8 -8
- package/dist/ops/webgl/gelu.js +2 -2
- package/dist/ops/webgl/log.js +3 -3
- package/dist/ops/webgl/matMulGelu.js +4 -4
- package/dist/ops/webgl/matMulMul.js +1 -1
- package/dist/ops/webgl/mulDropout.js +1 -1
- package/dist/ops/webgl/normRMS.js +2 -2
- package/dist/ops/webgl/qkv.js +1 -1
- package/dist/ops/webgl/rope.js +1 -1
- package/dist/ops/webgl/scatterSub.js +1 -1
- package/dist/ops/webgpu/adamAdjust.d.ts +1 -0
- package/dist/ops/webgpu/adamAdjust.js +52 -0
- package/dist/ops/webgpu/adamMoments.d.ts +1 -0
- package/dist/ops/webgpu/adamMoments.js +51 -0
- package/dist/ops/webgpu/appendCache.js +13 -12
- package/dist/ops/webgpu/attentionMask.js +11 -10
- package/dist/ops/webgpu/gatherSub.js +26 -11
- package/dist/ops/webgpu/gelu.js +7 -6
- package/dist/ops/webgpu/index.js +3 -0
- package/dist/ops/webgpu/normRMS.js +27 -101
- package/dist/ops/webgpu/normRMSGrad.d.ts +1 -0
- package/dist/ops/webgpu/normRMSGrad.js +128 -0
- package/dist/ops/webgpu/qkv.js +9 -8
- package/dist/ops/webgpu/rope.js +8 -7
- package/dist/ops/webgpu/scatterSub.js +8 -7
- package/dist/ops/webgpu/utils/reductions.d.ts +9 -0
- package/dist/ops/webgpu/utils/reductions.js +68 -0
- package/dist/{ops-CIQLNshk.js → ops-aRTXR2Sr.js} +195 -219
- package/dist/{random_width-DkYP8W8N.js → random_width-DbSpgl4o.js} +22 -21
- package/dist/{range-CYzpQY53.js → range-D9CZhVlR.js} +1 -1
- package/dist/{reciprocal-_A9yv27J.js → reciprocal-CGB48wZB.js} +1 -1
- package/dist/{register_all_kernels-guvSxp7M.js → register_all_kernels-DnbAyBXt.js} +30 -29
- package/dist/{reshape-BMUzc1UY.js → reshape-BR0eoLYN.js} +3 -3
- package/dist/{scatter_nd_util-IRBqKz_b.js → scatter_nd_util-OjyAxku2.js} +1 -1
- package/dist/{selu_util-Dt_iuXaq.js → selu_util-Ce6pu9IM.js} +41 -41
- package/dist/{shared-CDu9S76h.js → shared-Czipaeb6.js} +6 -6
- package/dist/{shared-BNa2q6jD.js → shared-DS5waSIY.js} +1 -1
- package/dist/{sin-Cocju-BY.js → sin-CiBxrDqX.js} +6 -6
- package/dist/slice-BHbDHObE.js +28 -0
- package/dist/{softmax-GPNK3o-U.js → softmax-JMEIUo2J.js} +3 -3
- package/dist/{split-CHzJjxDv.js → split-CRU0PjVV.js} +1 -1
- package/dist/{stack-Dpgg_1W1.js → stack-ikk2Y8_P.js} +1 -1
- package/dist/{sum-B8wEpKsg.js → sum-NLYbiDag.js} +3 -3
- package/dist/{tensor-RvZVNmg0.js → tensor-Do9PKbIE.js} +1 -1
- package/dist/{tensor2d-B_kyod7_.js → tensor2d-CWHxHpLh.js} +1 -1
- package/dist/training/Adam.d.ts +22 -0
- package/dist/training/Adam.js +93 -0
- package/dist/training/AdamExt.d.ts +1 -1
- package/dist/training/AdamExt.js +13 -12
- package/dist/training/DatasetBuilder.js +2 -2
- package/dist/training/FullTrainer.js +22 -22
- package/dist/training/Trainer.d.ts +1 -1
- package/dist/training/Trainer.js +32 -32
- package/dist/training/sparseCrossEntropy.d.ts +0 -4
- package/dist/training/sparseCrossEntropy.js +7 -7
- package/dist/utilities/arrayClose.d.ts +1 -0
- package/dist/utilities/arrayClose.js +11 -0
- package/dist/utilities/dummy.js +2 -2
- package/dist/utilities/generate.js +3 -3
- package/dist/utilities/multinomialCPU.js +2 -2
- package/dist/utilities/performance.d.ts +1 -1
- package/dist/utilities/performance.js +11 -11
- package/dist/utilities/profile.js +1 -1
- package/dist/utilities/safetensors.js +2 -2
- package/dist/utilities/weights.js +2 -2
- package/dist/{variable-DXEUOwew.js → variable-BTBkayv_.js} +1 -1
- package/dist/{webgpu_util-g13LvDIv.js → webgpu_program-WaoMq-WD.js} +138 -215
- package/dist/webgpu_util-DhSeP4b6.js +80 -0
- package/dist/{zeros-DCPCdFGq.js → zeros-DnPT2nD4.js} +4 -4
- package/package.json +1 -1
|
@@ -1,24 +1,25 @@
|
|
|
1
|
-
import {
|
|
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-
|
|
3
|
-
import {
|
|
4
|
-
import { r as N } from "./reshape-
|
|
5
|
-
import { s as W } from "./sum-
|
|
6
|
-
import { m as ct } from "./mat_mul-
|
|
7
|
-
import { s as Qt } from "./split-
|
|
8
|
-
import { s as Zu, c as ar } from "./sin-
|
|
9
|
-
import { e as Hn, g as lr, h as cs, c as Xu } from "./axis_util-
|
|
10
|
-
import { a as se, e as ie, l as Yu } from "./log_sum_exp-
|
|
11
|
-
import { s as Dn } from "./stack-
|
|
12
|
-
import { o as xe } from "./ones-
|
|
13
|
-
import {
|
|
14
|
-
import {
|
|
15
|
-
import {
|
|
16
|
-
import {
|
|
17
|
-
import {
|
|
18
|
-
import {
|
|
19
|
-
import {
|
|
20
|
-
import {
|
|
21
|
-
import {
|
|
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,34 +1,35 @@
|
|
|
1
|
-
import {
|
|
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-
|
|
3
|
-
import { r as re } from "./reshape-
|
|
4
|
-
import { b as jl } from "./broadcast_to-
|
|
5
|
-
import { c as os } from "./concat-
|
|
6
|
-
import { c as Yl, s as Ql } from "./sin-
|
|
7
|
-
import { w as Lr,
|
|
8
|
-
import { m as Tt } from "./mat_mul-
|
|
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-
|
|
10
|
-
import { e as $u, l as ku, a as wu } from "./log_sum_exp-
|
|
11
|
-
import { f as Nu } from "./dropout-
|
|
12
|
-
import { g as vu } from "./gather-
|
|
13
|
-
import { c as Hs, z as Hr } from "./zeros-
|
|
14
|
-
import {
|
|
15
|
-
import {
|
|
16
|
-
import {
|
|
17
|
-
import { s as
|
|
18
|
-
import { s as
|
|
19
|
-
import { s as
|
|
20
|
-
import {
|
|
21
|
-
import { m as
|
|
22
|
-
import {
|
|
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-
|
|
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-
|
|
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-
|
|
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-
|
|
29
|
-
import { R as hh, r as H, a as fh } from "./Reshape-
|
|
30
|
-
import { M as Ri } from "./mulmat_packed_gpu-
|
|
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-
|
|
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 {
|
|
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
|
|
22
|
+
const x = /* @__PURE__ */ o({ reshape_: c });
|
|
23
23
|
export {
|
|
24
|
-
|
|
24
|
+
x as r
|
|
25
25
|
};
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { j as p,
|
|
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,
|
|
2
|
-
import { r as ce } from "./reshape-
|
|
3
|
-
import { s as ae } from "./sum-
|
|
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
|
|
20
|
+
function Be(e, t, n, s, r = "NHWC", u) {
|
|
21
21
|
const i = e[3], o = [...t, i], c = Ae(r);
|
|
22
|
-
return
|
|
22
|
+
return B(e, o, n, u, s, null, null, c);
|
|
23
23
|
}
|
|
24
|
-
function
|
|
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
|
|
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
|
|
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),
|
|
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:
|
|
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:
|
|
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), [
|
|
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:
|
|
111
|
-
dilationHeight:
|
|
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 =
|
|
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 =
|
|
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
|
|
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,
|
|
179
|
-
g = { top:
|
|
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
|
|
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
|
|
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
|
|
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
|
|
604
|
-
if (
|
|
605
|
-
m.push(
|
|
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 ?
|
|
610
|
+
l.strides[f] > 0 ? E : E - 1
|
|
611
611
|
];
|
|
612
|
-
if (
|
|
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 (
|
|
618
|
-
const D = l.begin[f] < 0 ?
|
|
619
|
-
if (l.begin[f] = D, l.end[f] = l.begin[f] + 1, D < 0 || D >=
|
|
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],
|
|
623
|
-
const M = l.strides[f] === 1 && l.begin[f] === 0 && l.end[f] ===
|
|
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) :
|
|
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
|
|
636
|
-
|
|
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,
|
|
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
|
-
|
|
717
|
+
B as d,
|
|
718
718
|
he as e,
|
|
719
|
-
|
|
720
|
-
|
|
721
|
-
|
|
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
|
-
|
|
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,
|
|
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-
|
|
3
|
-
import { g as pn, a as wn, e as In, c as bn } from "./axis_util-
|
|
4
|
-
import { b as xn } from "./broadcast_to-
|
|
5
|
-
import { r as En } from "./reshape-
|
|
6
|
-
import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-
|
|
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 {
|
|
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
|
|
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_:
|
|
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
|
|
39
|
+
function x(n) {
|
|
40
40
|
const s = { x: t(n, "x", "sin", "float32") };
|
|
41
|
-
return c.runKernel(
|
|
41
|
+
return c.runKernel(e, s);
|
|
42
42
|
}
|
|
43
|
-
const p = /* @__PURE__ */ o({ sin_:
|
|
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 {
|
|
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
|
|
25
|
+
const p = /* @__PURE__ */ r({ softmax_: l });
|
|
26
26
|
export {
|
|
27
|
-
|
|
27
|
+
p as s
|
|
28
28
|
};
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
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
|
|
24
|
+
const f = /* @__PURE__ */ e({ sum_: i });
|
|
25
25
|
export {
|
|
26
|
-
|
|
26
|
+
f as s
|
|
27
27
|
};
|
|
@@ -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
|
+
}
|