@genai-fi/nanogpt 0.10.2 → 0.11.0

package/dist/ops/attentionMask.js

@@ -1,4 +1,4 @@
-import { e as r } from "../index-D6Q1lPZO.js";
+import { e as r } from "../index-D0RBWjq8.js";
 import "./cpu/attentionMask.js";
 import "./webgl/attentionMask.js";
 import "./grads/attentionMask.js";

package/dist/ops/concat16.js

@@ -1,6 +1,6 @@
 import { isPackedTensor as o } from "../utilities/packed.js";
-import { e } from "../index-D6Q1lPZO.js";
-import { c } from "../concat-DmBLPVGC.js";
+import { e } from "../index-D0RBWjq8.js";
+import { c } from "../concat-C6X3AAlQ.js";
 function p(r, n) {
   return o(r[0]) ? e().runKernel("Concat16", r, { axis: n ?? -1 }) : c(r, n);
 }

package/dist/ops/cpu/adamAdjust.js

@@ -1,19 +1,18 @@
-import { l as t, n as r, m as k, o as z } from "../../index-D6Q1lPZO.js";
-import { r as A } from "../../tensor_util-DfwaWayG.js";
-function C(o) {
-  const { moments: n, value: i } = o.inputs, { beta1: l, beta2: m, epsilon: u, learningRate: d } = o.attrs, e = n.shape.length, c = new Array(e).fill(0), s = n.shape.slice();
+import { h, l as t, n as i, m as k, o as z } from "../../index-D0RBWjq8.js";
+function A(c) {
+  const { moments: n, value: r } = c.inputs, { beta1: l, beta2: m, epsilon: u, learningRate: d } = c.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
   s[e - 1] = 1;
-  const a = c.slice();
-  a[e - 1] = 1;
-  const p = s.slice(), b = n.slice(c, s).squeeze([e - 1]), M = n.slice(a, p).squeeze([e - 1]), f = t(b, l), g = t(M, m);
-  return r(
-    k(t(f, r(z(g), u ?? 1e-8)), -d),
-    i
+  const o = a.slice();
+  o[e - 1] = 1;
+  const b = s.slice(), p = n.slice(a, s).squeeze([e - 1]), M = n.slice(o, b).squeeze([e - 1]), g = t(p, l), f = t(M, m);
+  return i(
+    k(t(g, i(z(f), u ?? 1e-8)), -d),
+    r
   );
 }
-const h = {
+const C = {
   kernelName: "AdamAdjust",
   backendName: "cpu",
-  kernelFunc: C
+  kernelFunc: A
 };
-A(h);
+h(C);

package/dist/ops/cpu/adamMoments.js

@@ -1,12 +1,11 @@
-import "../../index-D6Q1lPZO.js";
-import { r as p } from "../../tensor_util-DfwaWayG.js";
-import { s as b } from "../../stack-ChnHwRpX.js";
+import { h as p } from "../../index-D0RBWjq8.js";
+import { s as b } from "../../stack-WJK22CFn.js";
 function f(t) {
-  const { moments: n, gradient: o } = t.inputs, { beta1: c, beta2: m } = t.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
+  const { moments: n, gradient: c } = t.inputs, { beta1: o, beta2: m } = t.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
   s[e - 1] = 1;
-  const r = a.slice();
-  r[e - 1] = 1;
-  const i = s.slice(), l = n.slice(a, s).squeeze([e - 1]), u = n.slice(r, i).squeeze([e - 1]), M = l.mul(c).add(o.mul(1 - c)), d = u.mul(m).add(o.square().mul(1 - m));
+  const i = a.slice();
+  i[e - 1] = 1;
+  const r = s.slice(), l = n.slice(a, s).squeeze([e - 1]), u = n.slice(i, r).squeeze([e - 1]), M = l.mul(o).add(c.mul(1 - o)), d = u.mul(m).add(c.square().mul(1 - m));
   return b([M, d], -1);
 }
 const g = {

package/dist/ops/cpu/appendCache.js

@@ -1,13 +1,12 @@
-import "../../index-D6Q1lPZO.js";
-import { r as d } from "../../tensor_util-DfwaWayG.js";
-import { c as h } from "../../concat-DmBLPVGC.js";
+import { h as d } from "../../index-D0RBWjq8.js";
+import { c as m } from "../../concat-C6X3AAlQ.js";
 function u(p) {
-  const { cache: n, item: s } = p.inputs, { maxSize: a, pastLen: c } = p.attrs, t = n.shape[0], o = n.shape[1], r = n.shape[3], e = s.shape[2];
-  if (c + e <= a) {
-    const m = n.slice([0, 0, 0, 0], [t, o, c, r]), f = n.slice([0, 0, c + e, 0], [t, o, a - c - e, r]), i = e < e ? s.slice([0, 0, 0, 0], [t, o, e, r]) : s, k = h([m, i, f], 2);
-    return m.dispose(), f.dispose(), i !== s && i.dispose(), k;
+  const { cache: n, item: s } = p.inputs, { maxSize: i, pastLen: c } = p.attrs, t = n.shape[0], o = n.shape[1], a = n.shape[3], e = s.shape[2];
+  if (c + e <= i) {
+    const f = n.slice([0, 0, 0, 0], [t, o, c, a]), h = n.slice([0, 0, c + e, 0], [t, o, i - c - e, a]), r = e < e ? s.slice([0, 0, 0, 0], [t, o, e, a]) : s, k = m([f, r, h], 2);
+    return f.dispose(), h.dispose(), r !== s && r.dispose(), k;
   }
-  const l = n.slice([0, 0, e, 0], [t, o, a - e, r]), C = h([l, s], 2);
+  const l = n.slice([0, 0, e, 0], [t, o, i - e, a]), C = m([l, s], 2);
   return l.dispose(), C;
 }
 const w = {

package/dist/ops/cpu/attentionMask.js

@@ -1,22 +1,22 @@
-import { i as d, b as u } from "../../index-D6Q1lPZO.js";
-import { r as o } from "../../tensor_util-DfwaWayG.js";
-import { l as N, w as b } from "../../ops-BFDtP6th.js";
-import { o as A } from "../../ones-jU9jlQvM.js";
-import { z as I } from "../../zeros-DBFVbpv5.js";
-import { m as g } from "../../mat_mul-C59XWcJd.js";
+import { h as o, i as d, b as u } from "../../index-D0RBWjq8.js";
+import { l as N } from "../../ops-FJapAPfm.js";
+import { o as b } from "../../ones-BX_wEgzB.js";
+import { z as A } from "../../zeros-DeiE2zTa.js";
+import { w as I } from "../../resize_nearest_neighbor-DB8k9KN_.js";
+import { m as g } from "../../mat_mul-DxpNTCRz.js";
 function a(n) {
-  const { q: s, k: e } = n.inputs, { divisor: r } = n.attrs, c = s.shape[2], t = e.shape[2], m = N.bandPart(A([t, t]), -1, 0).cast("bool"), i = I([t, t]), l = d([t, t], Number.NEGATIVE_INFINITY), f = b(m, i, l), k = g(s, e, !1, !0).mul(u(r)), p = f.slice([0, 0], [c, t]).expandDims(0).expandDims(0);
+  const { q: s, k: e } = n.inputs, { divisor: r } = n.attrs, c = s.shape[2], t = e.shape[2], m = N.bandPart(b([t, t]), -1, 0).cast("bool"), i = A([t, t]), l = d([t, t], Number.NEGATIVE_INFINITY), f = I(m, i, l), k = g(s, e, !1, !0).mul(u(r)), p = f.slice([0, 0], [c, t]).expandDims(0).expandDims(0);
   return k.add(p);
 }
-const w = {
+const h = {
   kernelName: "AttentionMask",
   backendName: "cpu",
   kernelFunc: a
 };
-o(w);
-const M = {
+o(h);
+const w = {
   kernelName: "AttentionMask",
   backendName: "tensorflow",
   kernelFunc: a
 };
-o(M);
+o(w);

package/dist/ops/cpu/fusedSoftmax.js

@@ -1,30 +1,29 @@
-import "../../index-D6Q1lPZO.js";
-import { r as e } from "../../tensor_util-DfwaWayG.js";
-import { s as m } from "../../softmax-C9JQEtnO.js";
-function o(t) {
-  const { inputs: s, attrs: a } = t, { logits: n } = s, { dim: i, dropoutRate: r } = a;
-  if (!n)
+import { h as e } from "../../index-D0RBWjq8.js";
+import { s as m } from "../../softmax-faLoUZVT.js";
+function n(t) {
+  const { inputs: s, attrs: a } = t, { logits: o } = s, { dim: i, dropoutRate: r } = a;
+  if (!o)
     throw new Error("Error in softmax: input logits is null");
-  return r !== void 0 && r > 0 && console.warn("Dropout in fusedSoftmax not implemented for CPU backend, skipping dropout."), m(n, i);
+  return r !== void 0 && r > 0 && console.warn("Dropout in fusedSoftmax not implemented for CPU backend, skipping dropout."), m(o, i);
 }
 const f = {
   kernelName: "FusedSoftmax",
   backendName: "cpu",
-  kernelFunc: o
+  kernelFunc: n
 };
 e(f);
 const u = {
   kernelName: "FusedSoftmax",
   backendName: "tensorflow",
-  kernelFunc: o
+  kernelFunc: n
 };
 e(u);
 const l = {
   kernelName: "FusedSoftmax",
   backendName: "webgpu",
-  kernelFunc: o
+  kernelFunc: n
 };
 e(l);
 export {
-  o as softmaxCPU
+  n as softmaxCPU
 };

package/dist/ops/cpu/gatherSub.js

@@ -1,19 +1,18 @@
-import { A as u, B as c, E as m, c as g } from "../../index-D6Q1lPZO.js";
-import { k as p, r as h } from "../../tensor_util-DfwaWayG.js";
-import { r as f } from "../../range-ZZZD60Fx.js";
-import { s as l } from "../../stack-ChnHwRpX.js";
-function N(e, t) {
-  const n = c(t, "indices", "gatherND", "int32"), r = { params: c(e, "x", "gatherND", "string_or_numeric"), indices: n };
-  return m.runKernel(p, r);
+import { q as u, u as c, E as g, aj as h, h as m, c as p } from "../../index-D0RBWjq8.js";
+import { r as l } from "../../range-CuGvVN2c.js";
+import { s as N } from "../../stack-WJK22CFn.js";
+function f(e, t) {
+  const n = c(t, "indices", "gatherND", "int32"), s = { params: c(e, "x", "gatherND", "string_or_numeric"), indices: n };
+  return g.runKernel(h, s);
 }
-const b = /* @__PURE__ */ u({ gatherND_: N });
+const b = /* @__PURE__ */ u({ gatherND_: f });
 function d(e) {
-  const { values: t, labels: n, logits: s } = e.inputs, r = n.shape[0], o = f(0, r, 1, "int32"), i = l([o, n], 1), a = b(s, i);
-  return g(t, a);
+  const { values: t, labels: n, logits: r } = e.inputs, s = n.shape[0], a = l(0, s, 1, "int32"), i = N([a, n], 1), o = b(r, i);
+  return p(t, o);
 }
 const k = {
   kernelName: "EfficientGatherSub",
   backendName: "cpu",
   kernelFunc: d
 };
-h(k);
+m(k);

package/dist/ops/cpu/gelu.js

@@ -1,41 +1,40 @@
-import { t as d } from "../../index-D6Q1lPZO.js";
-import { r } from "../../tensor_util-DfwaWayG.js";
+import { h as t, t as d } from "../../index-D0RBWjq8.js";
 const o = 0.7978845608028654, c = 0.044715;
-function m(t) {
-  const { inputs: u } = t, { x: n } = u, e = n;
+function m(r) {
+  const { inputs: u } = r, { x: n } = u, e = n;
   return d(() => {
     const l = e.pow(3), s = e.add(l.mul(c)).mul(o).tanh().add(1).mul(0.5);
     return e.mul(s);
   });
 }
-const N = {
+const p = {
   kernelName: "Gelu",
   backendName: "cpu",
   kernelFunc: m
 };
-r(N);
+t(p);
 const K = {
   kernelName: "Gelu",
   backendName: "tensorflow",
   kernelFunc: m
 };
-r(K);
-function i(t) {
-  const { dy: u, x: n } = t.inputs;
+t(K);
+function i(r) {
+  const { dy: u, x: n } = r.inputs;
   return d(() => {
-    const e = n.square(), l = e.mul(n), a = n.add(l.mul(c)).mul(o).tanh(), f = a.square().neg().add(1), k = e.mul(3 * c).add(1), g = a.add(1).mul(0.5), G = n.mul(f).mul(o).mul(k).mul(0.5), p = g.add(G);
-    return u.mul(p);
+    const e = n.square(), l = e.mul(n), a = n.add(l.mul(c)).mul(o).tanh(), f = a.square().neg().add(1), k = e.mul(3 * c).add(1), g = a.add(1).mul(0.5), G = n.mul(f).mul(o).mul(k).mul(0.5), N = g.add(G);
+    return u.mul(N);
   });
 }
-const x = {
+const h = {
   kernelName: "GeluGrad",
   backendName: "cpu",
   kernelFunc: i
 };
-r(x);
-const h = {
+t(h);
+const x = {
   kernelName: "GeluGrad",
   backendName: "tensorflow",
   kernelFunc: i
 };
-r(h);
+t(x);

package/dist/ops/cpu/matMul16.js

@@ -1,16 +1,15 @@
 import { isPackedTensor as t } from "../../utilities/packed.js";
-import "../../index-D6Q1lPZO.js";
-import { r as p } from "../../tensor_util-DfwaWayG.js";
-import { m } from "../../mat_mul-C59XWcJd.js";
-function l(r) {
-  const { A: e, B: n } = r.inputs, { transposeA: o, transposeB: s } = r.attrs, a = !t(e), c = !t(n);
+import { h as p } from "../../index-D0RBWjq8.js";
+import { m as l } from "../../mat_mul-DxpNTCRz.js";
+function m(e) {
+  const { A: n, B: r } = e.inputs, { transposeA: o, transposeB: s } = e.attrs, a = !t(n), c = !t(r);
   if (a && c)
-    return m(e, n, o, s);
+    return l(n, r, o, s);
   throw new Error("MatMul16 CPU kernel only supports packed tensors currently.");
 }
 const u = {
   kernelName: "MatMul16",
   backendName: "cpu",
-  kernelFunc: l
+  kernelFunc: m
 };
 p(u);

package/dist/ops/cpu/matMulGelu.js

@@ -1,12 +1,11 @@
-import { t as m } from "../../index-D6Q1lPZO.js";
-import { g as i, d as M } from "../../gelu-Bmhopi0J.js";
-import { r as e } from "../../tensor_util-DfwaWayG.js";
-import { m as k } from "../../mat_mul-C59XWcJd.js";
+import { h as e, t as m } from "../../index-D0RBWjq8.js";
+import { g as M, d as i } from "../../gelu-DqTbCx5x.js";
+import { m as k } from "../../mat_mul-DxpNTCRz.js";
 function c(t) {
   const { inputs: u } = t, { x: n, kernel: r } = u, a = n, l = r;
   return m(() => {
     const o = k(a, l);
-    return i(o);
+    return M(o);
   });
 }
 const G = {
@@ -30,7 +29,7 @@ e(p);
 function s(t) {
   const { dy: u, x: n, kernel: r } = t.inputs;
   return m(() => {
-    const a = k(n, r), l = M(u, a), o = l.matMul(r.transpose()), d = n.transpose().matMul(l);
+    const a = k(n, r), l = i(u, a), o = l.matMul(r.transpose()), d = n.transpose().matMul(l);
     return [o, d];
   });
 }

package/dist/ops/cpu/matMulMul.js

@@ -1,8 +1,7 @@
-import { t as M } from "../../index-D6Q1lPZO.js";
-import { r as e } from "../../tensor_util-DfwaWayG.js";
+import { h as e, t as i } from "../../index-D0RBWjq8.js";
 function n(t) {
-  const { inputs: r, attrs: l } = t, { transposeA: o, transposeB: s } = l, { x: u, kernel: a, y: c } = r, m = u, i = a, k = c;
-  return M(() => m.matMul(i, o, s).mul(k));
+  const { inputs: r, attrs: l } = t, { transposeA: o, transposeB: s } = l, { x: u, kernel: a, y: c } = r, m = u, k = a, M = c;
+  return i(() => m.matMul(k, o, s).mul(M));
 }
 const p = {
   kernelName: "MatMulMul",

package/dist/ops/cpu/mulDropout.js

@@ -1,8 +1,7 @@
-import { m as u } from "../../index-D6Q1lPZO.js";
-import { r as e } from "../../tensor_util-DfwaWayG.js";
+import { h as e, m as t } from "../../index-D0RBWjq8.js";
 function n(o) {
-  const { inputs: r } = o, { a: l, b: t } = r;
-  return console.warn("Using fallback mulDrop implementation without dropout."), u(l, t);
+  const { inputs: r } = o, { a: l, b: u } = r;
+  return console.warn("Using fallback mulDrop implementation without dropout."), t(l, u);
 }
 const a = {
   kernelName: "MulDropout",

package/dist/ops/cpu/normRMS.js

@@ -1,29 +1,28 @@
-import { t as d } from "../../index-D6Q1lPZO.js";
-import { r as a } from "../../tensor_util-DfwaWayG.js";
+import { h as o, t as d } from "../../index-D0RBWjq8.js";
 function i(t) {
-  const { inputs: e } = t, { x: n, gamma: s } = e, r = n, m = s;
+  const { inputs: e } = t, { x: n, gamma: s } = e, r = n, a = s;
   return d(() => {
     const u = r.square().mean(-1, !0).add(1e-8).rsqrt();
-    return r.mul(u).mul(m);
+    return r.mul(u).mul(a);
   });
 }
-const k = {
+const f = {
   kernelName: "RMSNorm",
   backendName: "cpu",
   kernelFunc: i
 };
-a(k);
+o(f);
 const g = {
   kernelName: "RMSNorm",
   backendName: "tensorflow",
   kernelFunc: i
 };
-a(g);
+o(g);
 function N(t) {
   const { dy: e, x: n, gamma: s } = t.inputs;
   return d(() => {
-    const r = n.shape[n.shape.length - 1], m = n.square().mean(-1, !0), o = m.add(1e-8).rsqrt(), u = n.mul(o), l = e.mul(u).sum([0, 1]), c = e.mul(s), f = c.mul(n).sum(-1, !0).div(r);
-    return [c.mul(o).sub(n.mul(f).mul(o).div(m.add(1e-8))), l];
+    const r = n.shape[n.shape.length - 1], a = n.square().mean(-1, !0), m = a.add(1e-8).rsqrt(), u = n.mul(m), l = e.mul(u).sum([0, 1]), c = e.mul(s), k = c.mul(n).sum(-1, !0).div(r);
+    return [c.mul(m).sub(n.mul(k).mul(m).div(a.add(1e-8))), l];
   });
 }
 const S = {
@@ -31,10 +30,10 @@ const S = {
   backendName: "cpu",
   kernelFunc: N
 };
-a(S);
-const p = {
+o(S);
+const R = {
   kernelName: "RMSNormGrad",
   backendName: "tensorflow",
   kernelFunc: N
 };
-a(p);
+o(R);

package/dist/ops/cpu/qkv.js

@@ -1,7 +1,6 @@
-import "../../index-D6Q1lPZO.js";
-import { r as q } from "../../tensor_util-DfwaWayG.js";
-import { r as o } from "../../reshape-CaPQzFvz.js";
-import { s as x } from "../../split-DBck65sX.js";
+import { h as q } from "../../index-D0RBWjq8.js";
+import { r as o } from "../../reshape-CkjKPPqB.js";
+import { s as x } from "../../split-BNz5jcGc.js";
 function v(p) {
   const { x: c, kernel: K } = p.inputs, { heads: n, packed: C } = p.attrs;
   if (C)
@@ -10,18 +9,18 @@ function v(p) {
   a.dispose();
   const d = o(i, [s, e, 3 * t]);
   i.dispose();
-  const [k, m, l] = x(d, 3, -1);
+  const [k, l, m] = x(d, 3, -1);
   d.dispose();
   const r = t / n, f = o(k, [s, e, n, r]);
   k.dispose();
   const w = f.transpose([0, 2, 1, 3]);
   f.dispose();
-  const h = o(m, [s, e, n, r]);
-  m.dispose();
+  const h = o(l, [s, e, n, r]);
+  l.dispose();
   const N = h.transpose([0, 2, 1, 3]);
   h.dispose();
-  const u = o(l, [s, e, n, r]);
-  l.dispose();
+  const u = o(m, [s, e, n, r]);
+  m.dispose();
   const T = u.transpose([0, 2, 1, 3]);
   return u.dispose(), [w, N, T];
 }

package/dist/ops/cpu/rope.js

@@ -1,21 +1,20 @@
-import "../../index-D6Q1lPZO.js";
-import { r as I } from "../../tensor_util-DfwaWayG.js";
-import { r as y } from "../../range-ZZZD60Fx.js";
-import { g as F } from "../../gather-CH9sdacz.js";
-import { s as E } from "../../stack-ChnHwRpX.js";
-import { c as T } from "../../concat-DmBLPVGC.js";
+import { h as I } from "../../index-D0RBWjq8.js";
+import { r as y } from "../../range-CuGvVN2c.js";
+import { g as F } from "../../gather-BcO5UQNJ.js";
+import { s as E } from "../../stack-WJK22CFn.js";
+import { c as T } from "../../concat-C6X3AAlQ.js";
 function U(c, r, p, e, n) {
   const t = e.shape[3], s = p;
   if (s > t) return e;
-  const o = e.shape[2], i = s / 2, a = r.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), d = c.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), l = e.shape[0], m = e.shape[1], h = y(0, s, 2, "int32"), g = y(1, s, 2, "int32"), D = ((k) => {
-    const C = k.slice([0, 0, 0, 0], [l, m, o, s]), R = s < t ? k.slice([0, 0, 0, s], [l, m, o, t - s]) : null, u = F(C, h, 3), f = F(C, g, 3), v = u.mul(a), N = f.mul(d), S = v.sub(N), P = f.mul(a), b = u.mul(d), x = P.add(b);
-    u.dispose(), f.dispose(), a.dispose(), d.dispose(), v.dispose(), N.dispose(), P.dispose(), b.dispose();
+  const o = e.shape[2], i = s / 2, a = r.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), d = c.slice([n, 0, 0], [o, i, 1]).reshape([1, 1, o, i]), l = e.shape[0], m = e.shape[1], f = y(0, s, 2, "int32"), g = y(1, s, 2, "int32"), D = ((k) => {
+    const C = k.slice([0, 0, 0, 0], [l, m, o, s]), R = s < t ? k.slice([0, 0, 0, s], [l, m, o, t - s]) : null, u = F(C, f, 3), h = F(C, g, 3), v = u.mul(a), N = h.mul(d), S = v.sub(N), P = h.mul(a), b = u.mul(d), x = P.add(b);
+    u.dispose(), h.dispose(), a.dispose(), d.dispose(), v.dispose(), N.dispose(), P.dispose(), b.dispose();
     const K = E([S, x], -1);
     S.dispose(), x.dispose();
     const w = K.reshape([l, m, o, s]);
     return K.dispose(), R ? T([w, R], 3) : w;
   })(e);
-  return h.dispose(), g.dispose(), D;
+  return f.dispose(), g.dispose(), D;
 }
 function B(c) {
   const { x: r } = c.inputs, { pastLen: p, negSin: e, ropeCache: n } = c.attrs, t = r.shape[3], s = e ? n.getNegSin() : n.getSin(), o = n.getCos();

package/dist/ops/cpu/scatterSub.js

@@ -1,25 +1,23 @@
-import { A as f, B as c, E as g, c as l, m as N } from "../../index-D6Q1lPZO.js";
-import { j as b, r as S } from "../../tensor_util-DfwaWayG.js";
-import { d as h } from "../../tensor-CzmOBsdf.js";
-import { v as D } from "../../scatter_nd_util-C7zXRT_h.js";
-import { r as k } from "../../range-ZZZD60Fx.js";
-import { s as v } from "../../stack-ChnHwRpX.js";
-import { o as E } from "../../ones-jU9jlQvM.js";
-function I(r, e, s) {
-  h(s);
-  const n = c(r, "indices", "scatterND", "int32"), t = c(e, "updates", "scatterND");
+import { q as f, w as g, u as r, E as l, ai as N, h as b, c as S, m as h } from "../../index-D0RBWjq8.js";
+import { v as D } from "../../scatter_nd_util-BY22Cc-C.js";
+import { r as k } from "../../range-CuGvVN2c.js";
+import { s as v } from "../../stack-WJK22CFn.js";
+import { o as E } from "../../ones-BX_wEgzB.js";
+function I(a, e, s) {
+  g(s);
+  const n = r(a, "indices", "scatterND", "int32"), t = r(e, "updates", "scatterND");
   D(t, n, s);
-  const o = { indices: n, updates: t }, a = { shape: s };
-  return g.runKernel(b, o, a);
+  const c = { indices: n, updates: t }, o = { shape: s };
+  return l.runKernel(N, c, o);
 }
 const K = /* @__PURE__ */ f({ scatterND_: I });
-function L(r) {
-  const { logits: e, labels: s, dy: n } = r.inputs, t = s.shape[0], o = e.shape[1], a = k(0, t, 1, "int32"), i = v([a, s], 1), d = E([t]), p = K(i, d, [t, o]), u = l(e, p), m = n.reshape([t, 1]);
-  return N(u, m);
+function L(a) {
+  const { logits: e, labels: s, dy: n } = a.inputs, t = s.shape[0], c = e.shape[1], o = k(0, t, 1, "int32"), i = v([o, s], 1), d = E([t]), u = K(i, d, [t, c]), p = S(e, u), m = n.reshape([t, 1]);
+  return h(p, m);
 }
 const T = {
   kernelName: "EfficientScatterSub",
   backendName: "cpu",
   kernelFunc: L
 };
-S(T);
+b(T);

package/dist/ops/dot16.js

@@ -1,8 +1,8 @@
-import { b as d } from "../matMul16-fEAJ4smh.js";
+import { b as d } from "../matMul16-cDxwemKj.js";
 import { transpose16 as w } from "./transpose16.js";
 import { reshape16 as n } from "./reshape16.js";
 import { isPackedTensor as p } from "../utilities/packed.js";
-import { d as x } from "../tfjs_backend-CNkSTL0c.js";
+import { d as x } from "../tfjs_backend-Vi4JfLzT.js";
 function E(e, s, h = !1, c = !1) {
   if (!p(e) && !p(s))
     return x(e, s);

package/dist/ops/gatherSub.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-D6Q1lPZO.js";
+import { e as n } from "../index-D0RBWjq8.js";
 import "./cpu/gatherSub.js";
 import "./webgl/gatherSub.js";
 function f(r, e, t) {

package/dist/ops/gelu.js

@@ -1,7 +1,7 @@
-import "../index-D6Q1lPZO.js";
+import "../index-D0RBWjq8.js";
 import "./cpu/gelu.js";
 import "./webgl/gelu.js";
-import { d as e, g as i } from "../gelu-Bmhopi0J.js";
+import { d as e, g as i } from "../gelu-DqTbCx5x.js";
 export {
   e as dGelu,
   i as gelu

package/dist/ops/grads/add16.js

@@ -1,27 +1,26 @@
-import { j as u, q as d } from "../../index-D6Q1lPZO.js";
-import { sum16 as p } from "../sum16.js";
-import { reshape16 as c } from "../reshape16.js";
-import { a as h } from "../../tensor_util-DfwaWayG.js";
-const m = {
+import { j as u, $ as h, a0 as d } from "../../index-D0RBWjq8.js";
+import { sum16 as c } from "../sum16.js";
+import { reshape16 as p } from "../reshape16.js";
+const A = {
   kernelName: "Add16",
   inputsToSave: ["a", "b"],
   gradFunc: (s, i) => {
-    const [t, a] = i, n = u(t.shape, a.shape);
+    const [t, a] = i, n = h(t.shape, a.shape);
     if (Array.isArray(s))
       throw new Error("Add16 gradFunc expected dy to be a Tensor but got an array");
     return { a: () => {
       let e = s;
       const r = d(t.shape, n);
-      r.length > 0 && (e = p(e, r));
-      const o = c(e, t.shape);
+      r.length > 0 && (e = c(e, r));
+      const o = p(e, t.shape);
       return e.dispose(), o;
     }, b: () => {
       let e = s;
       const r = d(a.shape, n);
-      r.length > 0 && (e = p(e, r));
-      const o = c(e, a.shape);
+      r.length > 0 && (e = c(e, r));
+      const o = p(e, a.shape);
       return e.dispose(), o;
     } };
   }
 };
-h(m);
+u(A);

package/dist/ops/grads/attentionMask.js

@@ -1,7 +1,6 @@
-import "../../index-D6Q1lPZO.js";
-import { m as o } from "../../matMul16-fEAJ4smh.js";
-import { transpose16 as m } from "../transpose16.js";
-import { a as c } from "../../tensor_util-DfwaWayG.js";
+import { j as m } from "../../index-D0RBWjq8.js";
+import { m as o } from "../../matMul16-cDxwemKj.js";
+import { transpose16 as c } from "../transpose16.js";
 const l = {
   kernelName: "AttentionMask",
   inputsToSave: ["q", "k"],
@@ -13,10 +12,10 @@ const l = {
     return {
       q: () => o(r, i, e),
       k: () => {
-        const t = o(a, r, e, !0, !1), u = m(t, [0, 1, 3, 2]);
+        const t = o(a, r, e, !0, !1), u = c(t, [0, 1, 3, 2]);
         return t.dispose(), u;
       }
     };
   }
 };
-c(l);
+m(l);

package/dist/ops/grads/gelu.js

@@ -1,6 +1,5 @@
-import "../../index-D6Q1lPZO.js";
-import { a as m } from "../../gelu-Bmhopi0J.js";
-import "../../tensor_util-DfwaWayG.js";
+import "../../index-D0RBWjq8.js";
+import { a as e } from "../../gelu-DqTbCx5x.js";
 export {
-  m as geluGradConfig
+  e as geluGradConfig
 };

package/dist/ops/grads/matMul16.js

@@ -1,9 +1,8 @@
-import "../../index-D6Q1lPZO.js";
-import { a as f } from "../../matMul16-fEAJ4smh.js";
-import "../../gelu-Bmhopi0J.js";
+import "../../index-D0RBWjq8.js";
+import { a } from "../../matMul16-cDxwemKj.js";
+import "../../gelu-DqTbCx5x.js";
 import "../transpose16.js";
 import "../reshape16.js";
-import "../../tensor_util-DfwaWayG.js";
 export {
-  f as matMul16GradConfig
+  a as matMul16GradConfig
 };