@genai-fi/nanogpt 0.6.1 → 0.6.2
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 +6 -6
- package/dist/NanoGPTModel.js +9 -9
- package/dist/{RealDiv-7xu-pkZN.js → RealDiv-BYViZwhN.js} +6 -6
- package/dist/{Reshape-BYC1oUku.js → Reshape-t7Kcikjk.js} +2 -2
- package/dist/TeachableLLM.js +5 -5
- package/dist/{TiedEmbedding-C1HBot-5.js → TiedEmbedding-9WeDwvjO.js} +4 -4
- package/dist/{axis_util-CCNL7jea.js → axis_util-Bu4h7XWV.js} +1 -1
- package/dist/{broadcast_to-CddAF879.js → broadcast_to-DARN-DBD.js} +2 -2
- package/dist/{concat-XOK9ANZu.js → concat-5aPGqw3Z.js} +8 -8
- package/dist/{dataset-BFFipD1c.js → dataset-pgqp-YfL.js} +5 -5
- package/dist/{dropout-xlKRoJyU.js → dropout-Bciw46HT.js} +10 -10
- package/dist/{gather-DKtUaTtA.js → gather-DjyCjmOD.js} +1 -1
- package/dist/{gpgpu_math-B_ycgZ4W.js → gpgpu_math-CNslybmD.js} +31 -31
- package/dist/{index-CamYe_M8.js → index-BAzbokzv.js} +31 -31
- package/dist/{kernel_funcs_utils-D5MS0JFg.js → kernel_funcs_utils-CUxJCg0g.js} +5 -5
- package/dist/layers/BaseLayer.js +2 -2
- package/dist/layers/CausalSelfAttention.js +6 -6
- 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/load.d.ts +13 -0
- package/dist/loader/load.js +27 -0
- package/dist/loader/loadHF.d.ts +7 -0
- package/dist/loader/loadHF.js +22 -0
- package/dist/{utilities/load.d.ts → loader/loadTransformers.d.ts} +11 -11
- package/dist/loader/loadTransformers.js +28 -0
- package/dist/loader/newZipLoad.d.ts +8 -0
- package/dist/loader/newZipLoad.js +21 -0
- package/dist/loader/oldZipLoad.d.ts +7 -0
- package/dist/loader/oldZipLoad.js +76 -0
- package/dist/{log_sum_exp-CV_5-TTu.js → log_sum_exp-YEo2h3gb.js} +16 -16
- package/dist/main.js +4 -4
- package/dist/{mat_mul-CAbRFWUj.js → mat_mul-7121rsJk.js} +4 -4
- package/dist/{max-JBBv7aUf.js → max-DtlIuVeW.js} +3 -3
- package/dist/{mulmat_packed_gpu-DW4doKL_.js → mulmat_packed_gpu-D4nKF7Je.js} +1 -1
- package/dist/{norm-B9dQTFYn.js → norm-CzltS9Fz.js} +10 -10
- package/dist/{ones-CMHNqMr6.js → ones-BBlSRqn1.js} +2 -2
- package/dist/ops/appendCache.js +3 -3
- package/dist/ops/attentionMask.js +1 -1
- package/dist/ops/cpu/appendCache.js +2 -2
- package/dist/ops/cpu/attentionMask.js +6 -6
- package/dist/ops/cpu/fusedSoftmax.js +2 -2
- package/dist/ops/cpu/gatherSub.js +9 -9
- package/dist/ops/cpu/gelu.js +1 -1
- package/dist/ops/cpu/matMulGelu.js +1 -1
- 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 +14 -14
- package/dist/ops/fusedSoftmax.js +1 -1
- package/dist/ops/gatherSub.js +1 -1
- package/dist/ops/gelu.js +1 -1
- package/dist/ops/grads/attentionMask.js +1 -1
- package/dist/ops/grads/fusedSoftmax.js +4 -4
- package/dist/ops/grads/gelu.js +1 -1
- 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/node/sparseCrossEntropy.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/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 +1 -1
- package/dist/ops/webgl/gelu.js +2 -2
- package/dist/ops/webgl/log.js +3 -3
- package/dist/ops/webgl/matMulGelu.js +17 -17
- 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-DqtYemmV.js → ops-C0sQEcPw.js} +78 -78
- package/dist/{random_width-CLMQG5Jn.js → random_width-DWzaOgrn.js} +22 -22
- package/dist/{range-DqYjKnuG.js → range-DYsrnfiy.js} +1 -1
- package/dist/{reciprocal-z49filta.js → reciprocal-CJQeasVa.js} +1 -1
- package/dist/{register_all_kernels-COt6wLD0.js → register_all_kernels-BfFCQAqs.js} +28 -28
- package/dist/{reshape-C45vIIRU.js → reshape-krWGKraP.js} +1 -1
- package/dist/{scatter_nd_util-qgtnviTE.js → scatter_nd_util-93ln7Hut.js} +3 -3
- package/dist/{selu_util-4QV_GXTB.js → selu_util-sntGesxr.js} +41 -41
- package/dist/{shared-ByfrGA97.js → shared-Ca6iDobD.js} +6 -6
- package/dist/{sin-9JBrfVaB.js → sin-D_h-qCSx.js} +1 -1
- package/dist/{softmax-DvMvui-_.js → softmax-fsdtf6JC.js} +1 -1
- package/dist/{split-DxrHrPFK.js → split-eiktj-6L.js} +4 -4
- package/dist/{stack-DgaoDmnF.js → stack-dfEEz2OY.js} +2 -2
- package/dist/{sum-BpcpxNEh.js → sum-BE_Irnim.js} +3 -3
- package/dist/{tensor-CDz5x1mP.js → tensor-Xyi595sG.js} +1 -1
- package/dist/{tensor2d-jO8JY5Jd.js → tensor2d-CPEkynbH.js} +1 -1
- package/dist/training/AdamExt.js +1 -1
- package/dist/training/DatasetBuilder.js +2 -2
- package/dist/training/FullTrainer.js +1 -1
- package/dist/training/Trainer.js +3 -3
- package/dist/training/sparseCrossEntropy.js +5 -5
- package/dist/utilities/dummy.js +2 -2
- package/dist/utilities/generate.js +3 -3
- package/dist/utilities/profile.js +1 -1
- package/dist/utilities/safetensors.js +2 -2
- package/dist/utilities/save.js +1 -1
- package/dist/utilities/weights.js +2 -2
- package/dist/{variable-CLVXjN7F.js → variable-wSS22xj5.js} +1 -1
- package/dist/{zeros-DUkkVccu.js → zeros-YJDE7oRb.js} +10 -10
- package/package.json +3 -3
- package/dist/utilities/load.js +0 -99
|
@@ -1,16 +1,16 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { r as _ } from "./reshape-
|
|
3
|
-
import { s as un } from "./split-
|
|
4
|
-
import { s as
|
|
5
|
-
import { b as X } from "./broadcast_to-
|
|
6
|
-
import { r as
|
|
7
|
-
import { t as Ln } from "./tensor-
|
|
8
|
-
import { s as Y } from "./stack-
|
|
9
|
-
import { c as Xn, z as Qn } from "./zeros-
|
|
10
|
-
import { n as ln } from "./norm-
|
|
11
|
-
import { c as Q } from "./concat-
|
|
12
|
-
import { m as T } from "./mat_mul-
|
|
13
|
-
import { t as
|
|
1
|
+
import { o as h, q as l, E as f, aO as hn, i as p, aP as fn, h as w, aQ as bn, aR as dn, aS as gn, I as $n, ah as xn, B as j, aT as kn, aU as yn, aV as In, aW as _n, aX as En, aY as Mn, aZ as Nn, L as V, a_ as An, a$ as Tn, b0 as Sn, aK as wn, V as zn, O as Bn, Q as qn, b1 as Kn, t as Rn, b2 as J, b3 as Gn, b4 as Vn, b5 as Dn, b6 as Pn, b7 as vn, b8 as jn, s as Fn, b9 as On, ba as Wn, a as k, a7 as Z, aj as R, f as Yn, b as S, bb as Cn, H as L, d as Jn } from "./index-BAzbokzv.js";
|
|
2
|
+
import { r as _ } from "./reshape-krWGKraP.js";
|
|
3
|
+
import { s as un } from "./split-eiktj-6L.js";
|
|
4
|
+
import { s as q } from "./sum-BE_Irnim.js";
|
|
5
|
+
import { b as X } from "./broadcast_to-DARN-DBD.js";
|
|
6
|
+
import { r as W } from "./range-DYsrnfiy.js";
|
|
7
|
+
import { t as Ln } from "./tensor-Xyi595sG.js";
|
|
8
|
+
import { s as Y } from "./stack-dfEEz2OY.js";
|
|
9
|
+
import { c as Xn, z as Qn } from "./zeros-YJDE7oRb.js";
|
|
10
|
+
import { n as ln } from "./norm-CzltS9Fz.js";
|
|
11
|
+
import { c as Q } from "./concat-5aPGqw3Z.js";
|
|
12
|
+
import { m as T } from "./mat_mul-7121rsJk.js";
|
|
13
|
+
import { t as H } from "./tensor2d-CPEkynbH.js";
|
|
14
14
|
/**
|
|
15
15
|
* @license
|
|
16
16
|
* Copyright 2018 Google LLC. All Rights Reserved.
|
|
@@ -27,14 +27,14 @@ import { t as U } from "./tensor2d-jO8JY5Jd.js";
|
|
|
27
27
|
* limitations under the License.
|
|
28
28
|
* =============================================================================
|
|
29
29
|
*/
|
|
30
|
-
function
|
|
30
|
+
function Hn(r, t, n) {
|
|
31
31
|
const e = l(r, "x", "slice", "string_or_numeric");
|
|
32
32
|
if (e.rank === 0)
|
|
33
33
|
throw new Error("Slicing scalar is not possible");
|
|
34
34
|
const s = { x: e }, a = { begin: t, size: n };
|
|
35
35
|
return f.runKernel(hn, s, a);
|
|
36
36
|
}
|
|
37
|
-
const
|
|
37
|
+
const M = /* @__PURE__ */ h({ slice_: Hn });
|
|
38
38
|
/**
|
|
39
39
|
* @license
|
|
40
40
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -57,7 +57,7 @@ function Zn(r, t, n) {
|
|
|
57
57
|
const a = { x: e, weights: s }, o = { size: n };
|
|
58
58
|
return f.runKernel(fn, a, o);
|
|
59
59
|
}
|
|
60
|
-
const
|
|
60
|
+
const Un = /* @__PURE__ */ h({ bincount_: Zn });
|
|
61
61
|
/**
|
|
62
62
|
* @license
|
|
63
63
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -149,7 +149,7 @@ function te(r, t) {
|
|
|
149
149
|
const e = { x: n }, s = { reps: t };
|
|
150
150
|
return f.runKernel($n, e, s);
|
|
151
151
|
}
|
|
152
|
-
const
|
|
152
|
+
const O = /* @__PURE__ */ h({ tile_: te });
|
|
153
153
|
/**
|
|
154
154
|
* @license
|
|
155
155
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -175,11 +175,11 @@ function re(r, t, n, e = "float32") {
|
|
|
175
175
|
if (n == null)
|
|
176
176
|
return o;
|
|
177
177
|
if (n.length === 1)
|
|
178
|
-
return
|
|
178
|
+
return O(K(o, 0), [n[0], 1, 1]);
|
|
179
179
|
if (n.length === 2)
|
|
180
|
-
return
|
|
180
|
+
return O(K(K(o, 0), 0), [n[0], n[1], 1, 1]);
|
|
181
181
|
if (n.length === 3)
|
|
182
|
-
return
|
|
182
|
+
return O(K(K(K(o, 0), 0), 0), [
|
|
183
183
|
n[0],
|
|
184
184
|
n[1],
|
|
185
185
|
n[2],
|
|
@@ -211,7 +211,7 @@ function oe(r, t) {
|
|
|
211
211
|
const s = { a: n, b: e };
|
|
212
212
|
return f.runKernel(kn, s);
|
|
213
213
|
}
|
|
214
|
-
const
|
|
214
|
+
const U = /* @__PURE__ */ h({ greater_: oe });
|
|
215
215
|
/**
|
|
216
216
|
* @license
|
|
217
217
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -320,7 +320,7 @@ const pn = /* @__PURE__ */ h({ lessEqual_: me });
|
|
|
320
320
|
*/
|
|
321
321
|
function he(r) {
|
|
322
322
|
const n = { x: l(r, "x", "neg") };
|
|
323
|
-
return f.runKernel(
|
|
323
|
+
return f.runKernel(Mn, n);
|
|
324
324
|
}
|
|
325
325
|
const nn = /* @__PURE__ */ h({ neg_: he });
|
|
326
326
|
/**
|
|
@@ -343,7 +343,7 @@ function fe(r, t) {
|
|
|
343
343
|
const n = l(r, "a", "logicalAnd", "bool"), e = l(t, "b", "logicalAnd", "bool");
|
|
344
344
|
w(n.shape, e.shape);
|
|
345
345
|
const s = { a: n, b: e };
|
|
346
|
-
return f.runKernel(
|
|
346
|
+
return f.runKernel(Nn, s);
|
|
347
347
|
}
|
|
348
348
|
const be = /* @__PURE__ */ h({ logicalAnd_: fe });
|
|
349
349
|
/**
|
|
@@ -448,12 +448,12 @@ const Ie = /* @__PURE__ */ h({ squeeze_: ye });
|
|
|
448
448
|
* limitations under the License.
|
|
449
449
|
* =============================================================================
|
|
450
450
|
*/
|
|
451
|
-
function
|
|
451
|
+
function N(r, t) {
|
|
452
452
|
zn(r);
|
|
453
|
-
const n =
|
|
453
|
+
const n = Bn(r, t);
|
|
454
454
|
if (n.length !== 1)
|
|
455
455
|
throw new Error("tensor1d() requires values to be a flat/TypedArray");
|
|
456
|
-
return
|
|
456
|
+
return qn(r, null, n, t);
|
|
457
457
|
}
|
|
458
458
|
/**
|
|
459
459
|
* @license
|
|
@@ -523,13 +523,13 @@ const an = /* @__PURE__ */ h({ transpose_: Ee });
|
|
|
523
523
|
* limitations under the License.
|
|
524
524
|
* =============================================================================
|
|
525
525
|
*/
|
|
526
|
-
function
|
|
526
|
+
function Me(r, t, n, e, s = "bilinear", a = 0) {
|
|
527
527
|
const o = l(r, "image", "cropAndResize"), i = l(t, "boxes", "cropAndResize", "float32"), u = l(n, "boxInd", "cropAndResize", "int32"), c = i.shape[0];
|
|
528
528
|
p(o.rank === 4, () => `Error in cropAndResize: image must be rank 4,but got rank ${o.rank}.`), p(i.rank === 2 && i.shape[1] === 4, () => `Error in cropAndResize: boxes must be have size [${c},4] but had shape ${i.shape}.`), p(u.rank === 1 && u.shape[0] === c, () => `Error in cropAndResize: boxInd must be have size [${c}] but had shape ${i.shape}.`), p(e.length === 2, () => `Error in cropAndResize: cropSize must be of length 2, but got length ${e.length}.`), p(e[0] >= 1 && e[1] >= 1, () => `cropSize must be atleast [1,1], but was ${e}`), p(s === "bilinear" || s === "nearest", () => `method must be bilinear or nearest, but was ${s}`);
|
|
529
529
|
const b = { image: o, boxes: i, boxInd: u }, m = { method: s, extrapolationValue: a, cropSize: e };
|
|
530
530
|
return f.runKernel(Gn, b, m);
|
|
531
531
|
}
|
|
532
|
-
const
|
|
532
|
+
const Ne = /* @__PURE__ */ h({ cropAndResize_: Me });
|
|
533
533
|
/**
|
|
534
534
|
* @license
|
|
535
535
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -573,7 +573,7 @@ function Se(r) {
|
|
|
573
573
|
const t = l(r, "image", "grayscaleToRGB"), n = t.rank - 1, e = t.shape[n];
|
|
574
574
|
p(t.rank >= 2, () => `Error in grayscaleToRGB: images must be at least rank 2, but got rank ${t.rank}.`), p(e === 1, () => `Error in grayscaleToRGB: last dimension of a grayscale image should be size 1, but got size ${e}.`);
|
|
575
575
|
const s = new Array(t.rank);
|
|
576
|
-
return s.fill(1, 0, n), s[n] = 3,
|
|
576
|
+
return s.fill(1, 0, n), s[n] = 3, O(t, s);
|
|
577
577
|
}
|
|
578
578
|
const we = /* @__PURE__ */ h({ grayscaleToRGB_: Se });
|
|
579
579
|
/**
|
|
@@ -595,7 +595,7 @@ const we = /* @__PURE__ */ h({ grayscaleToRGB_: Se });
|
|
|
595
595
|
function ze(r) {
|
|
596
596
|
const t = l(r, "image", "RGBToGrayscale"), n = t.rank - 1, e = t.shape[n];
|
|
597
597
|
p(t.rank >= 2, () => `Error in RGBToGrayscale: images must be at least rank 2, but got rank ${t.rank}.`), p(e === 3, () => `Error in RGBToGrayscale: last dimension of an RGB image should be size 3, but got size ${e}.`);
|
|
598
|
-
const s = t.dtype, a = V(t, "float32"), o =
|
|
598
|
+
const s = t.dtype, a = V(t, "float32"), o = N([0.2989, 0.587, 0.114]);
|
|
599
599
|
let i;
|
|
600
600
|
switch (t.rank) {
|
|
601
601
|
case 2:
|
|
@@ -618,7 +618,7 @@ function ze(r) {
|
|
|
618
618
|
}
|
|
619
619
|
return i = K(i, -1), V(i, s);
|
|
620
620
|
}
|
|
621
|
-
const
|
|
621
|
+
const Be = /* @__PURE__ */ h({ rgbToGrayscale_: ze });
|
|
622
622
|
/**
|
|
623
623
|
* @license
|
|
624
624
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -635,13 +635,13 @@ const qe = /* @__PURE__ */ h({ rgbToGrayscale_: ze });
|
|
|
635
635
|
* limitations under the License.
|
|
636
636
|
* =============================================================================
|
|
637
637
|
*/
|
|
638
|
-
function
|
|
638
|
+
function qe(r, t, n = 0, e = 0.5) {
|
|
639
639
|
const s = l(r, "image", "rotateWithOffset", "float32");
|
|
640
640
|
p(s.rank === 4, () => `Error in rotateWithOffset: image must be rank 4,but got rank ${s.rank}.`);
|
|
641
641
|
const a = { image: s }, o = { radians: t, fillValue: n, center: e };
|
|
642
642
|
return f.runKernel(Dn, a, o);
|
|
643
643
|
}
|
|
644
|
-
const Ke = /* @__PURE__ */ h({ rotateWithOffset_:
|
|
644
|
+
const Ke = /* @__PURE__ */ h({ rotateWithOffset_: qe });
|
|
645
645
|
/**
|
|
646
646
|
* @license
|
|
647
647
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -762,7 +762,7 @@ function Fe(r, t, n, e, s, a) {
|
|
|
762
762
|
/* returnValidOutputs */
|
|
763
763
|
);
|
|
764
764
|
}
|
|
765
|
-
function
|
|
765
|
+
function Oe(r, t, n, e, s, a) {
|
|
766
766
|
return en(
|
|
767
767
|
r,
|
|
768
768
|
t,
|
|
@@ -785,8 +785,8 @@ function en(r, t, n, e, s, a, o = !1, i = !1, u = !1) {
|
|
|
785
785
|
if (E < s)
|
|
786
786
|
break;
|
|
787
787
|
let z = !1;
|
|
788
|
-
for (let
|
|
789
|
-
const P =
|
|
788
|
+
for (let B = m.length - 1; B >= A; --B) {
|
|
789
|
+
const P = We(r, I, m[B]);
|
|
790
790
|
if (P >= e) {
|
|
791
791
|
z = !0;
|
|
792
792
|
break;
|
|
@@ -801,7 +801,7 @@ function en(r, t, n, e, s, a, o = !1, i = !1, u = !1) {
|
|
|
801
801
|
const y = { selectedIndices: m };
|
|
802
802
|
return o && (y.selectedScores = $), u && (y.validOutputs = g), y;
|
|
803
803
|
}
|
|
804
|
-
function
|
|
804
|
+
function We(r, t, n) {
|
|
805
805
|
const e = r.subarray(t * 4, t * 4 + 4), s = r.subarray(n * 4, n * 4 + 4), a = Math.min(e[0], e[2]), o = Math.min(e[1], e[3]), i = Math.max(e[0], e[2]), u = Math.max(e[1], e[3]), c = Math.min(s[0], s[2]), b = Math.min(s[1], s[3]), m = Math.max(s[0], s[2]), $ = Math.max(s[1], s[3]), g = (i - a) * (u - o), x = (m - c) * ($ - b);
|
|
806
806
|
if (g <= 0 || x <= 0)
|
|
807
807
|
return 0;
|
|
@@ -835,7 +835,7 @@ async function Ce(r, t, n, e = 0.5, s = Number.NEGATIVE_INFINITY) {
|
|
|
835
835
|
const a = l(r, "boxes", "nonMaxSuppressionAsync"), o = l(t, "scores", "nonMaxSuppressionAsync"), i = D(a, o, n, e, s);
|
|
836
836
|
n = i.maxOutputSize, e = i.iouThreshold, s = i.scoreThreshold;
|
|
837
837
|
const u = await Promise.all([a.data(), o.data()]), c = u[0], b = u[1], { selectedIndices: m } = je(c, b, n, e, s);
|
|
838
|
-
return a !== r && a.dispose(), o !== t && o.dispose(),
|
|
838
|
+
return a !== r && a.dispose(), o !== t && o.dispose(), N(m, "int32");
|
|
839
839
|
}
|
|
840
840
|
const Je = Ce;
|
|
841
841
|
/**
|
|
@@ -880,13 +880,13 @@ const Xe = /* @__PURE__ */ h({ nonMaxSuppressionWithScore_: Le });
|
|
|
880
880
|
async function Qe(r, t, n, e = 0.5, s = Number.NEGATIVE_INFINITY, a = 0) {
|
|
881
881
|
const o = l(r, "boxes", "nonMaxSuppressionAsync"), i = l(t, "scores", "nonMaxSuppressionAsync"), u = D(o, i, n, e, s, a);
|
|
882
882
|
n = u.maxOutputSize, e = u.iouThreshold, s = u.scoreThreshold, a = u.softNmsSigma;
|
|
883
|
-
const c = await Promise.all([o.data(), i.data()]), b = c[0], m = c[1], { selectedIndices: $, selectedScores: g } =
|
|
883
|
+
const c = await Promise.all([o.data(), i.data()]), b = c[0], m = c[1], { selectedIndices: $, selectedScores: g } = Oe(b, m, n, e, s, a);
|
|
884
884
|
return o !== r && o.dispose(), i !== t && i.dispose(), {
|
|
885
|
-
selectedIndices:
|
|
886
|
-
selectedScores:
|
|
885
|
+
selectedIndices: N($, "int32"),
|
|
886
|
+
selectedScores: N(g)
|
|
887
887
|
};
|
|
888
888
|
}
|
|
889
|
-
const
|
|
889
|
+
const He = Qe;
|
|
890
890
|
/**
|
|
891
891
|
* @license
|
|
892
892
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -920,7 +920,7 @@ function Ze(r, t, n, e = 0.5, s = Number.NEGATIVE_INFINITY, a = !1) {
|
|
|
920
920
|
}, x = f.runKernel(jn, $, g);
|
|
921
921
|
return { selectedIndices: x[0], validOutputs: x[1] };
|
|
922
922
|
}
|
|
923
|
-
const
|
|
923
|
+
const Ue = /* @__PURE__ */ h({ nonMaxSuppressionPadded_: Ze });
|
|
924
924
|
/**
|
|
925
925
|
* @license
|
|
926
926
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -948,7 +948,7 @@ async function ns(r, t, n, e = 0.5, s = Number.NEGATIVE_INFINITY, a = !1) {
|
|
|
948
948
|
/* softNmsSigma */
|
|
949
949
|
), c = u.maxOutputSize, b = u.iouThreshold, m = u.scoreThreshold, [$, g] = await Promise.all([o.data(), i.data()]), { selectedIndices: x, validOutputs: y } = Fe($, g, c, b, m, a);
|
|
950
950
|
return o !== r && o.dispose(), i !== t && i.dispose(), {
|
|
951
|
-
selectedIndices:
|
|
951
|
+
selectedIndices: N(x, "int32"),
|
|
952
952
|
validOutputs: Fn(y, "int32")
|
|
953
953
|
};
|
|
954
954
|
}
|
|
@@ -974,7 +974,7 @@ function ss(r, t, n = !1, e = !1) {
|
|
|
974
974
|
p(s.rank === 3 || s.rank === 4, () => `Error in resizeBilinear: x must be rank 3 or 4, but got rank ${s.rank}.`), p(t.length === 2, () => `Error in resizeBilinear: new shape must 2D, but got shape ${t}.`), p(e === !1 || n === !1, () => "Error in resizeBilinear: If halfPixelCenters is true, alignCorners must be false.");
|
|
975
975
|
let a = s, o = !1;
|
|
976
976
|
s.rank === 3 && (o = !0, a = _(s, [1, s.shape[0], s.shape[1], s.shape[2]]));
|
|
977
|
-
const i = { images: a }, u = { alignCorners: n, halfPixelCenters: e, size: t }, c = f.runKernel(
|
|
977
|
+
const i = { images: a }, u = { alignCorners: n, halfPixelCenters: e, size: t }, c = f.runKernel(On, i, u);
|
|
978
978
|
return o ? _(c, [c.shape[1], c.shape[2], c.shape[3]]) : c;
|
|
979
979
|
}
|
|
980
980
|
const ts = /* @__PURE__ */ h({ resizeBilinear_: ss });
|
|
@@ -999,7 +999,7 @@ function rs(r, t, n = !1, e = !1) {
|
|
|
999
999
|
p(s.rank === 3 || s.rank === 4, () => `Error in resizeNearestNeighbor: x must be rank 3 or 4, but got rank ${s.rank}.`), p(t.length === 2, () => `Error in resizeNearestNeighbor: new shape must 2D, but got shape ${t}.`), p(s.dtype === "float32" || s.dtype === "int32", () => "`images` must have `int32` or `float32` as dtype"), p(e === !1 || n === !1, () => "Error in resizeNearestNeighbor: If halfPixelCenters is true, alignCorners must be false.");
|
|
1000
1000
|
let a = s, o = !1;
|
|
1001
1001
|
s.rank === 3 && (o = !0, a = _(s, [1, s.shape[0], s.shape[1], s.shape[2]]));
|
|
1002
|
-
const i = { images: a }, u = { alignCorners: n, halfPixelCenters: e, size: t }, c = f.runKernel(
|
|
1002
|
+
const i = { images: a }, u = { alignCorners: n, halfPixelCenters: e, size: t }, c = f.runKernel(Wn, i, u);
|
|
1003
1003
|
return o ? _(c, [c.shape[1], c.shape[2], c.shape[3]]) : c;
|
|
1004
1004
|
}
|
|
1005
1005
|
const as = /* @__PURE__ */ h({ resizeNearestNeighbor_: rs });
|
|
@@ -1021,7 +1021,7 @@ const as = /* @__PURE__ */ h({ resizeNearestNeighbor_: rs });
|
|
|
1021
1021
|
*/
|
|
1022
1022
|
function os(r, t = "binary", n = !1, e = 0.5) {
|
|
1023
1023
|
const s = l(r, "image", "threshold"), a = 0.2989, o = 0.587, i = 0.114, u = s.shape[0] * s.shape[1];
|
|
1024
|
-
let c = k(
|
|
1024
|
+
let c = k(N([e]), 255), b, m, $, g;
|
|
1025
1025
|
if (p(s.rank === 3, () => `Error in threshold: image must be rank 3,but got rank ${s.rank}.`), p(s.shape[2] === 3 || s.shape[2] === 1, () => `Error in threshold: image color channel must be equal to 3 or 1but got ${s.shape[2]}.`), p(s.dtype === "int32" || s.dtype === "float32", () => `Error in dtype: image dtype must be int32 or float32,but got dtype ${s.dtype}.`), p(t === "otsu" || t === "binary", () => `Method must be binary or otsu, but was ${t}`), s.shape[2] === 3) {
|
|
1026
1026
|
[b, m, $] = un(s, [1, 1, 1], -1);
|
|
1027
1027
|
const d = k(b, a), E = k(m, o), I = k($, i);
|
|
@@ -1029,24 +1029,24 @@ function os(r, t = "binary", n = !1, e = 0.5) {
|
|
|
1029
1029
|
} else
|
|
1030
1030
|
g = r;
|
|
1031
1031
|
if (t === "otsu") {
|
|
1032
|
-
const d =
|
|
1032
|
+
const d = Un(V(ke(g), "int32"), Ln([]), 256);
|
|
1033
1033
|
c = is(d, u);
|
|
1034
1034
|
}
|
|
1035
|
-
const x = n ? pn(g, c) :
|
|
1035
|
+
const x = n ? pn(g, c) : U(g, c);
|
|
1036
1036
|
return V(k(x, 255), "int32");
|
|
1037
1037
|
}
|
|
1038
1038
|
function is(r, t) {
|
|
1039
|
-
let n =
|
|
1039
|
+
let n = N([-1]), e = N([0]), s = N([0]), a, o, i, u, c, b;
|
|
1040
1040
|
for (let m = 0; m < r.size - 1; m++) {
|
|
1041
|
-
a =
|
|
1042
|
-
const $ =
|
|
1043
|
-
i = R($,
|
|
1044
|
-
const g = Yn(o.shape, a.size), x = Z(
|
|
1045
|
-
u = R(
|
|
1041
|
+
a = M(r, 0, m + 1), o = M(r, m + 1), c = R(q(a), t), b = R(q(o), t);
|
|
1042
|
+
const $ = q(k(a, W(0, a.size)));
|
|
1043
|
+
i = R($, q(a));
|
|
1044
|
+
const g = Yn(o.shape, a.size), x = Z(W(0, o.size), g), y = k(o, x);
|
|
1045
|
+
u = R(q(y), q(o));
|
|
1046
1046
|
const d = S(i, u), E = S(i, u), I = k(c, b);
|
|
1047
1047
|
s = k(k(I, d), E);
|
|
1048
|
-
const A =
|
|
1049
|
-
e = G(A, s, e), n = G(A,
|
|
1048
|
+
const A = U(s, e);
|
|
1049
|
+
e = G(A, s, e), n = G(A, N([m]), n);
|
|
1050
1050
|
}
|
|
1051
1051
|
return n;
|
|
1052
1052
|
}
|
|
@@ -1096,7 +1096,7 @@ function ps(r, t, n) {
|
|
|
1096
1096
|
const s = e.shape, [a, o] = e.shape.slice(-2);
|
|
1097
1097
|
let i, u;
|
|
1098
1098
|
typeof t == "number" ? (p(t % 1 === 0, () => `bandPart(): numLower must be an integer, got ${t}.`), p(t <= a, () => `bandPart(): numLower (${t}) must not be greater than the number of rows (${a}).`), i = l(t < 0 ? a : t, "numLower", "bandPart")) : (p(t.dtype === "int32", () => "bandPart(): numLower's dtype must be an int32."), i = G(tn(t, 0), a, rn(t, a))), typeof n == "number" ? (p(n % 1 === 0, () => `bandPart(): numUpper must be an integer, got ${n}.`), p(n <= o, () => `bandPart(): numUpper (${n}) must not be greater than the number of columns (${o}).`), u = l(n < 0 ? o : n, "numUpper", "bandPart")) : (p(n.dtype === "int32", () => "bandPart(): numUpper's dtype must be an int32."), u = G(tn(n, 0), o, rn(n, o)));
|
|
1099
|
-
const c = _(
|
|
1099
|
+
const c = _(W(0, a, 1, "int32"), [-1, 1]), b = W(0, o, 1, "int32"), m = S(c, b), $ = be(pn(m, i), ce(m, nn(u))), g = Qn([a, o], e.dtype);
|
|
1100
1100
|
return _(Y(mn(_(e, [-1, a, o])).map((x) => G($, x, g))), s);
|
|
1101
1101
|
}
|
|
1102
1102
|
const ms = /* @__PURE__ */ h({ bandPart_: ps });
|
|
@@ -1132,7 +1132,7 @@ function hs(r) {
|
|
|
1132
1132
|
let a = e[s];
|
|
1133
1133
|
if (s > 0)
|
|
1134
1134
|
for (let o = 0; o < s; ++o) {
|
|
1135
|
-
const i = k(
|
|
1135
|
+
const i = k(q(k(n[o], a)), n[o]);
|
|
1136
1136
|
a = S(a, i);
|
|
1137
1137
|
}
|
|
1138
1138
|
return R(a, ln(a, "euclidean"));
|
|
@@ -1178,35 +1178,35 @@ function cn(r, t = !1) {
|
|
|
1178
1178
|
p(r.shape.length === 2, () => `qr2d() requires a 2D Tensor, but got a ${r.shape.length}D Tensor.`);
|
|
1179
1179
|
const n = r.shape[0], e = r.shape[1];
|
|
1180
1180
|
let s = ae(n), a = L(r);
|
|
1181
|
-
const o =
|
|
1181
|
+
const o = H([[1]], [1, 1]);
|
|
1182
1182
|
let i = L(o);
|
|
1183
1183
|
const u = n >= e ? e : n;
|
|
1184
1184
|
for (let c = 0; c < u; ++c) {
|
|
1185
1185
|
const b = a, m = i, $ = s;
|
|
1186
1186
|
[i, a, s] = f.tidy(() => {
|
|
1187
|
-
const g =
|
|
1187
|
+
const g = M(a, [c, c], [n - c, 1]), x = ln(g), y = M(a, [c, c], [1, 1]), d = G(U(y, 0), H([[-1]]), H([[1]])), E = S(y, k(d, x)), I = R(g, E);
|
|
1188
1188
|
I.shape[0] === 1 ? i = L(o) : i = Q([
|
|
1189
1189
|
o,
|
|
1190
|
-
|
|
1190
|
+
M(I, [1, 0], [I.shape[0] - 1, I.shape[1]])
|
|
1191
1191
|
], 0);
|
|
1192
|
-
const A = nn(R(T(d, E), x)), z =
|
|
1192
|
+
const A = nn(R(T(d, E), x)), z = M(a, [c, 0], [n - c, e]), B = k(A, i), P = an(i);
|
|
1193
1193
|
if (c === 0)
|
|
1194
|
-
a = S(z, T(
|
|
1194
|
+
a = S(z, T(B, T(P, z)));
|
|
1195
1195
|
else {
|
|
1196
|
-
const C = S(z, T(
|
|
1197
|
-
a = Q([
|
|
1196
|
+
const C = S(z, T(B, T(P, z)));
|
|
1197
|
+
a = Q([M(a, [0, 0], [c, e]), C], 0);
|
|
1198
1198
|
}
|
|
1199
|
-
const sn = an(
|
|
1199
|
+
const sn = an(B), F = M(s, [0, c], [n, s.shape[1] - c]);
|
|
1200
1200
|
if (c === 0)
|
|
1201
1201
|
s = S(F, T(T(F, i), sn));
|
|
1202
1202
|
else {
|
|
1203
1203
|
const C = S(F, T(T(F, i), sn));
|
|
1204
|
-
s = Q([
|
|
1204
|
+
s = Q([M(s, [0, 0], [n, c]), C], 1);
|
|
1205
1205
|
}
|
|
1206
1206
|
return [i, a, s];
|
|
1207
1207
|
}), Jn([b, m, $]);
|
|
1208
1208
|
}
|
|
1209
|
-
return !t && n > e && (s =
|
|
1209
|
+
return !t && n > e && (s = M(s, [0, 0], [n, e]), a = M(a, [0, 0], [e, e])), [s, a];
|
|
1210
1210
|
});
|
|
1211
1211
|
}
|
|
1212
1212
|
const ds = /* @__PURE__ */ h({ qr_: bs });
|
|
@@ -1231,14 +1231,14 @@ const ws = {
|
|
|
1231
1231
|
grayscaleToRGB: we,
|
|
1232
1232
|
resizeNearestNeighbor: as,
|
|
1233
1233
|
resizeBilinear: ts,
|
|
1234
|
-
rgbToGrayscale:
|
|
1234
|
+
rgbToGrayscale: Be,
|
|
1235
1235
|
rotateWithOffset: Ke,
|
|
1236
|
-
cropAndResize:
|
|
1236
|
+
cropAndResize: Ne,
|
|
1237
1237
|
nonMaxSuppression: Ge,
|
|
1238
1238
|
nonMaxSuppressionAsync: Je,
|
|
1239
1239
|
nonMaxSuppressionWithScore: Xe,
|
|
1240
|
-
nonMaxSuppressionWithScoreAsync:
|
|
1241
|
-
nonMaxSuppressionPadded:
|
|
1240
|
+
nonMaxSuppressionWithScoreAsync: He,
|
|
1241
|
+
nonMaxSuppressionPadded: Ue,
|
|
1242
1242
|
nonMaxSuppressionPaddedAsync: es,
|
|
1243
1243
|
threshold: cs,
|
|
1244
1244
|
transform: ls
|
|
@@ -1251,8 +1251,8 @@ export {
|
|
|
1251
1251
|
be as a,
|
|
1252
1252
|
pn as b,
|
|
1253
1253
|
an as c,
|
|
1254
|
-
|
|
1255
|
-
|
|
1254
|
+
O as d,
|
|
1255
|
+
U as e,
|
|
1256
1256
|
tn as f,
|
|
1257
1257
|
ce as g,
|
|
1258
1258
|
K as h,
|
|
@@ -1266,12 +1266,12 @@ export {
|
|
|
1266
1266
|
ts as p,
|
|
1267
1267
|
as as q,
|
|
1268
1268
|
$e as r,
|
|
1269
|
-
|
|
1270
|
-
|
|
1269
|
+
M as s,
|
|
1270
|
+
N as t,
|
|
1271
1271
|
mn as u,
|
|
1272
1272
|
ke as v,
|
|
1273
1273
|
G as w,
|
|
1274
1274
|
je as x,
|
|
1275
1275
|
Fe as y,
|
|
1276
|
-
|
|
1276
|
+
Oe as z
|
|
1277
1277
|
};
|
|
@@ -1,25 +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 { s as Dt, n as wt, w as ne, a as Ze, g as Ye, b as ls, c as K, d as Ce, e as Yt, f as Ku, u as dn, h as ye, t as fn, i as Hu, l as qu, j as us, m as or, k as qt } from "./ops-
|
|
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 Ju, c as ar } from "./sin-
|
|
9
|
-
import { e as Hn, g as lr, h as cs, c as Zu } 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 { M as Xu, f as ur, r as Qu, d as tc, a as $n } from "./dropout-
|
|
14
|
-
import { z as vt } from "./zeros-
|
|
15
|
-
import { c as pe } from "./concat-
|
|
16
|
-
import { g as cr } from "./gather-
|
|
17
|
-
import { s as hr } from "./softmax-
|
|
18
|
-
import { m as Ee } from "./max-
|
|
19
|
-
import { t as ec } from "./tensor-
|
|
20
|
-
import { r as nc } from "./range-
|
|
21
|
-
import { m as sc } from "./norm-
|
|
22
|
-
import { v as ic } from "./variable-
|
|
1
|
+
import { o as T, q as I, E as O, bB as Oa, bC as Ma, bD as Ci, i as b, bE as Ii, L, bF as Di, bG as $i, bH as Ti, bI as zi, f as Ei, bJ as Li, bK as Fi, bL as Oi, bM as Mi, bN as _a, bO as _i, bP as Ra, bQ as Ri, bR as Ba, bS as Bi, B as Ge, h as kt, bn as Wa, bT as Wi, bU as Gi, bV as Pi, l as Pe, b as V, a as w, bW as Ga, bX as Ui, bY as Vi, bZ as ji, p as ce, a5 as pt, bs as Pa, b_ as Ki, b$ as Hi, c0 as qi, c1 as Ji, c2 as Zi, bu as Yi, c3 as Xi, c4 as Qi, x as Ua, ah as Va, ao as ja, c5 as tr, c6 as Ka, a7 as z, c7 as Ps, c8 as Ha, c9 as qa, g as pn, ca as Us, cb as Ja, cc as Za, cd as Ya, A as Xa, ce as Qa, cf as tl, cg as el, bc as nl, ch as sl, a4 as he, s as et, aj as U, ci as il, bj as rl, au as ht, cj as ol, z as Q, ck as al, cl as ll, cm as ul, cn as cl, co as hl, cp as pl, cq as dl, cr as fl, C as ml, cs as gl, bg as bl, bm as yl, ct as wl, J as kl, cu as xl, a0 as Nl, cv as vl, cw as Al, cx as Sl, aq as Cl, cy as Il, a8 as Dl, aS as $l, bo as Tl, ag as zl, bp as El, G as Ll, aU as Fl, ak as Ol, cz as Ml, cA as _l, cB as Rl, al as Bl, a9 as Wl, cC as Gl, cD as Pl, cE as Ul, M as Vl, bq as jl, cF as Kl, cG as Hl, a2 as ql, a_ as Jl, cH as Zl, cI as Yl, br as Xl, aY as Ql, P as tu, cJ as eu, j as rs, an as nu, bt as su, ax as iu, K as ru, as as ou, ar as au, D as lu, b9 as uu, cK as cu, ba as hu, cL as pu, b0 as du, aQ as fu, ap as mu, cM as gu, a1 as bu, aO as yu, S as wu, F as ku, bw as xu, cN as Nu, bx as vu, at as Au, bz as Su, N as Cu, cO as Iu, I as Du, b2 as $u, b1 as Tu, cP as Oe, cQ as zu, c as Eu, af as Vs, cR as Lu, t as x, a3 as $e, cS as S, cT as He, cU as qe, ai as Vt, d as J, ac as Fu, cV as js, k as Jt, H as Ou, T as Te, cW as Mu, cX as _u, m as Ks, cY as Ru, cZ as Hs, c_ as Bu } from "./index-BAzbokzv.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-sntGesxr.js";
|
|
3
|
+
import { s as Dt, n as wt, w as ne, a as Ze, g as Ye, b as ls, c as K, d as Ce, e as Yt, f as Ku, u as dn, h as ye, t as fn, i as Hu, l as qu, j as us, m as or, k as qt } from "./ops-C0sQEcPw.js";
|
|
4
|
+
import { r as N } from "./reshape-krWGKraP.js";
|
|
5
|
+
import { s as W } from "./sum-BE_Irnim.js";
|
|
6
|
+
import { m as ct } from "./mat_mul-7121rsJk.js";
|
|
7
|
+
import { s as Qt } from "./split-eiktj-6L.js";
|
|
8
|
+
import { s as Ju, c as ar } from "./sin-D_h-qCSx.js";
|
|
9
|
+
import { e as Hn, g as lr, h as cs, c as Zu } from "./axis_util-Bu4h7XWV.js";
|
|
10
|
+
import { a as se, e as ie, l as Yu } from "./log_sum_exp-YEo2h3gb.js";
|
|
11
|
+
import { s as Dn } from "./stack-dfEEz2OY.js";
|
|
12
|
+
import { o as xe } from "./ones-BBlSRqn1.js";
|
|
13
|
+
import { M as Xu, f as ur, r as Qu, d as tc, a as $n } from "./dropout-Bciw46HT.js";
|
|
14
|
+
import { z as vt } from "./zeros-YJDE7oRb.js";
|
|
15
|
+
import { c as pe } from "./concat-5aPGqw3Z.js";
|
|
16
|
+
import { g as cr } from "./gather-DjyCjmOD.js";
|
|
17
|
+
import { s as hr } from "./softmax-fsdtf6JC.js";
|
|
18
|
+
import { m as Ee } from "./max-DtlIuVeW.js";
|
|
19
|
+
import { t as ec } from "./tensor-Xyi595sG.js";
|
|
20
|
+
import { r as nc } from "./range-DYsrnfiy.js";
|
|
21
|
+
import { m as sc } from "./norm-CzltS9Fz.js";
|
|
22
|
+
import { v as ic } from "./variable-wSS22xj5.js";
|
|
23
23
|
/**
|
|
24
24
|
* @license
|
|
25
25
|
* Copyright 2020 Google LLC. All Rights Reserved.
|
|
@@ -1,32 +1,32 @@
|
|
|
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, n as ss, a as zs, r as Us, o as Vr, s as as, h as Zl, g as Jl, e as eu, b as tu, f as su, m as nu, p as ou, q as au, v as ru, j as iu, d as cu, c as lu, u as uu, x as Wr, y as Mr, z as Br } from "./ops-
|
|
8
|
-
import { m as Tt } from "./mat_mul-
|
|
9
|
-
import { w as Gr, k as Ye, h as Ve, x as du, z as pu, A as hu, y as fu, B as mu, v as gu, i as ot, j as lt, d as De, e as $t, g as Rs, S as zr, a as Ur, C as Hr, D as Xr, p as xu, q as Cu, r as bu, u as Iu } from "./selu_util-
|
|
10
|
-
import { n as Kr, m as yu } from "./norm-
|
|
11
|
-
import { e as $u, l as ku, a as wu } from "./log_sum_exp-
|
|
12
|
-
import { f as Nu } from "./dropout-
|
|
13
|
-
import { g as vu } from "./gather-
|
|
14
|
-
import { c as Hs, z as qr } from "./zeros-
|
|
15
|
-
import { m as Su } from "./max-
|
|
16
|
-
import { r as Ru } from "./reciprocal-
|
|
17
|
-
import { s as Xs } from "./split-
|
|
18
|
-
import { s as Tu } from "./softmax-
|
|
19
|
-
import { s as Du } from "./stack-
|
|
20
|
-
import { s as Eu } from "./sum-
|
|
21
|
-
import { a as j, u as ue, c as ut, i as dt, s as Fu, b as Pt, d as Pe, e as kt, f as gt, g as jr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, n as An, o as Yr, p as _u, q as Lu, v as Vu, w as Wu, x as Mu, y as Bu, A as Gu, B as zu, C as Uu, D as Hu, E as ft, F as Xu, G as Ku, H as Qr, I as qu, J as ju, K as Yu, L as Qu, M as Zu, N as Ju, O as ed, P as td, Q as sd, R as nd, S as od, T as ad, U as rd, V as id, W as cd, X as ld, Y as ud, Z as dd, _ as pd, $ as hd, a0 as fd, a1 as md, a2 as gd, a3 as xd, a4 as Cd, a5 as bd, a6 as Id, a7 as yd, a8 as $d, a9 as kd, aa as wd, ab as Nd, ac as vd, ad as Sd, ae as Rd, af as Td, ag as Dd, ah as Ed, ai as Fd, aj as Pd, ak as Zr, al as Od, t as Jr, am as Ad, an as _d, ao as ei, ap as Ld, aq as Vd, ar as Wd, as as Md, at as Bd, au as Gd, av as zd, aw as Ud, ax as Hd, ay as Xd, az as Kd, aA as qd, aB as jd, aC as Yd, aD as Qd, aE as Zd, aF as Jd, aG as ep, aH as tp, aI as sp, aJ as np, aK as op, aL as ap, aM as rp, aN as ip, aO as cp, aP as lp, aQ as up, aR as dp, aS as pp, aT as hp, aU as ti, aV as fp, aW as mp, aX as gp, aY as xp, aZ as Cp, a_ as bp, a$ as Ip, b0 as yp, b1 as $p, b2 as kp } from "./shared-
|
|
22
|
-
import { m as xt, D as Ts, E as Ds, F as Es, G as si, H as ni, I as oi, J as je, K as ai, L as ri, M as ii, N as ci, O as li, P as ui, Q as di, S as pi, T as hi, U as fi, V as mi, W as gi, X as wp, Y as Np, Z as vp, _ as Sp, $ as Rp, a0 as Tp, a1 as xi, a2 as Ci, a3 as bi, a4 as Ii, a5 as Qn, a6 as Dp, a7 as Ep, a8 as Fp, a9 as Pp, aa as Op, ab as Ap, ac as _n, ad as ze, u as We, ae as yi, c as $i, af as _e, ag as ki, g as wi, a as Ni, ah as _p, ai as Lp, aj as Z, ak as Mt, al as Vp, am as Zn, an as Wp, ao as Mp, ap as Bt, aq as Bp, ar as Gp, as as Ks, at as jt, au as Yt, av as zp, aw as Up, ax as Jn, ay as Hp, az as Xp, aA as Vs, aB as Kp, aC as qp, aD as jp, aE as Qt, aF as Ws, aG as Yp, f as Ce, aH as we, aI as Zt, aJ as Jt, aK as Qp, d as eo, e as to, i as vi, aL as Zp, aM as Jp, aN as eh, aO as th, aP as sh, aQ as nh, j as Si, aR as Gt, h as oh, aS as ah, aT as rh } from "./gpgpu_math-
|
|
23
|
-
import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-
|
|
1
|
+
import { o as oe, q as se, E as le, ch as ps, ci as hs, ck as en, cl as fs, cm as ms, co as gs, B as xs, cn as Cs, cp as bs, bm as To, i as Y, c$ as tn, cv as sn, aj as Do, z as nn, bo as Eo, cz as Is, cA as ys, cB as $s, ao as Nc, cE as on, l as vc, a as an, h as Ne, d0 as ks, cH as rn, cI as ws, L as It, bt as Fo, cM as Ns, d1 as cn, d2 as ln, s as Po, bx as Oo, d3 as un, cO as vs, d4 as dn, d5 as pn, ah as xe, d6 as S, a3 as Sc, a7 as Rc, T as Ao, d7 as Tc, cP as Dc, j as Ec, a4 as Fc, a5 as Pc, b as Oc, ac as L, d8 as _o, d9 as Lo, e as Me, aJ as Vt, da as Vo, aG as Wo, db as Ac, aF as Wt, af as Ke, dc as Mo, aq as hn, al as _c, g as z, bd as Ge, an as Lc, ar as fn, as as mn, D as Vc, aa as Wc, C as Bo, W as ne, ca as Go, cj as zo, bB as Uo, p as me, $ as st, bC as Ho, bD as Xo, bE as Ko, ai as ct, bF as qo, cq as jo, cr as Yo, bI as Qo, bH as Zo, aP as Jo, dd as ea, bJ as gn, ct as ta, aV as sa, J as na, bK as oa, bv as Te, c6 as aa, bL as ra, bM as ia, cu as ca, bN as la, a0 as xn, bO as Cn, b3 as ua, bP as da, u as nt, y as Mc, bQ as pa, bR as ha, bS as fa, c8 as ma, c9 as ga, de as xa, bl as Ss, bf as Ft, bi as bn, cw as Kn, df as mt, dg as Ca, cx as qn, dh as Bc, N as Gc, aR as ba, cy as Ia, bT as In, aS as ya, ax as $a, am as yt, di as ka, dj as wa, b4 as Na, c7 as va, dk as Sa, Y as Ra, G as Ta, dl as Da, bU as yn, aZ as $n, bW as kn, cD as Ea, M as zc, bX as Fa, bY as Pa, cF as Oa, cG as Aa, dm as _a, bZ as La, a2 as Va, be as Wa, S as Ma, a6 as Ba, b6 as Ga, b8 as za, b7 as Ua, b_ as Ha, cQ as Xa, b$ as Ka, P as qa, Z as ja, c0 as Ya, cJ as wn, dn as Qa, dp as Za, dq as Ja, R as er, K as Nn, b9 as tr, cK as sr, ba as nr, cL as or, c2 as ar, b5 as rr, b0 as vn, X as ir, dr as cr, aQ as lr, c3 as Sn, a1 as Rn, c4 as Tn, bV as Dn, c1 as ur, ds as dr, dt as pr, du as hr, dv as fr, dw as mr, F as gr, cN as xr, at as En, dx as Cr, dy as br, dz as Ir, bG as Fn, dA as yr, I as $r, bb as kr, av as Pn, b1 as wr, c5 as Nr, r as vr, dB as Sr, dC as jn, dD as Yn, dE as Uc, t as Hc, dF as Xc, dG as Kc, br as qc, b2 as jc, bc as Yc, bj as Qc, aO as Zc, bk as Jc, bs as el, a$ as tl, bg as sl, bh as nl, aM as Rr, bn as ol, a8 as al, ag as rl, bp as il, dH as cl, aT as ll, aU as ul, aW as dl, aX as pl, a9 as hl, bq as fl, a_ as ml, bz as gl, aY as xl, aw as Tr, bu as Cl, ap as bl, bw as Il, by as yl } from "./index-BAzbokzv.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-DWzaOgrn.js";
|
|
3
|
+
import { r as re } from "./reshape-krWGKraP.js";
|
|
4
|
+
import { b as jl } from "./broadcast_to-DARN-DBD.js";
|
|
5
|
+
import { c as os } from "./concat-5aPGqw3Z.js";
|
|
6
|
+
import { c as Yl, s as Ql } from "./sin-D_h-qCSx.js";
|
|
7
|
+
import { w as Lr, n as ss, a as zs, r as Us, o as Vr, s as as, h as Zl, g as Jl, e as eu, b as tu, f as su, m as nu, p as ou, q as au, v as ru, j as iu, d as cu, c as lu, u as uu, x as Wr, y as Mr, z as Br } from "./ops-C0sQEcPw.js";
|
|
8
|
+
import { m as Tt } from "./mat_mul-7121rsJk.js";
|
|
9
|
+
import { w as Gr, k as Ye, h as Ve, x as du, z as pu, A as hu, y as fu, B as mu, v as gu, i as ot, j as lt, d as De, e as $t, g as Rs, S as zr, a as Ur, C as Hr, D as Xr, p as xu, q as Cu, r as bu, u as Iu } from "./selu_util-sntGesxr.js";
|
|
10
|
+
import { n as Kr, m as yu } from "./norm-CzltS9Fz.js";
|
|
11
|
+
import { e as $u, l as ku, a as wu } from "./log_sum_exp-YEo2h3gb.js";
|
|
12
|
+
import { f as Nu } from "./dropout-Bciw46HT.js";
|
|
13
|
+
import { g as vu } from "./gather-DjyCjmOD.js";
|
|
14
|
+
import { c as Hs, z as qr } from "./zeros-YJDE7oRb.js";
|
|
15
|
+
import { m as Su } from "./max-DtlIuVeW.js";
|
|
16
|
+
import { r as Ru } from "./reciprocal-CJQeasVa.js";
|
|
17
|
+
import { s as Xs } from "./split-eiktj-6L.js";
|
|
18
|
+
import { s as Tu } from "./softmax-fsdtf6JC.js";
|
|
19
|
+
import { s as Du } from "./stack-dfEEz2OY.js";
|
|
20
|
+
import { s as Eu } from "./sum-BE_Irnim.js";
|
|
21
|
+
import { a as j, u as ue, c as ut, i as dt, s as Fu, b as Pt, d as Pe, e as kt, f as gt, g as jr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, n as An, o as Yr, p as _u, q as Lu, v as Vu, w as Wu, x as Mu, y as Bu, A as Gu, B as zu, C as Uu, D as Hu, E as ft, F as Xu, G as Ku, H as Qr, I as qu, J as ju, K as Yu, L as Qu, M as Zu, N as Ju, O as ed, P as td, Q as sd, R as nd, S as od, T as ad, U as rd, V as id, W as cd, X as ld, Y as ud, Z as dd, _ as pd, $ as hd, a0 as fd, a1 as md, a2 as gd, a3 as xd, a4 as Cd, a5 as bd, a6 as Id, a7 as yd, a8 as $d, a9 as kd, aa as wd, ab as Nd, ac as vd, ad as Sd, ae as Rd, af as Td, ag as Dd, ah as Ed, ai as Fd, aj as Pd, ak as Zr, al as Od, t as Jr, am as Ad, an as _d, ao as ei, ap as Ld, aq as Vd, ar as Wd, as as Md, at as Bd, au as Gd, av as zd, aw as Ud, ax as Hd, ay as Xd, az as Kd, aA as qd, aB as jd, aC as Yd, aD as Qd, aE as Zd, aF as Jd, aG as ep, aH as tp, aI as sp, aJ as np, aK as op, aL as ap, aM as rp, aN as ip, aO as cp, aP as lp, aQ as up, aR as dp, aS as pp, aT as hp, aU as ti, aV as fp, aW as mp, aX as gp, aY as xp, aZ as Cp, a_ as bp, a$ as Ip, b0 as yp, b1 as $p, b2 as kp } from "./shared-Ca6iDobD.js";
|
|
22
|
+
import { m as xt, D as Ts, E as Ds, F as Es, G as si, H as ni, I as oi, J as je, K as ai, L as ri, M as ii, N as ci, O as li, P as ui, Q as di, S as pi, T as hi, U as fi, V as mi, W as gi, X as wp, Y as Np, Z as vp, _ as Sp, $ as Rp, a0 as Tp, a1 as xi, a2 as Ci, a3 as bi, a4 as Ii, a5 as Qn, a6 as Dp, a7 as Ep, a8 as Fp, a9 as Pp, aa as Op, ab as Ap, ac as _n, ad as ze, u as We, ae as yi, c as $i, af as _e, ag as ki, g as wi, a as Ni, ah as _p, ai as Lp, aj as Z, ak as Mt, al as Vp, am as Zn, an as Wp, ao as Mp, ap as Bt, aq as Bp, ar as Gp, as as Ks, at as jt, au as Yt, av as zp, aw as Up, ax as Jn, ay as Hp, az as Xp, aA as Vs, aB as Kp, aC as qp, aD as jp, aE as Qt, aF as Ws, aG as Yp, f as Ce, aH as we, aI as Zt, aJ as Jt, aK as Qp, d as eo, e as to, i as vi, aL as Zp, aM as Jp, aN as eh, aO as th, aP as sh, aQ as nh, j as Si, aR as Gt, h as oh, aS as ah, aT as rh } from "./gpgpu_math-CNslybmD.js";
|
|
23
|
+
import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-Bu4h7XWV.js";
|
|
24
24
|
import { s as ih } from "./index-C4L8Cm77.js";
|
|
25
|
-
import { c as wt } from "./scatter_nd_util-
|
|
26
|
-
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-
|
|
27
|
-
import { R as hh, r as H, a as fh } from "./Reshape-
|
|
28
|
-
import { M as Ri } from "./mulmat_packed_gpu-
|
|
29
|
-
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-
|
|
25
|
+
import { c as wt } from "./scatter_nd_util-93ln7Hut.js";
|
|
26
|
+
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-CUxJCg0g.js";
|
|
27
|
+
import { R as hh, r as H, a as fh } from "./Reshape-t7Kcikjk.js";
|
|
28
|
+
import { M as Ri } from "./mulmat_packed_gpu-D4nKF7Je.js";
|
|
29
|
+
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-BYViZwhN.js";
|
|
30
30
|
/**
|
|
31
31
|
* @license
|
|
32
32
|
* Copyright 2018 Google LLC. All Rights Reserved.
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { g as p, W as w } from "./index-BAzbokzv.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)
|
|
@@ -31,7 +31,7 @@ function $(o, t, r) {
|
|
|
31
31
|
}
|
|
32
32
|
k(r, t, o);
|
|
33
33
|
}
|
|
34
|
-
function
|
|
34
|
+
function g(o, t, r) {
|
|
35
35
|
const n = t.shape.length, e = n > 1 ? t.shape[n - 1] : 1, h = r.length;
|
|
36
36
|
let a = 1;
|
|
37
37
|
for (let s = e; s < h; ++s)
|
|
@@ -41,6 +41,6 @@ function m(o, t, r) {
|
|
|
41
41
|
}
|
|
42
42
|
export {
|
|
43
43
|
k as a,
|
|
44
|
-
|
|
44
|
+
g as c,
|
|
45
45
|
$ as v
|
|
46
46
|
};
|