@genai-fi/nanogpt 0.11.0 → 0.12.0
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 +29 -29
- package/dist/{RealDiv-Ds-jvL09.js → RealDiv-C8neBwFi.js} +17 -17
- package/dist/{Reshape-Cd6e-Otn.js → Reshape-Bd4V_4X7.js} +1 -1
- package/dist/{Reshape-Ct266DEk.js → Reshape-Ck29jQSY.js} +7 -7
- package/dist/TeachableLLM.d.ts +2 -1
- package/dist/TeachableLLM.js +9 -9
- package/dist/Trainer.d.ts +4 -2
- package/dist/Trainer.js +11 -8
- package/dist/{axis_util-DofAuy0p.js → axis_util-DGqbT-FX.js} +1 -1
- package/dist/backend.js +2 -2
- package/dist/{backend_util-C7NWHpv7.js → backend_util-DC3rBo_H.js} +18 -18
- package/dist/{backend_webgpu-B0Vls736.js → backend_webgpu-mbhNnlx9.js} +10 -10
- package/dist/{broadcast_to-DDaNMbX7.js → broadcast_to-D1Dmg2Oz.js} +2 -2
- package/dist/checks/appendCache.js +2 -2
- package/dist/checks/attentionMask.js +3 -3
- package/dist/checks/gelu.js +2 -2
- package/dist/checks/matMulGelu.js +2 -2
- package/dist/checks/normRMS.js +4 -4
- package/dist/checks/normRMSGrad.js +3 -3
- package/dist/checks/packUnpack.js +2 -2
- package/dist/checks/qkv.js +2 -2
- package/dist/checks/rope.js +2 -2
- package/dist/clip_by_value-fg2aKzUy.js +12 -0
- package/dist/{complex-DClmWqJt.js → complex-Cyg-eQeZ.js} +1 -1
- package/dist/concat-CSm2rMwe.js +17 -0
- package/dist/{concat_util-CHsJFZJJ.js → concat_util-D0je5Ppu.js} +1 -1
- package/dist/{dataset-DcjWqUVQ.js → dataset-CVIJu7Xa.js} +3 -3
- package/dist/{dropout-OxuaJz6z.js → dropout-DLhSMNTZ.js} +14 -14
- package/dist/expand_dims-ChkuOp6I.js +11 -0
- package/dist/{exports_initializers-eS9QJ6ut.js → exports_initializers-1KWPiStI.js} +1 -1
- package/dist/{floor-DIb-lN_u.js → floor-BRMPgeIs.js} +1 -1
- package/dist/gather-BSULDalH.js +9 -0
- package/dist/{gelu-DqTbCx5x.js → gelu-BK1k-n1i.js} +1 -1
- package/dist/{gpgpu_math-CJcbnKPC.js → gpgpu_math-BJSTk_mW.js} +25 -25
- package/dist/{index-Dj5TkmPY.js → index-BBVLAXZD.js} +14 -14
- package/dist/{index-D0RBWjq8.js → index-Duu1Lvvv.js} +45 -45
- package/dist/{kernel_funcs_utils-CSaumNDs.js → kernel_funcs_utils-BtYrPoJu.js} +8 -8
- package/dist/layers/BaseLayer.js +2 -2
- package/dist/layers/CausalSelfAttention.js +6 -6
- package/dist/layers/MLP.js +4 -4
- package/dist/layers/PositionEmbedding.js +5 -5
- package/dist/layers/RMSNorm.js +3 -3
- package/dist/layers/RoPECache.js +4 -4
- package/dist/layers/TiedEmbedding.js +6 -6
- package/dist/layers/TransformerBlock.js +1 -1
- package/dist/loader/loadTransformers.js +1 -1
- package/dist/loader/oldZipLoad.js +17 -17
- package/dist/{log_sum_exp-VLZgbFAH.js → log_sum_exp-CVqLsVLl.js} +4 -4
- package/dist/main.d.ts +9 -0
- package/dist/main.js +68 -58
- package/dist/{matMul16-cDxwemKj.js → matMul16-xswmhSuF.js} +7 -7
- package/dist/{matMulGelu-B2s_80-H.js → matMulGelu-BpvgnYG8.js} +26 -26
- package/dist/mat_mul-Bn2BDpT4.js +11 -0
- package/dist/{mod-PrOKlFxH.js → mod-B4AUd1Np.js} +1 -1
- package/dist/models/NanoGPTV1.js +2 -2
- package/dist/models/model.js +9 -9
- package/dist/{ones-BX_wEgzB.js → ones-CBI1AQjb.js} +3 -3
- package/dist/ops/adamAdjust.js +1 -1
- package/dist/ops/adamMoments.js +1 -1
- package/dist/ops/add16.js +1 -1
- package/dist/ops/appendCache.js +3 -3
- package/dist/ops/attentionMask.js +1 -1
- package/dist/ops/concat16.js +2 -2
- package/dist/ops/cpu/adamAdjust.js +7 -7
- package/dist/ops/cpu/adamMoments.js +5 -5
- package/dist/ops/cpu/appendCache.js +6 -6
- package/dist/ops/cpu/attentionMask.js +6 -6
- package/dist/ops/cpu/fusedSoftmax.js +5 -5
- package/dist/ops/cpu/gatherSub.js +7 -7
- package/dist/ops/cpu/gelu.js +5 -5
- package/dist/ops/cpu/matMul16.js +2 -2
- package/dist/ops/cpu/matMulGelu.js +3 -3
- package/dist/ops/cpu/matMulMul.js +5 -5
- package/dist/ops/cpu/mulDropout.js +1 -1
- package/dist/ops/cpu/normRMS.js +5 -5
- package/dist/ops/cpu/qkv.js +3 -3
- package/dist/ops/cpu/rope.js +9 -9
- package/dist/ops/cpu/scatterSub.js +5 -5
- package/dist/ops/dot16.js +2 -2
- package/dist/ops/gatherSub.js +1 -1
- package/dist/ops/gelu.js +2 -2
- package/dist/ops/grads/add16.js +1 -1
- package/dist/ops/grads/attentionMask.js +2 -2
- package/dist/ops/grads/gelu.js +2 -2
- package/dist/ops/grads/matMul16.js +3 -3
- package/dist/ops/grads/matMulGelu.js +5 -5
- package/dist/ops/grads/normRMS.js +6 -6
- package/dist/ops/grads/pack16.js +3 -3
- package/dist/ops/grads/qkv.js +9 -9
- package/dist/ops/grads/rope.js +2 -2
- package/dist/ops/grads/softmax16.js +1 -1
- package/dist/ops/grads/unpack16.js +2 -2
- package/dist/ops/matMul16.js +3 -3
- package/dist/ops/matMulGelu.js +2 -2
- package/dist/ops/matMulMul.js +1 -1
- package/dist/ops/mul16.js +1 -1
- package/dist/ops/mulDrop.js +1 -1
- package/dist/ops/normRMS.js +1 -1
- package/dist/ops/pack16.js +2 -2
- package/dist/ops/qkv.js +1 -1
- package/dist/ops/reshape16.js +6 -6
- package/dist/ops/rope.js +2 -2
- package/dist/ops/scatterSub.js +1 -1
- package/dist/ops/slice16.js +2 -2
- package/dist/ops/softmax16.js +1 -1
- package/dist/ops/sub16.js +1 -1
- package/dist/ops/sum16.js +2 -2
- package/dist/ops/transpose16.js +6 -6
- package/dist/ops/unpack16.js +2 -2
- package/dist/ops/webgl/adamAdjust.js +2 -2
- package/dist/ops/webgl/adamMoments.js +1 -1
- package/dist/ops/webgl/appendCache.js +1 -1
- package/dist/ops/webgl/attentionMask.js +4 -4
- package/dist/ops/webgl/fusedSoftmax.js +6 -6
- 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/matMul16.js +10 -10
- package/dist/ops/webgl/matMulGelu.js +4 -4
- package/dist/ops/webgl/matMulMul.js +2 -2
- 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 +4 -4
- package/dist/ops/webgl/scatterSub.js +1 -1
- package/dist/ops/webgpu/adamAdjust.js +3 -3
- package/dist/ops/webgpu/adamMoments.js +5 -5
- package/dist/ops/webgpu/add16.js +1 -1
- package/dist/ops/webgpu/appendCache.js +3 -3
- package/dist/ops/webgpu/attentionMask.js +5 -5
- package/dist/ops/webgpu/attentionMask32_program.js +2 -2
- package/dist/ops/webgpu/concat16.js +5 -5
- package/dist/ops/webgpu/gatherSub.js +3 -3
- package/dist/ops/webgpu/gelu.js +3 -3
- package/dist/ops/webgpu/matMul16.js +19 -19
- package/dist/ops/webgpu/matMul16_program.js +2 -2
- package/dist/ops/webgpu/mul16.js +1 -1
- package/dist/ops/webgpu/normRMS.js +2 -2
- package/dist/ops/webgpu/normRMSGrad.js +4 -4
- package/dist/ops/webgpu/pack16.js +3 -3
- package/dist/ops/webgpu/pack16_program.js +2 -2
- package/dist/ops/webgpu/qkv.js +4 -4
- package/dist/ops/webgpu/rope.js +3 -3
- package/dist/ops/webgpu/scatterSub.js +3 -3
- package/dist/ops/webgpu/slice16.js +4 -4
- package/dist/ops/webgpu/softmax16.js +4 -4
- package/dist/ops/webgpu/softmax16_program.js +2 -2
- package/dist/ops/webgpu/softmax16_subgroup_program.js +2 -2
- package/dist/ops/webgpu/softmax16grad.js +1 -1
- package/dist/ops/webgpu/sub16.js +1 -1
- package/dist/ops/webgpu/sum16.js +5 -5
- package/dist/ops/webgpu/transpose16.js +2 -2
- package/dist/ops/webgpu/transpose16_program.js +2 -2
- package/dist/ops/webgpu/transpose16_shared_program.js +3 -3
- package/dist/ops/webgpu/unpack16.js +5 -5
- package/dist/ops/webgpu/utils/binary_op.js +3 -3
- package/dist/ops/webgpu/utils/reductions.js +4 -4
- package/dist/{ops-FJapAPfm.js → ops-C2_OXuZ4.js} +35 -35
- package/dist/{pack16-k4jq6aMX.js → pack16-atD0eYRm.js} +6 -6
- package/dist/patches/webgpu_backend.js +8 -8
- package/dist/patches/webgpu_base.js +1 -1
- package/dist/patches/webgpu_program.js +2 -2
- package/dist/{random_width-UGQn4OWb.js → random_width-BN4wGJaW.js} +33 -33
- package/dist/{range-CuGvVN2c.js → range-DKmP1-OQ.js} +1 -1
- package/dist/relu-BsXmGzzu.js +9 -0
- package/dist/{reshape-CkjKPPqB.js → reshape-BI0yzp1T.js} +1 -1
- package/dist/{resize_nearest_neighbor-DB8k9KN_.js → resize_nearest_neighbor-BA_BX-ub.js} +25 -25
- package/dist/{rope-BmZmp9uP.js → rope-DJ7Y7c-u.js} +1 -1
- package/dist/{scatter_nd_util-BY22Cc-C.js → scatter_nd_util-k9MUVUkn.js} +1 -1
- package/dist/{selu_util-BuLbmbrl.js → selu_util-DyW0X1WG.js} +5 -5
- package/dist/{shared-B7USJZgw.js → shared-Q3BS6T03.js} +1 -1
- package/dist/{shared-BQboIImQ.js → shared-nnSWpC3u.js} +6 -6
- package/dist/{slice-Aqy7KbJh.js → slice-wBNvzVyz.js} +3 -3
- package/dist/{slice_util-D8CQRenR.js → slice_util-zN8KFC5I.js} +7 -7
- package/dist/{softmax-faLoUZVT.js → softmax-DfuYyjMh.js} +1 -1
- package/dist/split-BYrLboMq.js +9 -0
- package/dist/squeeze-Bk8Brcct.js +10 -0
- package/dist/{stack-WJK22CFn.js → stack-CDWShFHF.js} +1 -1
- package/dist/{step-dXR33iOg.js → step-BS5JXRR6.js} +14 -14
- package/dist/sum-BPUfDB2X.js +11 -0
- package/dist/{tensor-BQqrDvpx.js → tensor-CEt9Nm2s.js} +1 -1
- package/dist/{tensor1d-LxP9asMm.js → tensor1d-Cc_KCIDg.js} +1 -1
- package/dist/{tensor2d-BN1sSfQO.js → tensor2d-BN97fF71.js} +1 -1
- package/dist/{tensor4d-DVwr7pLF.js → tensor4d-vuDDgdUI.js} +1 -1
- package/dist/{tfjs_backend-Vi4JfLzT.js → tfjs_backend-806hyYve.js} +36 -36
- package/dist/tile-OWUvpIVt.js +11 -0
- package/dist/tokeniser/BaseTokeniser.d.ts +6 -8
- package/dist/tokeniser/BaseTokeniser.js +6 -6
- package/dist/tokeniser/CharTokeniser.d.ts +6 -6
- package/dist/tokeniser/CharTokeniser.js +26 -26
- package/dist/tokeniser/bpe.d.ts +6 -6
- package/dist/tokeniser/bpe.js +9 -9
- package/dist/tokeniser/type.d.ts +6 -8
- package/dist/training/Adam.js +2 -2
- package/dist/training/AdamExt.js +1 -1
- package/dist/training/DatasetBuilder.d.ts +1 -1
- package/dist/training/DatasetBuilder.js +29 -29
- package/dist/training/FullTrainer.js +1 -1
- package/dist/training/Trainer.d.ts +5 -4
- package/dist/training/Trainer.js +22 -25
- package/dist/training/sparseCrossEntropy.js +3 -3
- package/dist/training/tasks/ConversationTask.d.ts +11 -0
- package/dist/training/tasks/ConversationTask.js +26 -0
- package/dist/training/tasks/PretrainingTask.d.ts +11 -0
- package/dist/training/tasks/PretrainingTask.js +34 -0
- package/dist/training/tasks/StartSentenceTask.d.ts +12 -0
- package/dist/training/tasks/StartSentenceTask.js +42 -0
- package/dist/training/tasks/Task.d.ts +8 -0
- package/dist/training/tasks/Task.js +41 -0
- package/dist/{transpose-JawVKyZy.js → transpose-BUkQCJp9.js} +7 -7
- package/dist/{unsorted_segment_sum-LAbmE9G4.js → unsorted_segment_sum-BljxHhCY.js} +78 -78
- package/dist/utilities/dummy.js +3 -3
- package/dist/utilities/multinomialCPU.js +2 -2
- package/dist/utilities/packed.js +1 -1
- package/dist/utilities/performance.js +1 -1
- package/dist/utilities/profile.js +1 -1
- package/dist/utilities/safetensors.js +2 -2
- package/dist/utilities/sentences.d.ts +1 -1
- package/dist/utilities/sentences.js +11 -11
- package/dist/utilities/weights.js +2 -2
- package/dist/{variable-DQ9yYgEU.js → variable-DPt_Iuog.js} +1 -1
- package/dist/{webgpu_program-CAE4RICo.js → webgpu_program-BpWRlghH.js} +1 -1
- package/dist/{webgpu_util-BdovYhXr.js → webgpu_util-DMiKzzQM.js} +7 -7
- package/dist/{zeros-DeiE2zTa.js → zeros-5YROwwUH.js} +2 -2
- package/dist/{zeros_like-BAz3iKru.js → zeros_like-De4n1C3m.js} +57 -57
- package/package.json +1 -1
- package/dist/clip_by_value-Dn5tzexi.js +0 -12
- package/dist/concat-C6X3AAlQ.js +0 -17
- package/dist/expand_dims-BzfJK2uc.js +0 -11
- package/dist/gather-BcO5UQNJ.js +0 -9
- package/dist/mat_mul-DxpNTCRz.js +0 -11
- package/dist/relu-Cf80uA2p.js +0 -9
- package/dist/split-BNz5jcGc.js +0 -9
- package/dist/squeeze--YMgaAAf.js +0 -10
- package/dist/sum-BdplSvq_.js +0 -11
- package/dist/tile-CvN_LyVr.js +0 -11
package/dist/checks/rope.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import t from "../layers/RoPECache.js";
|
|
2
|
-
import { s as c, e as i } from "../index-
|
|
3
|
-
import { t as p } from "../tensor4d-
|
|
2
|
+
import { s as c, e as i } from "../index-Duu1Lvvv.js";
|
|
3
|
+
import { t as p } from "../tensor4d-vuDDgdUI.js";
|
|
4
4
|
async function y(a) {
|
|
5
5
|
await c(a);
|
|
6
6
|
const o = p(
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import { o as a, q as m, x as y, h as c, E as f, C as p } from "./index-Duu1Lvvv.js";
|
|
2
|
+
function u(o, s, t) {
|
|
3
|
+
const r = m(o, "x", "clipByValue");
|
|
4
|
+
if (y(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)
|
|
5
|
+
return c(r.shape, s, r.dtype);
|
|
6
|
+
const n = { x: r }, e = { clipValueMin: s, clipValueMax: t };
|
|
7
|
+
return f.runKernel(p, n, e);
|
|
8
|
+
}
|
|
9
|
+
const E = /* @__PURE__ */ a({ clipByValue_: u });
|
|
10
|
+
export {
|
|
11
|
+
E as c
|
|
12
|
+
};
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { o as c, q as s, a6 as n, E as m, a7 as r } from "./index-Duu1Lvvv.js";
|
|
2
2
|
function l(o, p) {
|
|
3
3
|
const a = s(o, "real", "complex"), e = s(p, "imag", "complex");
|
|
4
4
|
n(a.shape, e.shape, `real and imag shapes, ${a.shape} and ${e.shape}, must match in call to tf.complex().`);
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { o as s, x as a, y as p, w as i, E as l, z as f } from "./index-Duu1Lvvv.js";
|
|
2
|
+
function h(o, e = 0) {
|
|
3
|
+
a(o.length >= 1, () => "Pass at least one tensor to concat");
|
|
4
|
+
const t = p(o, "tensors", "concat", "string_or_numeric");
|
|
5
|
+
if (t[0].dtype === "complex64" && t.forEach((n) => {
|
|
6
|
+
if (n.dtype !== "complex64")
|
|
7
|
+
throw new Error(`Cannot concatenate complex64 tensors with a tensor
|
|
8
|
+
with dtype ${n.dtype}. `);
|
|
9
|
+
}), t.length === 1)
|
|
10
|
+
return i(t[0]);
|
|
11
|
+
const r = t, c = { axis: e };
|
|
12
|
+
return l.runKernel(f, r, c);
|
|
13
|
+
}
|
|
14
|
+
const u = /* @__PURE__ */ s({ concat_: h });
|
|
15
|
+
export {
|
|
16
|
+
u as c
|
|
17
|
+
};
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import { aC as S,
|
|
1
|
+
import { aC as S, a3 as h, aa as N, d as v, aD as o, aE as p, aF as g, x as k, t as y } from "./index-Duu1Lvvv.js";
|
|
2
2
|
import { s as R } from "./index-CieiGp4Y.js";
|
|
3
|
-
import { s as $ } from "./stack-
|
|
4
|
-
import { t as B } from "./tensor-
|
|
3
|
+
import { s as $ } from "./stack-CDWShFHF.js";
|
|
4
|
+
import { t as B } from "./tensor-CEt9Nm2s.js";
|
|
5
5
|
function _(s, t) {
|
|
6
6
|
return b(s, t);
|
|
7
7
|
}
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { f as w } from "./floor-
|
|
1
|
+
import { o as l, u as h, at as d, ab as f, q as m, x as u, a3 as p, j as g, l as V, m as v } from "./index-Duu1Lvvv.js";
|
|
2
|
+
import { f as w } from "./floor-BRMPgeIs.js";
|
|
3
3
|
import { s as c } from "./index-CieiGp4Y.js";
|
|
4
|
-
class
|
|
4
|
+
class b {
|
|
5
5
|
constructor(t, e, s, n, o) {
|
|
6
6
|
this.mean = t, this.stdDev = e, this.dtype = s, this.nextVal = NaN, this.truncated = n, this.truncated && (this.upper = this.mean + this.stdDev * 2, this.lower = this.mean - this.stdDev * 2);
|
|
7
7
|
const a = o || Math.random();
|
|
@@ -33,7 +33,7 @@ class N {
|
|
|
33
33
|
return t <= this.upper && t >= this.lower;
|
|
34
34
|
}
|
|
35
35
|
}
|
|
36
|
-
class
|
|
36
|
+
class N {
|
|
37
37
|
constructor(t = 0, e = 1, s, n) {
|
|
38
38
|
if (this.canReturnFloat = () => this.dtype == null || this.dtype === "float32", this.min = t, this.range = e - t, this.dtype = s, n == null && (n = Math.random()), typeof n == "number" && (n = n.toString()), !this.canReturnFloat() && this.range <= 1)
|
|
39
39
|
throw new Error(`The difference between ${t} - ${e} <= 1 and dtype is not float`);
|
|
@@ -46,24 +46,24 @@ class b {
|
|
|
46
46
|
return this.convertValue(this.min + this.range * this.random());
|
|
47
47
|
}
|
|
48
48
|
}
|
|
49
|
-
function
|
|
49
|
+
function T(r, t = 0, e = 1, s, n) {
|
|
50
50
|
if (h(r), s != null && s === "bool")
|
|
51
51
|
throw new Error(`Unsupported data type ${s}`);
|
|
52
|
-
const o = new
|
|
52
|
+
const o = new b(t, e, s, !1, n), a = d(r, s);
|
|
53
53
|
for (let i = 0; i < a.values.length; i++)
|
|
54
54
|
a.values[i] = o.nextValue();
|
|
55
55
|
return a.toTensor();
|
|
56
56
|
}
|
|
57
|
-
const U = /* @__PURE__ */ l({ randomNormal_:
|
|
58
|
-
function
|
|
57
|
+
const U = /* @__PURE__ */ l({ randomNormal_: T });
|
|
58
|
+
function x(r, t = 0, e = 1, s = "float32", n) {
|
|
59
59
|
h(r);
|
|
60
|
-
const o = d(r, s), a = new
|
|
60
|
+
const o = d(r, s), a = new N(t, e, null, n);
|
|
61
61
|
for (let i = 0; i < o.values.length; i++)
|
|
62
62
|
o.values[i] = a.nextValue();
|
|
63
63
|
return o.toTensor();
|
|
64
64
|
}
|
|
65
|
-
const
|
|
66
|
-
function
|
|
65
|
+
const y = /* @__PURE__ */ l({ randomUniform_: x });
|
|
66
|
+
function M(r, t) {
|
|
67
67
|
if (t == null)
|
|
68
68
|
return r.shape.slice();
|
|
69
69
|
if (f(r.shape, t))
|
|
@@ -80,13 +80,13 @@ function D(r, t, e, s) {
|
|
|
80
80
|
const n = m(r, "x", "dropout");
|
|
81
81
|
if (u(n.dtype === "float32", () => `x has to be a floating point tensor since it's going to be scaled, but got a ${n.dtype} tensor instead.`), u(t >= 0 && t < 1, () => `rate must be a float in the range [0, 1), but got ${t}.`), t === 0)
|
|
82
82
|
return r instanceof p ? n.clone() : n;
|
|
83
|
-
const o =
|
|
83
|
+
const o = M(n, e), a = 1 - t, i = g(w(V(y(o, 0, 1, "float32", s), a)), a);
|
|
84
84
|
return v(n, i);
|
|
85
85
|
}
|
|
86
86
|
const q = /* @__PURE__ */ l({ dropout_: D });
|
|
87
87
|
export {
|
|
88
|
-
|
|
89
|
-
|
|
88
|
+
b as M,
|
|
89
|
+
y as a,
|
|
90
90
|
q as d,
|
|
91
91
|
U as r
|
|
92
92
|
};
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { o as e, q as a, x as i, E as m, A as p } from "./index-Duu1Lvvv.js";
|
|
2
|
+
function x(t, n = 0) {
|
|
3
|
+
const s = a(t, "x", "expandDims", "string_or_numeric");
|
|
4
|
+
i(n <= s.rank, () => "Axis must be <= rank of the tensor");
|
|
5
|
+
const r = { input: s }, o = { dim: n };
|
|
6
|
+
return m.runKernel(p, r, o);
|
|
7
|
+
}
|
|
8
|
+
const u = /* @__PURE__ */ e({ expandDims_: x });
|
|
9
|
+
export {
|
|
10
|
+
u as e
|
|
11
|
+
};
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { o as g, q as t, E as h, G as p } from "./index-Duu1Lvvv.js";
|
|
2
|
+
function u(n, s, r = 0, e = 0) {
|
|
3
|
+
const o = t(n, "x", "gather"), a = t(s, "indices", "gather", "int32"), c = { x: o, indices: a }, i = { axis: r, batchDims: e };
|
|
4
|
+
return h.runKernel(p, c, i);
|
|
5
|
+
}
|
|
6
|
+
const d = /* @__PURE__ */ g({ gather_: u });
|
|
7
|
+
export {
|
|
8
|
+
d as g
|
|
9
|
+
};
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { b as se } from "./backend_util-
|
|
1
|
+
import { aa as $, U as E, aN as X, ab as T, aO as ae, K as N, aP as G, x as ie, a8 as k, ac as ce } from "./index-Duu1Lvvv.js";
|
|
2
|
+
import { b as se } from "./backend_util-DC3rBo_H.js";
|
|
3
3
|
const w = {}, M = {
|
|
4
4
|
alpha: !1,
|
|
5
5
|
antialias: !1,
|
|
@@ -215,7 +215,7 @@ function Ot(e, t, n) {
|
|
|
215
215
|
function It(e, t) {
|
|
216
216
|
v(e, () => e.bindFramebuffer(e.FRAMEBUFFER, t)), v(e, () => e.framebufferTexture2D(e.FRAMEBUFFER, e.COLOR_ATTACHMENT0, e.TEXTURE_2D, null, 0));
|
|
217
217
|
}
|
|
218
|
-
function
|
|
218
|
+
function Ut(e) {
|
|
219
219
|
const t = e.checkFramebufferStatus(e.FRAMEBUFFER);
|
|
220
220
|
if (t !== e.FRAMEBUFFER_COMPLETE)
|
|
221
221
|
throw new Error("Error binding framebuffer: " + Se(e, t));
|
|
@@ -258,7 +258,7 @@ function J(e) {
|
|
|
258
258
|
e[e.length - 1]
|
|
259
259
|
];
|
|
260
260
|
}
|
|
261
|
-
function
|
|
261
|
+
function Lt(e) {
|
|
262
262
|
let t = [1, 1, 1];
|
|
263
263
|
return e.length === 0 || e.length === 1 && e[0] === 1 || (t = [Z(e), ...J(e)]), t;
|
|
264
264
|
}
|
|
@@ -378,7 +378,7 @@ function Xt(e, t) {
|
|
|
378
378
|
n != null && ie(n.dtype !== "complex64", () => `${t} does not support complex64 tensors in the WebGL backend.`);
|
|
379
379
|
});
|
|
380
380
|
}
|
|
381
|
-
function
|
|
381
|
+
function U() {
|
|
382
382
|
let e, t, n, r, o, a, i, c, s, l;
|
|
383
383
|
return $().getNumber("WEBGL_VERSION") === 2 ? (e = "#version 300 es", t = "in", n = "out", r = "in", o = "texture", a = "outputColor", i = "out vec4 outputColor;", c = $().getBool("WEBGL2_ISNAN_CUSTOM") ? `
|
|
384
384
|
bool isnan_custom(float val) {
|
|
@@ -567,8 +567,8 @@ function Fe(e, t, n) {
|
|
|
567
567
|
});
|
|
568
568
|
const o = r.join(`
|
|
569
569
|
`), a = e.map((x) => _e(x, t, n.packedInputs, n.enableShapeUniforms)).join(`
|
|
570
|
-
`), i = t.texShape, c =
|
|
571
|
-
let l, u, f =
|
|
570
|
+
`), i = t.texShape, c = U(), s = Ne(c);
|
|
571
|
+
let l, u, f = Ue(c);
|
|
572
572
|
return t.isPacked ? (l = Ae(t.logicalShape, i, n.enableShapeUniforms), u = Ie(c)) : (l = we(t.logicalShape, i, n.enableShapeUniforms), u = Oe(c)), n.packedInputs && (f += be), [
|
|
573
573
|
f,
|
|
574
574
|
s,
|
|
@@ -580,7 +580,7 @@ function Fe(e, t, n) {
|
|
|
580
580
|
].join(`
|
|
581
581
|
`);
|
|
582
582
|
}
|
|
583
|
-
function
|
|
583
|
+
function L(e, t = !1) {
|
|
584
584
|
const n = e.shapeInfo.logicalShape;
|
|
585
585
|
switch (n.length) {
|
|
586
586
|
case 0:
|
|
@@ -617,7 +617,7 @@ function te(e, t) {
|
|
|
617
617
|
}
|
|
618
618
|
function _e(e, t, n = !1, r) {
|
|
619
619
|
let o = "";
|
|
620
|
-
n ? o += te(e, r) : o +=
|
|
620
|
+
n ? o += te(e, r) : o += L(e, r);
|
|
621
621
|
const a = e.shapeInfo.logicalShape, i = t.logicalShape;
|
|
622
622
|
return a.length <= i.length && (n ? o += ot(e, t) : o += at(e, t)), o;
|
|
623
623
|
}
|
|
@@ -676,7 +676,7 @@ function Ie(e) {
|
|
|
676
676
|
}
|
|
677
677
|
`;
|
|
678
678
|
}
|
|
679
|
-
function
|
|
679
|
+
function Ue(e) {
|
|
680
680
|
return `${e.version}
|
|
681
681
|
precision highp float;
|
|
682
682
|
precision highp int;
|
|
@@ -732,12 +732,12 @@ function Le(e) {
|
|
|
732
732
|
return fract((p3.x + p3.y) * p3.z);
|
|
733
733
|
}
|
|
734
734
|
|
|
735
|
-
${
|
|
735
|
+
${Le}
|
|
736
736
|
${De}
|
|
737
737
|
${ge}
|
|
738
738
|
`;
|
|
739
739
|
}
|
|
740
|
-
const
|
|
740
|
+
const Le = `
|
|
741
741
|
vec2 uvFromFlat(int texNumR, int texNumC, int index) {
|
|
742
742
|
int texR = index / texNumC;
|
|
743
743
|
int texC = index - texR * texNumC;
|
|
@@ -1112,7 +1112,7 @@ function O(e) {
|
|
|
1112
1112
|
return `offset${e}`;
|
|
1113
1113
|
}
|
|
1114
1114
|
function He(e) {
|
|
1115
|
-
const t = e.name, n = "get" + t.charAt(0).toUpperCase() + t.slice(1), r =
|
|
1115
|
+
const t = e.name, n = "get" + t.charAt(0).toUpperCase() + t.slice(1), r = U();
|
|
1116
1116
|
return `
|
|
1117
1117
|
vec4 ${n}() {
|
|
1118
1118
|
return ${r.texture2D}(${t}, halfCR);
|
|
@@ -1147,7 +1147,7 @@ function qe(e, t) {
|
|
|
1147
1147
|
`;
|
|
1148
1148
|
}
|
|
1149
1149
|
function je(e, t) {
|
|
1150
|
-
const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = e.shapeInfo.texShape, a =
|
|
1150
|
+
const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = e.shapeInfo.texShape, a = U();
|
|
1151
1151
|
if (t)
|
|
1152
1152
|
return `
|
|
1153
1153
|
vec4 ${r}(int index) {
|
|
@@ -1215,7 +1215,7 @@ function Ye(e, t) {
|
|
|
1215
1215
|
`;
|
|
1216
1216
|
}
|
|
1217
1217
|
function Ke(e, t) {
|
|
1218
|
-
const n = e.shapeInfo.logicalShape, r = e.name, o = "get" + r.charAt(0).toUpperCase() + r.slice(1), a = e.shapeInfo.texShape, i = a[0], c = a[1], s =
|
|
1218
|
+
const n = e.shapeInfo.logicalShape, r = e.name, o = "get" + r.charAt(0).toUpperCase() + r.slice(1), a = e.shapeInfo.texShape, i = a[0], c = a[1], s = U();
|
|
1219
1219
|
if (a != null && T(n, a))
|
|
1220
1220
|
return t ? `
|
|
1221
1221
|
vec4 ${o}(int row, int col) {
|
|
@@ -1269,7 +1269,7 @@ function Ze(e, t) {
|
|
|
1269
1269
|
if (s.length < n.length) {
|
|
1270
1270
|
const d = g(e, s), x = ["row", "col"];
|
|
1271
1271
|
return `
|
|
1272
|
-
${
|
|
1272
|
+
${L(d, t)}
|
|
1273
1273
|
float ${o}(int row, int col) {
|
|
1274
1274
|
return ${o}(${b(x, c)});
|
|
1275
1275
|
}
|
|
@@ -1334,7 +1334,7 @@ function Je(e, t) {
|
|
|
1334
1334
|
}
|
|
1335
1335
|
`;
|
|
1336
1336
|
}
|
|
1337
|
-
const c =
|
|
1337
|
+
const c = U();
|
|
1338
1338
|
if (t)
|
|
1339
1339
|
return `
|
|
1340
1340
|
vec4 ${o}(int b, int row, int col) {
|
|
@@ -1360,7 +1360,7 @@ function Qe(e, t) {
|
|
|
1360
1360
|
if (l.length < n.length) {
|
|
1361
1361
|
const p = g(e, l), m = ["row", "col", "depth"];
|
|
1362
1362
|
return `
|
|
1363
|
-
${
|
|
1363
|
+
${L(p, t)}
|
|
1364
1364
|
float ${o}(int row, int col, int depth) {
|
|
1365
1365
|
return ${o}(${b(m, s)});
|
|
1366
1366
|
}
|
|
@@ -1430,7 +1430,7 @@ function Qe(e, t) {
|
|
|
1430
1430
|
`;
|
|
1431
1431
|
}
|
|
1432
1432
|
function et(e, t) {
|
|
1433
|
-
const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o =
|
|
1433
|
+
const n = e.name, r = "get" + n.charAt(0).toUpperCase() + n.slice(1), o = U();
|
|
1434
1434
|
if (t)
|
|
1435
1435
|
return `
|
|
1436
1436
|
vec4 ${r}(int b2, int b, int row, int col) {
|
|
@@ -1464,7 +1464,7 @@ function tt(e, t) {
|
|
|
1464
1464
|
if (s.length < n.length) {
|
|
1465
1465
|
const R = g(e, s), F = ["row", "col", "depth", "depth2"];
|
|
1466
1466
|
return `
|
|
1467
|
-
${
|
|
1467
|
+
${L(R, t)}
|
|
1468
1468
|
float ${o}(int row, int col, int depth, int depth2) {
|
|
1469
1469
|
return ${o}(${b(F, l)});
|
|
1470
1470
|
}
|
|
@@ -1550,7 +1550,7 @@ function nt(e) {
|
|
|
1550
1550
|
if (s.length < t.length) {
|
|
1551
1551
|
const p = g(e, s), m = ["row", "col", "depth", "depth2", "depth3"];
|
|
1552
1552
|
return `
|
|
1553
|
-
${
|
|
1553
|
+
${L(p)}
|
|
1554
1554
|
float ${r}(int row, int col, int depth, int depth2, int depth3) {
|
|
1555
1555
|
return ${r}(${b(m, l)});
|
|
1556
1556
|
}
|
|
@@ -1607,7 +1607,7 @@ function rt(e) {
|
|
|
1607
1607
|
if (o.length < t.length) {
|
|
1608
1608
|
const m = g(e, o), S = ["row", "col", "depth", "depth2", "depth3", "depth4"];
|
|
1609
1609
|
return `
|
|
1610
|
-
${
|
|
1610
|
+
${L(m)}
|
|
1611
1611
|
float ${r}(int row, int col, int depth,
|
|
1612
1612
|
int depth2, int depth3, int depth4) {
|
|
1613
1613
|
return ${r}(${b(S, a)});
|
|
@@ -1979,12 +1979,12 @@ export {
|
|
|
1979
1979
|
wt as O,
|
|
1980
1980
|
H as P,
|
|
1981
1981
|
Nt as Q,
|
|
1982
|
-
|
|
1982
|
+
Ut as R,
|
|
1983
1983
|
Ot as S,
|
|
1984
1984
|
q as T,
|
|
1985
1985
|
ut as U,
|
|
1986
1986
|
j as V,
|
|
1987
|
-
|
|
1987
|
+
Lt as W,
|
|
1988
1988
|
xt as X,
|
|
1989
1989
|
ht as Y,
|
|
1990
1990
|
Yt as Z,
|
|
@@ -2008,7 +2008,7 @@ export {
|
|
|
2008
2008
|
yt as m,
|
|
2009
2009
|
Pt as n,
|
|
2010
2010
|
Vt as o,
|
|
2011
|
-
|
|
2011
|
+
U as p,
|
|
2012
2012
|
Q as q,
|
|
2013
2013
|
St as r,
|
|
2014
2014
|
Rt as s,
|
|
@@ -1,19 +1,19 @@
|
|
|
1
|
-
import { W as Mt } from "./backend_webgpu-
|
|
2
|
-
import {
|
|
3
|
-
import { i as Qi, a as Zi, c as b, f as v, M as Y, b as nt, d as ut, e as dt } from "./webgpu_util-
|
|
1
|
+
import { W as Mt } from "./backend_webgpu-mbhNnlx9.js";
|
|
2
|
+
import { ad as Et, aa as X, x as L, de as Ut, df as Ht, bZ as Gt, U as D, _ as j, aX as Xt, ag as Ye, aQ as Kt, a7 as qt, ak as fe, bR as Yt, c9 as jt, ca as Qt, bX as Zt, cQ as Jt, as as es, ab as De, af as te, aS as ts, bo as ss, bp as os, bq as is, cb as as, cc as rs, cd as ns, ce as us, cf as ds, cg as ls, am as cs, b7 as hs, br as ps, cA as fs, cR as ms, cS as gs, D as xs, S as Cs, bt as ws, bf as ys, dg as Ss, b9 as bs, ar as vs, bU as ks, bV as Is, ae as Rs, b_ as Ps, C as $s, cU as Ds, ap as Ns, z as zs, bv as As, cF as Fs, bw as Ws, cB as Ls, cV as Vs, cC as Bs, bx as _s, by as Ts, bh as Os, bz as Ms, bA as Es, cD as Us, ch as Hs, bB as Gs, cH as Xs, cI as Ks, dh as qs, ci as Ys, cW as js, cX as Qs, di as Zs, c2 as Js, N as eo, bg as to, aI as so, cY as oo, bC as io, bD as ao, an as ro, A as no, b$ as uo, cr as lo, bi as co, F as ho, c0 as po, dj as fo, a8 as at, bu as mo, cG as go, dk as xo, aj as Co, G as wo, at as ke, b1 as yo, b2 as So, cs as bo, cj as vo, ck as ko, cl as Io, aJ as Ro, b3 as Po, b4 as $o, dl as Do, ao as No, b5 as zo, b6 as Ao, bF as Fo, cn as Wo, cm as Lo, c_ as Vo, c1 as Bo, bG as _o, cE as To, c$ as Oo, d0 as Mo, dm as Eo, a$ as Uo, b8 as Ho, co as Go, M as Xo, I as Ko, dn as qo, aq as Yo, bk as jo, bl as Qo, bH as Zo, d5 as Jo, bI as ei, P as ti, a6 as si, bJ as oi, d1 as ii, aK as ai, c3 as ri, Z as ni, aT as ui, cp as di, H as li, aL as ci, bd as hi, d2 as pi, be as fi, d3 as mi, bL as gi, bj as xi, ba as Ci, bM as wi, ai as yi, dp as Si, a_ as bi, bN as vi, aH as ki, cq as Ii, bO as Ri, bP as Pi, bE as $i, bK as Di, dq as Ni, dr as zi, T as Ai, ax as rt, ds as Fi, Q as Wi, J as Li, c5 as Vi, d4 as Bi, bb as _i, aM as Ti, ct as Oi, dt as Mi, c7 as Ei, cu as Ui, bs as Hi, du as Gi, cv as Xi, bn as Ki, bc as qi, bQ as Yi, f as ji } from "./index-Duu1Lvvv.js";
|
|
3
|
+
import { i as Qi, a as Zi, c as b, f as v, M as Y, b as nt, d as ut, e as dt } from "./webgpu_util-DMiKzzQM.js";
|
|
4
4
|
import { g as Te, B as F } from "./binary_op_util-pKXltfxI.js";
|
|
5
|
-
import { S as Ji, a as ea } from "./selu_util-
|
|
6
|
-
import { E as ta, t as sa, u as oa, w as ia, x as aa, y as ra, f as je, z as lt, A as ct, B as ht, C as na, D as ua, F as da, G as la, H as ca, I as ha, J as pa, K as fa, L as ma, M as ga, N as xa, O as Ca } from "./backend_util-
|
|
7
|
-
import { t as W, e as S, h as Q, b as G, c as Ie, P as pt, d as wa, a as ya } from "./webgpu_program-
|
|
8
|
-
import { r as R, a as Sa } from "./Reshape-
|
|
9
|
-
import { s as ba } from "./shared-
|
|
10
|
-
import { c as Oe, a as Ce, b as we, d as Me, e as va, g as ft } from "./axis_util-
|
|
11
|
-
import { h as ye, i as Ne, j as Se, b as Z, d as Ee, g as Ue, k as mt } from "./step-
|
|
12
|
-
import { p as ka, a as Ia, b as Ra, d as Pa } from "./slice_util-
|
|
13
|
-
import { z as $a } from "./zeros-
|
|
14
|
-
import { c as me, a as Da } from "./concat_util-
|
|
5
|
+
import { S as Ji, a as ea } from "./selu_util-DyW0X1WG.js";
|
|
6
|
+
import { E as ta, t as sa, u as oa, w as ia, x as aa, y as ra, f as je, z as lt, A as ct, B as ht, C as na, D as ua, F as da, G as la, H as ca, I as ha, J as pa, K as fa, L as ma, M as ga, N as xa, O as Ca } from "./backend_util-DC3rBo_H.js";
|
|
7
|
+
import { t as W, e as S, h as Q, b as G, c as Ie, P as pt, d as wa, a as ya } from "./webgpu_program-BpWRlghH.js";
|
|
8
|
+
import { r as R, a as Sa } from "./Reshape-Bd4V_4X7.js";
|
|
9
|
+
import { s as ba } from "./shared-nnSWpC3u.js";
|
|
10
|
+
import { c as Oe, a as Ce, b as we, d as Me, e as va, g as ft } from "./axis_util-DGqbT-FX.js";
|
|
11
|
+
import { h as ye, i as Ne, j as Se, b as Z, d as Ee, g as Ue, k as mt } from "./step-BS5JXRR6.js";
|
|
12
|
+
import { p as ka, a as Ia, b as Ra, d as Pa } from "./slice_util-zN8KFC5I.js";
|
|
13
|
+
import { z as $a } from "./zeros-5YROwwUH.js";
|
|
14
|
+
import { c as me, a as Da } from "./concat_util-D0je5Ppu.js";
|
|
15
15
|
import { n as Na, a as za } from "./non_max_suppression_impl-B2W7YjZB.js";
|
|
16
|
-
import { c as He } from "./scatter_nd_util-
|
|
16
|
+
import { c as He } from "./scatter_nd_util-k9MUVUkn.js";
|
|
17
17
|
Qi() && Et(
|
|
18
18
|
"webgpu",
|
|
19
19
|
async () => {
|
|
@@ -3228,46 +3228,46 @@ function vs() {
|
|
|
3228
3228
|
}
|
|
3229
3229
|
vs();
|
|
3230
3230
|
export {
|
|
3231
|
-
|
|
3232
|
-
|
|
3233
|
-
|
|
3231
|
+
fi as $,
|
|
3232
|
+
Kr as A,
|
|
3233
|
+
K as B,
|
|
3234
3234
|
yr as C,
|
|
3235
|
-
|
|
3235
|
+
ur as D,
|
|
3236
3236
|
f as E,
|
|
3237
3237
|
Yr as F,
|
|
3238
3238
|
Qr as G,
|
|
3239
|
-
|
|
3240
|
-
|
|
3241
|
-
|
|
3242
|
-
|
|
3243
|
-
|
|
3239
|
+
Ja as H,
|
|
3240
|
+
So as I,
|
|
3241
|
+
wo as J,
|
|
3242
|
+
Gs as K,
|
|
3243
|
+
Rt as L,
|
|
3244
3244
|
Na as M,
|
|
3245
|
-
|
|
3245
|
+
yo as N,
|
|
3246
3246
|
H as O,
|
|
3247
3247
|
Oa as P,
|
|
3248
|
-
|
|
3248
|
+
se as Q,
|
|
3249
3249
|
Xa as R,
|
|
3250
3250
|
uo as S,
|
|
3251
3251
|
Co as T,
|
|
3252
|
-
|
|
3253
|
-
|
|
3254
|
-
|
|
3255
|
-
|
|
3256
|
-
|
|
3257
|
-
|
|
3258
|
-
|
|
3252
|
+
z as U,
|
|
3253
|
+
Ue as V,
|
|
3254
|
+
Cs as W,
|
|
3255
|
+
Tn as X,
|
|
3256
|
+
vn as Y,
|
|
3257
|
+
qa as Z,
|
|
3258
|
+
os as _,
|
|
3259
3259
|
si as a,
|
|
3260
3260
|
Ba as a$,
|
|
3261
|
-
|
|
3262
|
-
|
|
3263
|
-
|
|
3264
|
-
|
|
3265
|
-
|
|
3266
|
-
|
|
3267
|
-
|
|
3268
|
-
|
|
3269
|
-
|
|
3270
|
-
|
|
3261
|
+
mi as a0,
|
|
3262
|
+
pi as a1,
|
|
3263
|
+
C as a2,
|
|
3264
|
+
F as a3,
|
|
3265
|
+
gi as a4,
|
|
3266
|
+
Ds as a5,
|
|
3267
|
+
te as a6,
|
|
3268
|
+
br as a7,
|
|
3269
|
+
$t as a8,
|
|
3270
|
+
ei as a9,
|
|
3271
3271
|
xt as aA,
|
|
3272
3272
|
he as aB,
|
|
3273
3273
|
R as aC,
|
|
@@ -3295,10 +3295,10 @@ export {
|
|
|
3295
3295
|
ci as aY,
|
|
3296
3296
|
rs as aZ,
|
|
3297
3297
|
co as a_,
|
|
3298
|
-
|
|
3299
|
-
|
|
3300
|
-
|
|
3301
|
-
|
|
3298
|
+
S as aa,
|
|
3299
|
+
Dt as ab,
|
|
3300
|
+
di as ac,
|
|
3301
|
+
ii as ad,
|
|
3302
3302
|
Us as ae,
|
|
3303
3303
|
Ce as af,
|
|
3304
3304
|
Ps as ag,
|
|
@@ -3496,25 +3496,25 @@ export {
|
|
|
3496
3496
|
Wa as dy,
|
|
3497
3497
|
ja as dz,
|
|
3498
3498
|
ni as e,
|
|
3499
|
-
|
|
3499
|
+
Jo as f,
|
|
3500
3500
|
oi as g,
|
|
3501
|
-
|
|
3502
|
-
|
|
3503
|
-
|
|
3501
|
+
as as h,
|
|
3502
|
+
Xo as i,
|
|
3503
|
+
$ as j,
|
|
3504
3504
|
Fn as k,
|
|
3505
|
-
|
|
3505
|
+
k as l,
|
|
3506
3506
|
b as m,
|
|
3507
|
-
|
|
3508
|
-
|
|
3507
|
+
nt as n,
|
|
3508
|
+
N as o,
|
|
3509
3509
|
Qt as p,
|
|
3510
|
-
|
|
3510
|
+
T as q,
|
|
3511
3511
|
ai as r,
|
|
3512
3512
|
ri as s,
|
|
3513
3513
|
E as t,
|
|
3514
|
-
|
|
3514
|
+
Ct as u,
|
|
3515
3515
|
hs as v,
|
|
3516
|
-
|
|
3517
|
-
|
|
3518
|
-
|
|
3519
|
-
|
|
3516
|
+
Jn as w,
|
|
3517
|
+
p as x,
|
|
3518
|
+
En as y,
|
|
3519
|
+
wr as z
|
|
3520
3520
|
};
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { u as O, f as Y } from "./gpgpu_math-
|
|
3
|
-
import { f as v } from "./backend_util-
|
|
1
|
+
import { _ as B, U as K, aQ as G, a7 as W, aJ as z, aR as V, aa as N, aK as F, ak as S } from "./index-Duu1Lvvv.js";
|
|
2
|
+
import { u as O, f as Y } from "./gpgpu_math-BJSTk_mW.js";
|
|
3
|
+
import { f as v } from "./backend_util-DC3rBo_H.js";
|
|
4
4
|
function H(t, e) {
|
|
5
5
|
return ["x", "y", "z", "w", "u", "v"].slice(0, e).map((s) => `${t}.${s}`);
|
|
6
6
|
}
|
|
@@ -200,12 +200,12 @@ const be = {
|
|
|
200
200
|
backendName: "webgl",
|
|
201
201
|
kernelFunc: oe
|
|
202
202
|
};
|
|
203
|
-
const k = "return (a < 0.) ? b * a : a;",
|
|
203
|
+
const k = "return (a < 0.) ? b * a : a;", U = `
|
|
204
204
|
vec4 aLessThanZero = vec4(lessThan(a, vec4(0.)));
|
|
205
205
|
return (aLessThanZero * (b * a)) + ((vec4(1.0) - aLessThanZero) * a);
|
|
206
206
|
`;
|
|
207
207
|
function ue(t) {
|
|
208
|
-
const { inputs: e, backend: s } = t, { x: r, alpha: u } = e, n = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(
|
|
208
|
+
const { inputs: e, backend: s } = t, { x: r, alpha: u } = e, n = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(U, r.shape, u.shape) : new b(k, r.shape, u.shape);
|
|
209
209
|
return s.runWebGLProgram(n, [r, u], "float32");
|
|
210
210
|
}
|
|
211
211
|
const Ne = {
|
|
@@ -234,7 +234,7 @@ function Ie({ opSnippet: t, packedOpSnippet: e, checkOutOfBounds: s = !1, suppor
|
|
|
234
234
|
[h.complexTensorInfos.real, f.complexTensorInfos.real],
|
|
235
235
|
[h.complexTensorInfos.imag, f.complexTensorInfos.imag]
|
|
236
236
|
].map((C) => {
|
|
237
|
-
const [p, x] = C,
|
|
237
|
+
const [p, x] = C, $ = {
|
|
238
238
|
dataId: p.dataId,
|
|
239
239
|
dtype: p.dtype,
|
|
240
240
|
shape: a.shape
|
|
@@ -243,7 +243,7 @@ function Ie({ opSnippet: t, packedOpSnippet: e, checkOutOfBounds: s = !1, suppor
|
|
|
243
243
|
dtype: x.dtype,
|
|
244
244
|
shape: c.shape
|
|
245
245
|
}, D = new b(t, a.shape, c.shape);
|
|
246
|
-
return l.runWebGLProgram(D, [
|
|
246
|
+
return l.runWebGLProgram(D, [$, _], S(p.dtype, x.dtype));
|
|
247
247
|
}), A = L({ inputs: { real: g, imag: m }, backend: l });
|
|
248
248
|
return l.disposeIntermediateTensorInfo(g), l.disposeIntermediateTensorInfo(m), A;
|
|
249
249
|
}
|
|
@@ -273,7 +273,7 @@ function Ae(t, e = !1) {
|
|
|
273
273
|
if (t === "relu6")
|
|
274
274
|
return e ? ae : Q;
|
|
275
275
|
if (t === "prelu")
|
|
276
|
-
return e ?
|
|
276
|
+
return e ? U : k;
|
|
277
277
|
if (t === "leakyrelu")
|
|
278
278
|
return e ? R : w;
|
|
279
279
|
if (t === "sigmoid")
|
package/dist/layers/BaseLayer.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { v as V } from "../variable-
|
|
1
|
+
import { w as p, a3 as g, a1 as v, e as o, a4 as _ } from "../index-Duu1Lvvv.js";
|
|
2
|
+
import { v as V } from "../variable-DPt_Iuog.js";
|
|
3
3
|
class T {
|
|
4
4
|
parent;
|
|
5
5
|
config;
|
|
@@ -1,18 +1,18 @@
|
|
|
1
1
|
import { attentionMask as P } from "../ops/attentionMask.js";
|
|
2
2
|
import _ from "./BaseLayer.js";
|
|
3
|
-
import { r as k } from "../rope-
|
|
3
|
+
import { r as k } from "../rope-DJ7Y7c-u.js";
|
|
4
4
|
import { appendCache as b } from "../ops/appendCache.js";
|
|
5
|
-
import { k as u, t as S } from "../index-
|
|
5
|
+
import { k as u, t as S } from "../index-Duu1Lvvv.js";
|
|
6
6
|
import { softmax16 as C } from "../ops/softmax16.js";
|
|
7
|
-
import { b as R } from "../matMul16-
|
|
8
|
-
import { p as V } from "../pack16-
|
|
7
|
+
import { b as R } from "../matMul16-xswmhSuF.js";
|
|
8
|
+
import { p as V } from "../pack16-atD0eYRm.js";
|
|
9
9
|
import { transpose16 as J } from "../ops/transpose16.js";
|
|
10
10
|
import { dot16 as L } from "../ops/dot16.js";
|
|
11
11
|
import { reshape16 as N } from "../ops/reshape16.js";
|
|
12
12
|
import { isPackedTensor as l } from "../utilities/packed.js";
|
|
13
13
|
import { qkv as j } from "../ops/qkv.js";
|
|
14
|
-
import { v as T } from "../variable-
|
|
15
|
-
import { r as v, d as y } from "../dropout-
|
|
14
|
+
import { v as T } from "../variable-DPt_Iuog.js";
|
|
15
|
+
import { r as v, d as y } from "../dropout-DLhSMNTZ.js";
|
|
16
16
|
class X extends _ {
|
|
17
17
|
divisor;
|
|
18
18
|
index;
|
package/dist/layers/MLP.js
CHANGED
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import { t as p } from "../index-
|
|
1
|
+
import { t as p } from "../index-Duu1Lvvv.js";
|
|
2
2
|
import u from "./BaseLayer.js";
|
|
3
|
-
import { c, b as M } from "../matMul16-
|
|
3
|
+
import { c, b as M } from "../matMul16-xswmhSuF.js";
|
|
4
4
|
import { reshape16 as o } from "../ops/reshape16.js";
|
|
5
|
-
import { v as d } from "../variable-
|
|
6
|
-
import { r as h, d as b } from "../dropout-
|
|
5
|
+
import { v as d } from "../variable-DPt_Iuog.js";
|
|
6
|
+
import { r as h, d as b } from "../dropout-DLhSMNTZ.js";
|
|
7
7
|
class O extends u {
|
|
8
8
|
index;
|
|
9
9
|
hiddenUnits;
|