@genai-fi/nanogpt 0.18.1 → 0.19.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 +30 -30
- package/dist/{RealDiv-ioj6Z-ox.js → RealDiv-CGwv0liw.js} +9 -9
- package/dist/{Reshape-BZC-ebeR.js → Reshape-BW__R4mZ.js} +7 -7
- package/dist/{Reshape-pwprEaej.js → Reshape-CPBkTIH2.js} +1 -1
- package/dist/TeachableLLM.js +17 -17
- package/dist/Trainer.js +97 -95
- package/dist/{axis_util-QWWgLjut.js → axis_util-GTVlo58H.js} +1 -1
- package/dist/backend.js +2 -2
- package/dist/{backend_util-qwSFfxYx.js → backend_util-GaFarB78.js} +21 -21
- package/dist/{backend_webgpu-DI2wXEC2.js → backend_webgpu-BqASlsbV.js} +8 -8
- package/dist/{broadcast_to-C_EJTVTZ.js → broadcast_to-eS93CCN_.js} +2 -2
- package/dist/checks/appendCache.js +2 -2
- package/dist/checks/attentionMask.js +5 -5
- package/dist/checks/gelu.js +2 -2
- package/dist/checks/matMulGelu.js +2 -2
- package/dist/checks/normRMS.js +6 -6
- package/dist/checks/normRMSGrad.js +3 -3
- package/dist/checks/packUnpack.js +6 -6
- package/dist/checks/qkv.js +2 -2
- package/dist/checks/rope.js +2 -2
- package/dist/{clip_by_value-CLAD4h_I.js → clip_by_value-DDA7rrcT.js} +1 -1
- package/dist/complex-DI35Q-gW.js +11 -0
- package/dist/{concat-Dqk7Xk7h.js → concat-CAQpCret.js} +5 -5
- package/dist/{concat_util-C1Mxe27t.js → concat_util-D18dJ4fD.js} +1 -1
- package/dist/data/textLoader.d.ts +1 -1
- package/dist/data/textLoader.js +68 -58
- package/dist/{dataset-DlqAN81i.js → dataset-CGGp1z9P.js} +3 -3
- package/dist/{dropout_util-N0z8Os-K.js → dropout_util--NxWuYg2.js} +1 -1
- package/dist/{expand_dims-D0rBtgT1.js → expand_dims-Bkd1YD5x.js} +4 -4
- package/dist/{exports_initializers-DIOZQt_L.js → exports_initializers-CYzKLjN7.js} +1 -1
- package/dist/{floor-CymuCmTO.js → floor-BQtb-Azg.js} +1 -1
- package/dist/{gather-DEyjXNb1.js → gather-qIqEqaGn.js} +1 -1
- package/dist/{gelu-DpTCC3eB.js → gelu-B220X1Go.js} +1 -1
- package/dist/{gpgpu_math-3bCb5ooU.js → gpgpu_math-BwvV12df.js} +25 -25
- package/dist/{index-DSGwv2Yx.js → index-CUXkjxiT.js} +33 -33
- package/dist/{index-BQvB7LCC.js → index-CjOWnMXP.js} +15 -15
- package/dist/{kernel_funcs_utils-DGqzNlHT.js → kernel_funcs_utils-pq0CK9co.js} +6 -6
- package/dist/layers/BaseLayer.js +4 -4
- package/dist/layers/CausalSelfAttention.d.ts +1 -0
- package/dist/layers/CausalSelfAttention.js +14 -14
- package/dist/layers/LoRA.js +4 -4
- 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.d.ts +1 -0
- package/dist/layers/TransformerBlock.js +1 -1
- package/dist/layers/WeightStore.js +2 -2
- package/dist/loader/loadTransformers.js +1 -1
- package/dist/loader/oldZipLoad.js +9 -9
- package/dist/loader/save.js +38 -31
- package/dist/main.js +9 -9
- package/dist/{matMul16-BIT70Vya.js → matMul16-BcVC_E62.js} +3 -3
- package/dist/{matMulGelu-CsZnh18H.js → matMulGelu-JNLZqKQp.js} +18 -18
- package/dist/mat_mul-DhG0Newp.js +11 -0
- package/dist/mod-CSdCpRjf.js +11 -0
- package/dist/models/NanoGPTV1.js +2 -2
- package/dist/models/NanoGPTV2.js +2 -2
- package/dist/models/model.d.ts +1 -0
- package/dist/models/model.js +9 -9
- package/dist/{not_equal-CkQKkKZy.js → not_equal-hurPF26l.js} +15 -15
- package/dist/{ones-DbVB5N58.js → ones-BytntneX.js} +3 -3
- package/dist/ops/adamAdjust.js +3 -3
- package/dist/ops/adamMoments.js +3 -3
- package/dist/ops/add16.js +1 -1
- package/dist/ops/appendCache.js +6 -6
- package/dist/ops/attentionMask.js +3 -3
- package/dist/ops/concat16.js +3 -3
- package/dist/ops/cpu/adamAdjust.js +9 -9
- package/dist/ops/cpu/adamMoments.js +5 -5
- package/dist/ops/cpu/appendCache.js +2 -2
- package/dist/ops/cpu/attentionMask.js +6 -6
- package/dist/ops/cpu/fusedSoftmax.js +4 -4
- package/dist/ops/cpu/gatherSub.js +5 -5
- package/dist/ops/cpu/gelu.js +4 -4
- package/dist/ops/cpu/matMul16.js +2 -2
- package/dist/ops/cpu/matMulGelu.js +7 -7
- package/dist/ops/cpu/matMulMul.js +2 -2
- package/dist/ops/cpu/mulDropout.js +5 -5
- 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 +5 -5
- package/dist/ops/dot16.js +2 -2
- package/dist/ops/dropout.js +6 -6
- package/dist/ops/dropout16.js +1 -1
- package/dist/ops/gatherSub.js +1 -1
- package/dist/ops/gelu.js +2 -2
- package/dist/ops/globalNorm.js +7 -7
- package/dist/ops/grads/add16.js +1 -1
- package/dist/ops/grads/attentionMask.js +2 -2
- package/dist/ops/grads/dropout16.js +1 -1
- package/dist/ops/grads/gelu.js +2 -2
- package/dist/ops/grads/matMul16.js +3 -3
- package/dist/ops/grads/matMulGelu.js +1 -1
- package/dist/ops/grads/mul16.js +1 -1
- package/dist/ops/grads/normRMS.js +7 -7
- package/dist/ops/grads/pack16.js +3 -3
- package/dist/ops/grads/qkv.js +11 -11
- 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 +6 -6
- package/dist/ops/matMulMul.js +3 -3
- package/dist/ops/mul16.js +1 -1
- package/dist/ops/mulDrop.js +3 -3
- package/dist/ops/normRMS.js +4 -4
- package/dist/ops/pack16.js +2 -2
- package/dist/ops/qkv.js +3 -3
- 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 +6 -6
- package/dist/ops/transpose16.js +3 -3
- 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 +1 -1
- package/dist/ops/webgl/dropout16.js +1 -1
- package/dist/ops/webgl/fusedSoftmax.js +7 -7
- package/dist/ops/webgl/gatherSub.js +3 -3
- package/dist/ops/webgl/gelu.js +2 -2
- package/dist/ops/webgl/log.js +3 -3
- package/dist/ops/webgl/matMul16.js +13 -13
- 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 +1 -1
- package/dist/ops/webgl/scatterSub.js +2 -2
- package/dist/ops/webgpu/adamAdjust.js +3 -3
- package/dist/ops/webgpu/adamMoments.js +3 -3
- package/dist/ops/webgpu/add16.js +6 -6
- package/dist/ops/webgpu/appendCache.js +3 -3
- package/dist/ops/webgpu/attentionMask.js +2 -2
- package/dist/ops/webgpu/attentionMask32_program.js +2 -2
- package/dist/ops/webgpu/clipScale.js +7 -7
- package/dist/ops/webgpu/concat16.js +5 -5
- package/dist/ops/webgpu/dropout16.js +6 -6
- package/dist/ops/webgpu/gatherSub.js +3 -3
- package/dist/ops/webgpu/gelu.js +8 -8
- package/dist/ops/webgpu/matMul16.js +16 -16
- package/dist/ops/webgpu/matMul16_program.js +2 -2
- package/dist/ops/webgpu/mul16.js +5 -5
- package/dist/ops/webgpu/norm2.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 +4 -4
- package/dist/ops/webgpu/pack16_program.js +2 -2
- package/dist/ops/webgpu/qkv.js +2 -2
- 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 +4 -4
- package/dist/ops/webgpu/sub16.js +6 -6
- package/dist/ops/webgpu/sum16.js +3 -3
- package/dist/ops/webgpu/transpose16.js +8 -8
- 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 +3 -3
- package/dist/ops/webgpu/utils/binary_op.js +3 -3
- package/dist/ops/webgpu/utils/reductions.js +5 -5
- package/dist/{ops-CURIZSVt.js → ops-CsXeTq1P.js} +100 -100
- package/dist/{pack16-WlOSOuZA.js → pack16-bqltoUlR.js} +2 -2
- package/dist/patches/webgpu_backend.js +6 -6
- package/dist/patches/webgpu_base.js +1 -1
- package/dist/patches/webgpu_program.js +2 -2
- package/dist/{random_normal-CIm8lk2-.js → random_normal-IBRrha8a.js} +1 -1
- package/dist/{random_width-B_fVXhGx.js → random_width-DN5ZtQkM.js} +131 -131
- package/dist/{range-BDxO73mk.js → range-C-CjF-LI.js} +1 -1
- package/dist/relu-J_X6MUzx.js +9 -0
- package/dist/{reshape-BIN71H3p.js → reshape-BDOuCSNW.js} +1 -1
- package/dist/{resize_nearest_neighbor-C6_0dAnK.js → resize_nearest_neighbor-BojqlfRe.js} +41 -41
- package/dist/{rope-CC5RjmKU.js → rope-DcrZM_e6.js} +5 -5
- package/dist/{scatter_nd_util-C-x73Cj6.js → scatter_nd_util-ByNJaL6I.js} +1 -1
- package/dist/{segment_util-4zuHV5IG.js → segment_util-Dasb2Zaf.js} +2 -2
- package/dist/{selu_util-BXdhy_W6.js → selu_util-BLhIqRkw.js} +5 -5
- package/dist/{shared-zTaJ5siv.js → shared-3agzAqQ_.js} +1 -1
- package/dist/{shared-DRWDyk9w.js → shared-CagdqkLh.js} +6 -6
- package/dist/slice-BzS11Qh0.js +12 -0
- package/dist/{slice_util-DPY56GzQ.js → slice_util-CC35pLmT.js} +5 -5
- package/dist/{softmax-BLGJqdwx.js → softmax-D4q1LJN7.js} +1 -1
- package/dist/split-C2Sj255c.js +9 -0
- package/dist/{squeeze-O_YWJpw_.js → squeeze-ho4wLUek.js} +2 -2
- package/dist/{stack-z6QE7kmP.js → stack-DudVrtmG.js} +1 -1
- package/dist/{step-DQY6_ABw.js → step-BTxPtq1r.js} +4 -4
- package/dist/{sum-D39FeU5h.js → sum-BpiwSWvg.js} +3 -3
- package/dist/{tensor-D8e0Gd7c.js → tensor-BWFldCso.js} +1 -1
- package/dist/{tensor1d-BMl0eZYV.js → tensor1d-LMGMIUlr.js} +1 -1
- package/dist/{tensor2d-DTtQ1QcT.js → tensor2d-BnXMKScO.js} +1 -1
- package/dist/{tensor4d-Dj4rDssL.js → tensor4d-C6UCG_u8.js} +1 -1
- package/dist/{tfjs_backend-Bk3PmK91.js → tfjs_backend-BGnG-ppu.js} +65 -65
- package/dist/{tile-CsWlVKKz.js → tile-CFy-xTO6.js} +1 -1
- package/dist/tokeniser/BaseTokeniser.d.ts +4 -0
- package/dist/tokeniser/BaseTokeniser.js +61 -56
- package/dist/tokeniser/type.d.ts +8 -0
- package/dist/training/AdamW.js +2 -2
- package/dist/training/BasicTrainer.d.ts +1 -0
- package/dist/training/BasicTrainer.js +95 -79
- package/dist/training/DatasetBuilder.d.ts +6 -2
- package/dist/training/DatasetBuilder.js +60 -41
- package/dist/training/Evaluator.d.ts +1 -2
- package/dist/training/Evaluator.js +21 -31
- package/dist/training/SFTTrainer.d.ts +3 -2
- package/dist/training/SFTTrainer.js +4 -3
- package/dist/training/orthoGrad.js +1 -1
- package/dist/training/sparseCrossEntropy.js +38 -38
- package/dist/training/tasks/ConversationTask.d.ts +4 -0
- package/dist/training/tasks/ConversationTask.js +7 -7
- package/dist/training/tasks/PretrainingTask.d.ts +4 -0
- package/dist/training/tasks/PretrainingTask.js +11 -7
- package/dist/training/tasks/StartSentenceTask.d.ts +4 -0
- package/dist/training/tasks/StartSentenceTask.js +5 -5
- package/dist/training/tasks/Task.d.ts +12 -0
- package/dist/training/tasks/Task.js +55 -31
- package/dist/training/types.d.ts +1 -0
- package/dist/training/validation.d.ts +1 -1
- package/dist/training/validation.js +33 -32
- package/dist/{transpose-Qxz-4os3.js → transpose-9kRxIXWR.js} +7 -7
- package/dist/{unsorted_segment_sum-BfFVV9Zm.js → unsorted_segment_sum-DJvk5xnh.js} +20 -20
- package/dist/utilities/dummy.js +6 -6
- 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.js +5 -5
- package/dist/utilities/weights.js +2 -2
- package/dist/{variable-SSATClyt.js → variable-Ck482e3n.js} +1 -1
- package/dist/{webgpu_program-CbjdYLYk.js → webgpu_program-B4HmApL1.js} +1 -1
- package/dist/{webgpu_util-DuofJBMo.js → webgpu_util-DYlGSwOJ.js} +7 -7
- package/dist/{zeros-Bw0puq_w.js → zeros-DvZpK8s6.js} +2 -2
- package/dist/{zeros_like-rOHr54NY.js → zeros_like-CWjDdwr-.js} +69 -69
- package/package.json +1 -1
- package/dist/complex-3DpPEG9B.js +0 -11
- package/dist/mat_mul-DP86qZtZ.js +0 -11
- package/dist/mod-BXjLYwvM.js +0 -11
- package/dist/readers-17HLdxVM.js +0 -12
- package/dist/relu-DTvZKBsZ.js +0 -9
- package/dist/slice-BvItlgXu.js +0 -12
- package/dist/split-BN9LkEgS.js +0 -9
- package/dist/training/SFTDatasetBuilder.d.ts +0 -23
- package/dist/training/SFTDatasetBuilder.js +0 -85
|
@@ -1,9 +1,9 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { d as T, f as L, h as
|
|
3
|
-
import { a as z, c as
|
|
4
|
-
import { c as
|
|
5
|
-
import { S as k, a as Q, b as ee, g as te, c as se, s as ne } from "./selu_util-
|
|
6
|
-
import { c as re, v as oe, a as ie } from "./scatter_nd_util-
|
|
1
|
+
import { U as d, a9 as A, a8 as O, x as g, av as _, az as w, ad as D, _ as x, $ as b, am as y, aY as M } from "./index-CUXkjxiT.js";
|
|
2
|
+
import { d as T, f as L, h as W, c as v, e as F, a as N, b as C, g as P } from "./axis_util-GTVlo58H.js";
|
|
3
|
+
import { a as z, c as U } from "./concat_util-D18dJ4fD.js";
|
|
4
|
+
import { c as B, b as H, d as V, f as G, g as Z, h as j, i as q, j as J, k as K, m as X, t as Y } from "./step-BTxPtq1r.js";
|
|
5
|
+
import { S as k, a as Q, b as ee, g as te, c as se, s as ne } from "./selu_util-BLhIqRkw.js";
|
|
6
|
+
import { c as re, v as oe, a as ie } from "./scatter_nd_util-ByNJaL6I.js";
|
|
7
7
|
import { a as ae, c as ue, b as ce, e as pe, d as le, g as fe, m as he, s as ge } from "./complex_util-Yc1A_gV1.js";
|
|
8
8
|
function de(e, t) {
|
|
9
9
|
const r = e.shape.length, n = t.shape.length;
|
|
@@ -146,10 +146,10 @@ function De(e, t, r) {
|
|
|
146
146
|
return n;
|
|
147
147
|
}
|
|
148
148
|
const xe = 0.3275911, be = 0.254829592, ye = -0.284496736, Me = 1.421413741, Te = -1.453152027, Le = 1.061405429;
|
|
149
|
-
const I = "->",
|
|
150
|
-
function
|
|
149
|
+
const I = "->", We = /->/g, E = ",", $ = "...";
|
|
150
|
+
function ve(e, t) {
|
|
151
151
|
e = e.replace(/\s/g, "");
|
|
152
|
-
const r = (e.length - e.replace(
|
|
152
|
+
const r = (e.length - e.replace(We, "").length) / I.length;
|
|
153
153
|
if (r < 1)
|
|
154
154
|
throw new Error("Equations without an arrow are not supported.");
|
|
155
155
|
if (r > 1)
|
|
@@ -226,7 +226,7 @@ function ze(e, t) {
|
|
|
226
226
|
(e[n].length === 0 || e[n].indexOf(t) !== -1 || t === -1) && r.push(n);
|
|
227
227
|
return r;
|
|
228
228
|
}
|
|
229
|
-
function
|
|
229
|
+
function Ue(e, t, r = 0) {
|
|
230
230
|
let n = [];
|
|
231
231
|
if (typeof t == "number")
|
|
232
232
|
g(e.shape[r] % t === 0, () => "Number of splits must evenly divide the axis."), n = new Array(t).fill(e.shape[r] / t);
|
|
@@ -242,7 +242,7 @@ function Be(e, t, r = 0) {
|
|
|
242
242
|
}
|
|
243
243
|
return n;
|
|
244
244
|
}
|
|
245
|
-
function
|
|
245
|
+
function Be(e) {
|
|
246
246
|
return `Received SparseTensor with denseShape[0] = 0 but
|
|
247
247
|
indices.shape[0] = ${e}`;
|
|
248
248
|
}
|
|
@@ -314,8 +314,8 @@ const ut = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
|
|
|
314
314
|
axesAreInnerMostDims: L,
|
|
315
315
|
calculateShapes: re,
|
|
316
316
|
checkEinsumDimSizes: Ne,
|
|
317
|
-
checkPadOnDimRoundingMode:
|
|
318
|
-
combineLocations:
|
|
317
|
+
checkPadOnDimRoundingMode: B,
|
|
318
|
+
combineLocations: W,
|
|
319
319
|
combineRaggedTensorToTensorShapes: me,
|
|
320
320
|
complexWithEvenIndex: ue,
|
|
321
321
|
complexWithOddIndex: ce,
|
|
@@ -324,12 +324,12 @@ const ut = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
|
|
|
324
324
|
computeDefaultPad: G,
|
|
325
325
|
computeDilation2DInfo: Z,
|
|
326
326
|
computeOptimalWindowSize: Se,
|
|
327
|
-
computeOutAndReduceShapes:
|
|
328
|
-
computeOutShape:
|
|
327
|
+
computeOutAndReduceShapes: v,
|
|
328
|
+
computeOutShape: U,
|
|
329
329
|
computePool2DInfo: j,
|
|
330
330
|
computePool3DInfo: q,
|
|
331
331
|
convertConv2DDataFormat: J,
|
|
332
|
-
decodeEinsumEquation:
|
|
332
|
+
decodeEinsumEquation: ve,
|
|
333
333
|
eitherStridesOrDilationsAreOne: K,
|
|
334
334
|
expandShapeToKeepDim: F,
|
|
335
335
|
exponent: pe,
|
|
@@ -353,7 +353,7 @@ const ut = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
|
|
|
353
353
|
getRowPartitionTypesHelper: Ie,
|
|
354
354
|
getSliceBeginCoords: we,
|
|
355
355
|
getSliceSize: De,
|
|
356
|
-
getSparseFillEmptyRowsIndicesDenseShapeMismatch:
|
|
356
|
+
getSparseFillEmptyRowsIndicesDenseShapeMismatch: Be,
|
|
357
357
|
getSparseFillEmptyRowsNegativeIndexErrorMessage: He,
|
|
358
358
|
getSparseFillEmptyRowsOutOfRangeIndexErrorMessage: Ve,
|
|
359
359
|
getSparseReshapeEmptyTensorZeroOutputDimErrorMessage: je,
|
|
@@ -369,7 +369,7 @@ const ut = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
|
|
|
369
369
|
isIdentityPermutation: Pe,
|
|
370
370
|
mergeRealAndImagArrays: he,
|
|
371
371
|
prepareAndValidate: de,
|
|
372
|
-
prepareSplitSize:
|
|
372
|
+
prepareSplitSize: Ue,
|
|
373
373
|
shouldFuse: ne,
|
|
374
374
|
splitRealAndImagArrays: ge,
|
|
375
375
|
stridesOrDilationsArePositive: X,
|
|
@@ -386,14 +386,14 @@ export {
|
|
|
386
386
|
we as C,
|
|
387
387
|
De as D,
|
|
388
388
|
xe as E,
|
|
389
|
-
|
|
389
|
+
ve as F,
|
|
390
390
|
Ne as G,
|
|
391
391
|
Ce as H,
|
|
392
392
|
Fe as I,
|
|
393
393
|
Pe as J,
|
|
394
394
|
de as K,
|
|
395
395
|
Re as L,
|
|
396
|
-
|
|
396
|
+
Ue as M,
|
|
397
397
|
S as P,
|
|
398
398
|
f as R,
|
|
399
399
|
Ee as a,
|
|
@@ -403,7 +403,7 @@ export {
|
|
|
403
403
|
et as e,
|
|
404
404
|
Qe as f,
|
|
405
405
|
Ie as g,
|
|
406
|
-
|
|
406
|
+
Be as h,
|
|
407
407
|
He as i,
|
|
408
408
|
Ve as j,
|
|
409
409
|
Ge as k,
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import { ab as g, as as $, at as K,
|
|
2
|
-
import { m as te, f as se, P as re } from "./webgpu_program-
|
|
3
|
-
import { i as ne, G as
|
|
4
|
-
import { m as
|
|
1
|
+
import { ab as g, as as $, at as K, h as D, x as _, au as O, U as x, av as Z, a5 as W, aw as F, ax as j, ay as X, az as J, af as ee, a9 as k } from "./index-CUXkjxiT.js";
|
|
2
|
+
import { m as te, f as se, P as re } from "./webgpu_program-B4HmApL1.js";
|
|
3
|
+
import { i as ne, G as q } from "./webgpu_util-DYlGSwOJ.js";
|
|
4
|
+
import { m as N } from "./complex_util-Yc1A_gV1.js";
|
|
5
5
|
const d = g();
|
|
6
6
|
d.registerFlag("WEBGPU_DEFERRED_SUBMIT_BATCH_SIZE", () => 15);
|
|
7
7
|
d.registerFlag("WEBGPU_CPU_FORWARD", () => !0);
|
|
@@ -248,7 +248,7 @@ class R extends $ {
|
|
|
248
248
|
if (s != null || t.dtype === "string")
|
|
249
249
|
return s;
|
|
250
250
|
if (t.dtype === "complex64") {
|
|
251
|
-
const E = this.readSync(n.real.dataId), B = this.readSync(n.imag.dataId), y = O(
|
|
251
|
+
const E = this.readSync(n.real.dataId), B = this.readSync(n.imag.dataId), y = O(N(E, B).buffer, "float32");
|
|
252
252
|
return this.convertAndCacheOnCPU(e, y), y;
|
|
253
253
|
}
|
|
254
254
|
this.hasReadSyncWarned || (this.hasReadSyncWarned = !0, console.warn("The performance of synchronously reading data from GPU to CPU is poor on the webgpu backend, please use asynchronous APIs instead."));
|
|
@@ -309,7 +309,7 @@ class R extends $ {
|
|
|
309
309
|
this.read(t.complexTensorInfos.real.dataId),
|
|
310
310
|
this.read(t.complexTensorInfos.imag.dataId)
|
|
311
311
|
]), a = r[0], i = r[1];
|
|
312
|
-
n =
|
|
312
|
+
n = N(a, i);
|
|
313
313
|
} else {
|
|
314
314
|
const r = await this.getBufferData(t.resource);
|
|
315
315
|
n = O(r, t.dtype);
|
|
@@ -337,7 +337,7 @@ class R extends $ {
|
|
|
337
337
|
refCount: 1,
|
|
338
338
|
external: e.zeroCopy
|
|
339
339
|
});
|
|
340
|
-
const a = this.tensorMap.get(r), i =
|
|
340
|
+
const a = this.tensorMap.get(r), i = q(a.dtype) * x(a.shape);
|
|
341
341
|
if (e.buffer.size < i)
|
|
342
342
|
throw new Error(`GPUBuffer size(${e.buffer.size}) is smaller than tensor size(${i})!`);
|
|
343
343
|
if ((e.buffer.usage & (GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC)) !== (GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC))
|
|
@@ -398,7 +398,7 @@ class R extends $ {
|
|
|
398
398
|
const t = this.tensorMap.get(e);
|
|
399
399
|
if (t.resource != null)
|
|
400
400
|
return;
|
|
401
|
-
const s =
|
|
401
|
+
const s = q(t.dtype) * x(t.shape);
|
|
402
402
|
let n;
|
|
403
403
|
const r = GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST;
|
|
404
404
|
if (t.values) {
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { o as h,
|
|
2
|
-
import { r as b } from "./reshape-
|
|
1
|
+
import { o as h, q as f, u as p, w as g, E as u, T } from "./index-CUXkjxiT.js";
|
|
2
|
+
import { r as b } from "./reshape-BDOuCSNW.js";
|
|
3
3
|
function m(e, r) {
|
|
4
4
|
let n = f(e, "broadcastTo", "x");
|
|
5
5
|
const a = n.shape;
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { s as i,
|
|
2
|
-
import { t } from "../tensor4d-
|
|
3
|
-
import { t as
|
|
1
|
+
import { s as i, h as a } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t } from "../tensor4d-C6UCG_u8.js";
|
|
3
|
+
import { t as e } from "../tensor2d-BnXMKScO.js";
|
|
4
4
|
async function k(n) {
|
|
5
5
|
await i(n);
|
|
6
6
|
const s = t(
|
|
@@ -23,14 +23,14 @@ async function k(n) {
|
|
|
23
23
|
]
|
|
24
24
|
],
|
|
25
25
|
[1, 1, 2, 4]
|
|
26
|
-
), r =
|
|
26
|
+
), r = e(
|
|
27
27
|
[
|
|
28
28
|
[0, -1 / 0, -1 / 0, -1 / 0],
|
|
29
29
|
[0, 0, 0, -1 / 0]
|
|
30
30
|
],
|
|
31
31
|
[2, 4]
|
|
32
32
|
);
|
|
33
|
-
return await
|
|
33
|
+
return await a().runKernel("AttentionMask", { q: s, k: o, mask: r }, { divisor: 0.5, pastLen: 0 }).array();
|
|
34
34
|
}
|
|
35
35
|
export {
|
|
36
36
|
k as execute
|
package/dist/checks/gelu.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { s as e,
|
|
2
|
-
import { t as s } from "../tensor2d-
|
|
1
|
+
import { s as e, h as o } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t as s } from "../tensor2d-BnXMKScO.js";
|
|
3
3
|
async function m(t) {
|
|
4
4
|
await e(t);
|
|
5
5
|
const r = s(
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { s as o,
|
|
2
|
-
import { t as e } from "../tensor2d-
|
|
1
|
+
import { s as o, h as s } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t as e } from "../tensor2d-BnXMKScO.js";
|
|
3
3
|
async function i(t) {
|
|
4
4
|
await o(t);
|
|
5
5
|
const r = e(
|
package/dist/checks/normRMS.js
CHANGED
|
@@ -1,13 +1,13 @@
|
|
|
1
|
-
import { s as u, a0 as A,
|
|
2
|
-
import { a as
|
|
3
|
-
import { t as p } from "../tensor1d-
|
|
4
|
-
import { t as r } from "../tensor-
|
|
1
|
+
import { s as u, a0 as A, h } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { a as y } from "../ops-CsXeTq1P.js";
|
|
3
|
+
import { t as p } from "../tensor1d-LMGMIUlr.js";
|
|
4
|
+
import { t as r } from "../tensor-BWFldCso.js";
|
|
5
5
|
const w = Array.from({ length: 2048 * 192 }, () => Math.random()), x = Array.from({ length: 192 }, () => Math.random()), M = Array.from({ length: 2048 * 192 }, () => Math.random());
|
|
6
6
|
async function k(t) {
|
|
7
7
|
await u(t);
|
|
8
8
|
const o = p(x, "float32"), n = r(w, [16, 128, 192], "float32"), s = r(M, [16, 128, 192], "float32"), e = (d, g) => {
|
|
9
|
-
const i =
|
|
10
|
-
return
|
|
9
|
+
const i = h().runKernel("RMSNorm", { x: d, gamma: g });
|
|
10
|
+
return y.meanSquaredError(i, s);
|
|
11
11
|
}, { value: m, grads: a } = A(e)([n, o]), c = await m.array(), f = await a[0].array(), l = await a[1].array();
|
|
12
12
|
return [c, f, l];
|
|
13
13
|
}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { s as c,
|
|
2
|
-
import { t as f } from "../tensor1d-
|
|
3
|
-
import { t as r } from "../tensor-
|
|
1
|
+
import { s as c, h as d } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t as f } from "../tensor1d-LMGMIUlr.js";
|
|
3
|
+
import { t as r } from "../tensor-BWFldCso.js";
|
|
4
4
|
const y = Array.from({ length: 2048 * 192 }, () => Math.random()), i = Array.from({ length: 192 }, () => Math.random()), l = Array.from({ length: 2048 * 192 }, () => Math.random());
|
|
5
5
|
async function x(t) {
|
|
6
6
|
await c(t);
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import { s as a,
|
|
2
|
-
import { t as c } from "../tensor2d-
|
|
3
|
-
async function i(
|
|
4
|
-
await a(
|
|
1
|
+
import { s as a, h as n } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t as c } from "../tensor2d-BnXMKScO.js";
|
|
3
|
+
async function i(e) {
|
|
4
|
+
await a(e);
|
|
5
5
|
const r = c(
|
|
6
6
|
[
|
|
7
7
|
[0.1, 0.2, 0, 0, 1230, 1232331234, -12234234],
|
|
@@ -10,8 +10,8 @@ async function i(n) {
|
|
|
10
10
|
[0, 0, 0, 0, -0.1, 1e-3, 0]
|
|
11
11
|
],
|
|
12
12
|
[4, 7]
|
|
13
|
-
), t =
|
|
14
|
-
return await
|
|
13
|
+
), t = n().runKernel("Pack16", { x: r });
|
|
14
|
+
return await n().runKernel("Unpack16", { x: t }).array();
|
|
15
15
|
}
|
|
16
16
|
export {
|
|
17
17
|
i as execute
|
package/dist/checks/qkv.js
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import { t as f } from "../tensor2d-
|
|
1
|
+
import { W as i, X as u, Y as c, s as l, h } from "../index-CUXkjxiT.js";
|
|
2
|
+
import { t as f } from "../tensor2d-BnXMKScO.js";
|
|
3
3
|
function m(t, e, n) {
|
|
4
4
|
if (i(t), e != null && e.length !== 3)
|
|
5
5
|
throw new Error("tensor3d() requires shape to have three numbers");
|
package/dist/checks/rope.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import s from "../layers/RoPECache.js";
|
|
2
|
-
import { s as c,
|
|
3
|
-
import { t as p } from "../tensor4d-
|
|
2
|
+
import { s as c, h as i } from "../index-CUXkjxiT.js";
|
|
3
|
+
import { t as p } from "../tensor4d-C6UCG_u8.js";
|
|
4
4
|
async function f(r) {
|
|
5
5
|
await c(r);
|
|
6
6
|
const n = p(
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { o as a,
|
|
1
|
+
import { o as a, q as f, x as m, f as y, E as c, C as p } from "./index-CUXkjxiT.js";
|
|
2
2
|
function u(o, s, t) {
|
|
3
3
|
const r = f(o, "x", "clipByValue");
|
|
4
4
|
if (m(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { o as c, q as s, a6 as n, E as m, a7 as r } from "./index-CUXkjxiT.js";
|
|
2
|
+
function l(o, p) {
|
|
3
|
+
const a = s(o, "real", "complex"), e = s(p, "imag", "complex");
|
|
4
|
+
n(a.shape, e.shape, `real and imag shapes, ${a.shape} and ${e.shape}, must match in call to tf.complex().`);
|
|
5
|
+
const t = { real: a, imag: e };
|
|
6
|
+
return m.runKernel(r, t);
|
|
7
|
+
}
|
|
8
|
+
const i = /* @__PURE__ */ c({ complex_: l });
|
|
9
|
+
export {
|
|
10
|
+
i as c
|
|
11
|
+
};
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { o as s,
|
|
2
|
-
function
|
|
1
|
+
import { o as s, x as a, y as p, w as i, E as l, z as f } from "./index-CUXkjxiT.js";
|
|
2
|
+
function h(o, e = 0) {
|
|
3
3
|
a(o.length >= 1, () => "Pass at least one tensor to concat");
|
|
4
4
|
const t = p(o, "tensors", "concat", "string_or_numeric");
|
|
5
5
|
if (t[0].dtype === "complex64" && t.forEach((n) => {
|
|
@@ -9,9 +9,9 @@ function f(o, e = 0) {
|
|
|
9
9
|
}), t.length === 1)
|
|
10
10
|
return i(t[0]);
|
|
11
11
|
const r = t, c = { axis: e };
|
|
12
|
-
return l.runKernel(
|
|
12
|
+
return l.runKernel(f, r, c);
|
|
13
13
|
}
|
|
14
|
-
const
|
|
14
|
+
const u = /* @__PURE__ */ s({ concat_: h });
|
|
15
15
|
export {
|
|
16
|
-
|
|
16
|
+
u as c
|
|
17
17
|
};
|
|
@@ -4,4 +4,4 @@ export interface DataOptions {
|
|
|
4
4
|
column?: string;
|
|
5
5
|
hasHeader?: boolean;
|
|
6
6
|
}
|
|
7
|
-
export default function loadTextData(file: File, options?: DataOptions): Promise<Conversation[][]>;
|
|
7
|
+
export default function loadTextData(file: File, options?: DataOptions, cb?: (progress: number) => void): Promise<Conversation[][]>;
|
package/dist/data/textLoader.js
CHANGED
|
@@ -1,20 +1,20 @@
|
|
|
1
|
-
import { p as
|
|
2
|
-
import { loadParquet as
|
|
3
|
-
import { loadPDF as
|
|
4
|
-
import { loadDOCX as
|
|
5
|
-
import { z as
|
|
6
|
-
function
|
|
7
|
-
const
|
|
8
|
-
return
|
|
1
|
+
import { p as x } from "../papaparse.min-C0cScC2i.js";
|
|
2
|
+
import { loadParquet as m } from "./parquet.js";
|
|
3
|
+
import { loadPDF as y } from "./pdf.js";
|
|
4
|
+
import { loadDOCX as g } from "./docx.js";
|
|
5
|
+
import { z as w } from "../jszip.min-BZhlzntC.js";
|
|
6
|
+
function h(t, e) {
|
|
7
|
+
const o = t.findIndex((s) => s.toLowerCase() === e.toLowerCase());
|
|
8
|
+
return o === -1 ? 0 : o;
|
|
9
9
|
}
|
|
10
|
-
function
|
|
11
|
-
return t.every((
|
|
10
|
+
function z(t) {
|
|
11
|
+
return t.every((e) => e.length < 64);
|
|
12
12
|
}
|
|
13
|
-
function
|
|
13
|
+
function j(t) {
|
|
14
14
|
return t.split(".").pop() || "";
|
|
15
15
|
}
|
|
16
|
-
function
|
|
17
|
-
switch (
|
|
16
|
+
function v(t) {
|
|
17
|
+
switch (j(t)) {
|
|
18
18
|
case "json":
|
|
19
19
|
return "application/json";
|
|
20
20
|
case "jsonl":
|
|
@@ -35,74 +35,84 @@ function g(t) {
|
|
|
35
35
|
return "unknown";
|
|
36
36
|
}
|
|
37
37
|
}
|
|
38
|
-
function
|
|
38
|
+
function E(t) {
|
|
39
39
|
if (!Array.isArray(t)) return !1;
|
|
40
|
-
const
|
|
41
|
-
return typeof
|
|
40
|
+
const e = t[0];
|
|
41
|
+
return typeof e == "object" && e !== null && "role" in e && "content" in e && typeof e.role == "string" && typeof e.content == "string";
|
|
42
42
|
}
|
|
43
|
-
async function
|
|
44
|
-
const
|
|
45
|
-
if (
|
|
46
|
-
return
|
|
47
|
-
if (
|
|
48
|
-
return
|
|
49
|
-
if (
|
|
50
|
-
return
|
|
51
|
-
if (
|
|
52
|
-
const
|
|
53
|
-
if (Array.isArray(
|
|
54
|
-
return
|
|
55
|
-
typeof
|
|
43
|
+
async function O(t, e, o) {
|
|
44
|
+
const s = t.type !== "" ? t.type : v(t.name);
|
|
45
|
+
if (s === "application/parquet")
|
|
46
|
+
return m(t, e?.maxSize, e?.column);
|
|
47
|
+
if (s === "application/pdf")
|
|
48
|
+
return y(t, e?.maxSize);
|
|
49
|
+
if (s === "application/vnd.openxmlformats-officedocument.wordprocessingml.document")
|
|
50
|
+
return g(t);
|
|
51
|
+
if (s === "application/json") {
|
|
52
|
+
const c = await t.text(), a = JSON.parse(c);
|
|
53
|
+
if (Array.isArray(a))
|
|
54
|
+
return a.map((r) => [
|
|
55
|
+
typeof r == "string" ? { role: "text", content: r } : "text" in r ? { role: "text", content: r.text } : { role: "text", content: JSON.stringify(r) }
|
|
56
56
|
]);
|
|
57
57
|
throw new Error("Expected JSON array");
|
|
58
58
|
}
|
|
59
|
-
if (
|
|
60
|
-
|
|
61
|
-
|
|
59
|
+
if (s === "application/jsonl") {
|
|
60
|
+
const c = await t.text();
|
|
61
|
+
return o && o(0.1), c.split(`
|
|
62
|
+
`).filter((a) => a.trim() !== "").map((a, r, n) => {
|
|
63
|
+
o && r % 1e3 === 0 && o(0.1 + r / n.length * 0.9);
|
|
62
64
|
try {
|
|
63
|
-
const
|
|
64
|
-
return
|
|
65
|
-
typeof
|
|
65
|
+
const i = JSON.parse(a);
|
|
66
|
+
return E(i) ? i : [
|
|
67
|
+
typeof i == "string" ? { role: "text", content: i } : "text" in i ? { role: "text", content: i.text } : { role: "text", content: JSON.stringify(i) }
|
|
66
68
|
];
|
|
67
69
|
} catch {
|
|
68
|
-
return [{ role: "text", content:
|
|
70
|
+
return [{ role: "text", content: a }];
|
|
69
71
|
}
|
|
70
72
|
});
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
73
|
+
}
|
|
74
|
+
if (s === "application/zip") {
|
|
75
|
+
const c = await w.loadAsync(t);
|
|
76
|
+
let a = [];
|
|
77
|
+
const r = Object.keys(c.files);
|
|
78
|
+
for (let n = 0; n < r.length; n++) {
|
|
79
|
+
const i = r[n], p = c.file(i);
|
|
80
|
+
if (p) {
|
|
81
|
+
const l = await p.async("blob", (u) => {
|
|
82
|
+
if (o) {
|
|
83
|
+
const d = 0.1 + (u.percent / 100 * 0.9 / r.length + n / r.length * 0.9);
|
|
84
|
+
o(d);
|
|
85
|
+
}
|
|
86
|
+
}), f = await O(new File([l], i), e);
|
|
87
|
+
o && o(0.1 + (n + 1) / r.length * 0.9), a = a.concat(f);
|
|
78
88
|
}
|
|
79
89
|
}
|
|
80
|
-
return
|
|
90
|
+
return a;
|
|
81
91
|
}
|
|
82
|
-
if (
|
|
83
|
-
const
|
|
84
|
-
return new Promise((
|
|
85
|
-
|
|
92
|
+
if (s === "text/csv") {
|
|
93
|
+
const c = await t.text();
|
|
94
|
+
return o && o(0.1), new Promise((a, r) => {
|
|
95
|
+
x.parse(c, {
|
|
86
96
|
header: !1,
|
|
87
97
|
skipEmptyLines: !0,
|
|
88
98
|
delimiter: ",",
|
|
89
|
-
complete: (
|
|
90
|
-
if (
|
|
91
|
-
console.error(
|
|
99
|
+
complete: (n) => {
|
|
100
|
+
if (n.errors.length > 0)
|
|
101
|
+
console.error(n.errors), r(new Error("Error parsing file"));
|
|
92
102
|
else {
|
|
93
|
-
const
|
|
94
|
-
|
|
103
|
+
const i = h(n.data[0], e?.column || "text"), l = e?.hasHeader ?? z(n.data[0]) ? n.data.slice(1) : n.data;
|
|
104
|
+
a(l.map((f) => [{ role: "text", content: f[i] }]));
|
|
95
105
|
}
|
|
96
106
|
},
|
|
97
|
-
error: (
|
|
98
|
-
|
|
107
|
+
error: (n) => {
|
|
108
|
+
r(n);
|
|
99
109
|
}
|
|
100
110
|
});
|
|
101
111
|
});
|
|
102
|
-
} else if (
|
|
112
|
+
} else if (s === "text/plain")
|
|
103
113
|
return [[{ role: "text", content: await t.text() }]];
|
|
104
|
-
throw new Error(`Unsupported file type: ${
|
|
114
|
+
throw new Error(`Unsupported file type: ${s}`);
|
|
105
115
|
}
|
|
106
116
|
export {
|
|
107
|
-
|
|
117
|
+
O as default
|
|
108
118
|
};
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import { ag as h, aA as S, ab as N,
|
|
1
|
+
import { ag as h, aA as S, ab as N, d as v, aB as o, aC as p, aD as g, x as k, t as y } from "./index-CUXkjxiT.js";
|
|
2
2
|
import { s as R } from "./index-CieiGp4Y.js";
|
|
3
|
-
import { s as B } from "./stack-
|
|
4
|
-
import { t as $ } from "./tensor-
|
|
3
|
+
import { s as B } from "./stack-DudVrtmG.js";
|
|
4
|
+
import { t as $ } from "./tensor-BWFldCso.js";
|
|
5
5
|
function _(s, t) {
|
|
6
6
|
return b(s, t);
|
|
7
7
|
}
|
|
@@ -1,11 +1,11 @@
|
|
|
1
|
-
import { o as e,
|
|
2
|
-
function
|
|
1
|
+
import { o as e, q as a, x as i, E as m, A as p } from "./index-CUXkjxiT.js";
|
|
2
|
+
function x(t, n = 0) {
|
|
3
3
|
const s = a(t, "x", "expandDims", "string_or_numeric");
|
|
4
4
|
i(n <= s.rank, () => "Axis must be <= rank of the tensor");
|
|
5
5
|
const r = { input: s }, o = { dim: n };
|
|
6
6
|
return m.runKernel(p, r, o);
|
|
7
7
|
}
|
|
8
|
-
const
|
|
8
|
+
const u = /* @__PURE__ */ e({ expandDims_: x });
|
|
9
9
|
export {
|
|
10
|
-
|
|
10
|
+
u as e
|
|
11
11
|
};
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { o as g,
|
|
1
|
+
import { o as g, q as t, E as h, G as p } from "./index-CUXkjxiT.js";
|
|
2
2
|
function u(n, s, r = 0, e = 0) {
|
|
3
3
|
const o = t(n, "x", "gather"), a = t(s, "indices", "gather", "int32"), c = { x: o, indices: a }, i = { axis: r, batchDims: e };
|
|
4
4
|
return h.runKernel(p, c, i);
|