@genai-fi/nanogpt 0.10.3 → 0.11.0

package/dist/ops/webgpu/sum16.js

@@ -1,10 +1,10 @@
-import { createReduceInfo as g, reduce as h, ReduceProgram as x } from "./utils/reductions.js";
-import { p as k, af as A } from "../../index-DOvlwCh-.js";
+import { createReduceInfo as h, reduce as g, ReduceProgram as x } from "./utils/reductions.js";
+import { h as k, af as A } from "../../index-D0RBWjq8.js";
 import { isPackedTensor as P } from "../../utilities/packed.js";
 import { transpose16 as b } from "../transpose16.js";
 import I from "./utils/deviceInfo.js";
-import { s as w } from "../../sum-DWAtNGez.js";
-import { a as D, b as K } from "../../axis_util-BaG7mf5A.js";
+import { s as w } from "../../sum-BdplSvq_.js";
+import { a as D, b as K } from "../../axis_util-DofAuy0p.js";
 class v extends x {
   shaderKey = "sum16";
   constructor(e, t, o) {
@@ -20,14 +20,14 @@ class v extends x {
   }
 }
 function y(r) {
-  const { x: e } = r.inputs, { axis: t, keepDims: o } = r.attrs, a = r.backend, c = [], p = I(a), m = P(e);
+  const { x: e } = r.inputs, { axis: t, keepDims: o } = r.attrs, a = r.backend, c = [], u = I(a), m = P(e);
   if (!m)
     return w(e, t, o);
   let n = A(t ?? -1, e.shape);
   const i = D(n, e.shape.length);
   let s = e;
   i != null && (s = b(e, i), n = K(n.length, s.shape.length), c.push(s));
-  const u = g([s], -1), f = new v(p, u, m), d = h(f, [s], a);
+  const p = h([s], -1), f = new v(u, p, m), d = g(f, [s], a);
   return c.forEach((l) => l.dispose()), d;
 }
 const N = {

package/dist/ops/webgpu/transpose16.js

@@ -1,9 +1,9 @@
 import { isPackedTensor as d } from "../../utilities/packed.js";
-import { p as k, e as l } from "../../index-DOvlwCh-.js";
+import { h as k, e as l } from "../../index-D0RBWjq8.js";
 import { reshape16 as u } from "../reshape16.js";
 import P from "./transpose16_shared_program.js";
 import b from "./transpose16_program.js";
-import { t as T } from "../../transpose-ClWiBS_b.js";
+import { t as T } from "../../transpose-JawVKyZy.js";
 function w(p) {
   const { inputs: h, attrs: i } = p, { x: e } = h, { perm: r } = i, a = p.backend, m = d(e);
   if (m && r[r.length - 1] !== e.shape.length - 1) {

package/dist/ops/webgpu/transpose16_program.js

@@ -1,5 +1,5 @@
-import { f as a, c as i } from "../../webgpu_util-RxEF33Rj.js";
-import { b as h, e as d, c as n } from "../../webgpu_program-DuOXPQol.js";
+import { f as a, c as i } from "../../webgpu_util-BdovYhXr.js";
+import { b as h, e as d, c as n } from "../../webgpu_program-CAE4RICo.js";
 function p(r) {
   const e = r.length;
   if (e > 6)

package/dist/ops/webgpu/transpose16_shared_program.js

@@ -1,6 +1,6 @@
-import { c as a } from "../../webgpu_util-RxEF33Rj.js";
-import { e as l } from "../../webgpu_program-DuOXPQol.js";
-import { l as p } from "../../index-DOvlwCh-.js";
+import { c as a } from "../../webgpu_util-BdovYhXr.js";
+import { e as l } from "../../webgpu_program-CAE4RICo.js";
+import { y as p } from "../../index-D0RBWjq8.js";
 class n {
   variableNames = ["A"];
   outputShape;

package/dist/ops/webgpu/unpack16.js

@@ -1,6 +1,6 @@
-import { f as c, c as r } from "../../webgpu_util-RxEF33Rj.js";
-import { e as u } from "../../webgpu_program-DuOXPQol.js";
-import { p } from "../../index-DOvlwCh-.js";
+import { f as c, c as r } from "../../webgpu_util-BdovYhXr.js";
+import { e as u } from "../../webgpu_program-CAE4RICo.js";
+import { h as p } from "../../index-D0RBWjq8.js";
 class l {
   outputShape;
   shaderKey = "Unpack16";

package/dist/ops/webgpu/utils/binary_op.js

@@ -1,8 +1,8 @@
-import { f as s, c as n } from "../../../webgpu_util-RxEF33Rj.js";
+import { f as s, c as n } from "../../../webgpu_util-BdovYhXr.js";
 import { g as p } from "../../../binary_op_util-pKXltfxI.js";
 import { B as b } from "../../../binary_op_util-pKXltfxI.js";
-import { a3 as u, ag as r } from "../../../index-DOvlwCh-.js";
-import { e as l } from "../../../webgpu_program-DuOXPQol.js";
+import { $ as u, ag as r } from "../../../index-D0RBWjq8.js";
+import { e as l } from "../../../webgpu_program-CAE4RICo.js";
 class x {
   dispatch;
   dispatchLayout;

package/dist/ops/webgpu/utils/reductions.js

@@ -1,8 +1,8 @@
-import { af as l, h as c, e as S } from "../../../index-DOvlwCh-.js";
-import { e as a } from "../../../webgpu_program-DuOXPQol.js";
+import { af as l, V as c, e as S } from "../../../index-D0RBWjq8.js";
+import { e as a } from "../../../webgpu_program-CAE4RICo.js";
 import { reshape16 as b } from "../../reshape16.js";
-import { f } from "../../../webgpu_util-RxEF33Rj.js";
-import { c as h } from "../../../axis_util-BaG7mf5A.js";
+import { f } from "../../../webgpu_util-BdovYhXr.js";
+import { c as h } from "../../../axis_util-DofAuy0p.js";
 function d(e, u, t, i) {
   return e && !u ? `
             bestValue = subgroupAdd(bestValue);

package/dist/{ops-B5yanEdW.js → ops-FJapAPfm.js}

@@ -1,26 +1,26 @@
-import { A as b, B as u, l as p, E as k, bf as hs, bg as bs, at as ds, bh as gs, bi as Es, X as W, bj as $s, bk as ys, bl as _s, bm as Ns, b as $, m as h, x as G, w as S, q as Ss, c as d, bn as ks, D as K, d as Is, ab as B, aZ as ns, b0 as Ts, a6 as Ms } from "./index-DOvlwCh-.js";
-import { t as j } from "./tile-mbfagpsB.js";
-import { e as R } from "./expand_dims-DT4tEPwA.js";
-import { t as x } from "./tensor1d-D11P_7Dp.js";
+import { q as b, u, y as p, E as k, bf as hs, bg as bs, at as ds, bh as gs, bi as Es, N as W, bj as $s, bk as ys, bl as _s, bm as Ns, b as $, m as h, n as G, l as S, i as Ss, c as d, bn as ks, x as K, d as Is, a7 as L, aZ as ns, b0 as Ts, a2 as Ms } from "./index-D0RBWjq8.js";
+import { t as j } from "./tile-CvN_LyVr.js";
+import { e as R } from "./expand_dims-BzfJK2uc.js";
+import { t as x } from "./tensor1d-LxP9asMm.js";
 import { n as xs, a as As, b as ws } from "./non_max_suppression_impl-B2W7YjZB.js";
-import { r as Os, l as ls, g as ts, w as q, a as rs, m as Q, b as Gs, c as Bs, u as us, n as ps, d as Ds, e as Ls, s as Ws, f as Rs, h as qs, i as zs } from "./resize_nearest_neighbor-B19mCEg2.js";
-import { s as ms } from "./split-CVLc0w--.js";
-import { s as _ } from "./sum-DWAtNGez.js";
-import { s as T } from "./slice-BfEGSH82.js";
-import { r as F } from "./range-BklejeeW.js";
-import { t as Vs } from "./tensor-DJoc7gJU.js";
-import { n as C, t as as } from "./transpose-ClWiBS_b.js";
-import { r as O } from "./reshape-ByE68wS9.js";
-import { s as Z } from "./stack-Cf4n9h0N.js";
-import { z as Cs } from "./zeros-KnWaWf-X.js";
-import { s as Ps } from "./squeeze-C7Z2srUo.js";
-import { c as X } from "./concat-BV8bt5H-.js";
-import { m as w } from "./mat_mul-SjpJRLyL.js";
-import { t as J } from "./tensor2d-Bs9wZRc7.js";
-import { o as Ys } from "./ones-D2rT0xk2.js";
-import { r as fs } from "./relu-CP0ZcxWO.js";
-import { a as is, e as ss, l as vs } from "./log_sum_exp-ngO0-4pK.js";
-import { e as js } from "./axis_util-BaG7mf5A.js";
+import { r as Os, l as ls, g as ts, w as q, a as rs, m as Q, b as Gs, c as Ls, u as us, n as ps, d as Bs, e as Ds, s as Ws, f as Rs, h as qs, i as zs } from "./resize_nearest_neighbor-DB8k9KN_.js";
+import { s as ms } from "./split-BNz5jcGc.js";
+import { s as _ } from "./sum-BdplSvq_.js";
+import { s as T } from "./slice-Aqy7KbJh.js";
+import { r as F } from "./range-CuGvVN2c.js";
+import { t as Vs } from "./tensor-BQqrDvpx.js";
+import { n as C, t as as } from "./transpose-JawVKyZy.js";
+import { r as O } from "./reshape-CkjKPPqB.js";
+import { s as Z } from "./stack-WJK22CFn.js";
+import { z as Cs } from "./zeros-DeiE2zTa.js";
+import { s as Ps } from "./squeeze--YMgaAAf.js";
+import { c as X } from "./concat-C6X3AAlQ.js";
+import { m as w } from "./mat_mul-DxpNTCRz.js";
+import { t as J } from "./tensor2d-BN1sSfQO.js";
+import { o as Ys } from "./ones-BX_wEgzB.js";
+import { r as fs } from "./relu-Cf80uA2p.js";
+import { a as is, e as ss, l as vs } from "./log_sum_exp-VLZgbFAH.js";
+import { e as js } from "./axis_util-DofAuy0p.js";
 function Fs(e, a, n) {
   const r = u(e, "x", "bincount"), t = u(a, "weights", "bincount");
   p(r.dtype === "int32", () => `Error in bincount: input dtype must be int32, but got ${r.dtype}`), p(n >= 0, () => `size must be non-negative, but got ${n}.`), p(t.size === r.size || t.size === 0, () => `Error in bincount: weights must have the same size as input or0-length, but got input shape: ${r.shape}, weights shape: ${t.shape}.`);
@@ -186,8 +186,8 @@ function yn(e, a = "binary", n = !1, r = 0.5) {
   let c = h(x([r]), 255), f, m, E, g;
   if (p(t.rank === 3, () => `Error in threshold: image must be rank 3,but got rank ${t.rank}.`), p(t.shape[2] === 3 || t.shape[2] === 1, () => `Error in threshold: image color channel must be equal to 3 or 1but got ${t.shape[2]}.`), p(t.dtype === "int32" || t.dtype === "float32", () => `Error in dtype: image dtype must be int32 or float32,but got dtype ${t.dtype}.`), p(a === "otsu" || a === "binary", () => `Method must be binary or otsu, but was ${a}`), t.shape[2] === 3) {
     [f, m, E] = ms(t, [1, 1, 1], -1);
-    const I = h(f, s), A = h(m, o), L = h(E, i);
-    g = G(G(I, A), L);
+    const I = h(f, s), A = h(m, o), D = h(E, i);
+    g = G(G(I, A), D);
   } else
     g = e;
   if (a === "otsu") {
@@ -205,8 +205,8 @@ function _n(e, a) {
     i = S(E, _(s));
     const g = Ss(o.shape, s.size), y = G(F(0, o.size), g), M = h(o, y);
     l = S(_(M), _(o));
-    const I = d(i, l), A = d(i, l), L = h(c, f);
-    t = h(h(L, I), A);
+    const I = d(i, l), A = d(i, l), D = h(c, f);
+    t = h(h(D, I), A);
     const P = ts(t, r);
     r = q(P, t, r), n = q(P, x([m]), n);
   }
@@ -226,7 +226,7 @@ function In(e, a, n) {
   const t = r.shape, [s, o] = r.shape.slice(-2);
   let i, l;
   typeof a == "number" ? (p(a % 1 === 0, () => `bandPart(): numLower must be an integer, got ${a}.`), p(a <= s, () => `bandPart(): numLower (${a}) must not be greater than the number of rows (${s}).`), i = u(a < 0 ? s : a, "numLower", "bandPart")) : (p(a.dtype === "int32", () => "bandPart(): numLower's dtype must be an int32."), i = q(rs(a, 0), s, Q(a, s))), typeof n == "number" ? (p(n % 1 === 0, () => `bandPart(): numUpper must be an integer, got ${n}.`), p(n <= o, () => `bandPart(): numUpper (${n}) must not be greater than the number of columns (${o}).`), l = u(n < 0 ? o : n, "numUpper", "bandPart")) : (p(n.dtype === "int32", () => "bandPart(): numUpper's dtype must be an int32."), l = q(rs(n, 0), o, Q(n, o)));
-  const c = O(F(0, s, 1, "int32"), [-1, 1]), f = F(0, o, 1, "int32"), m = d(c, f), E = Gs(ls(m, i), Bs(m, C(l))), g = Cs([s, o], r.dtype);
+  const c = O(F(0, s, 1, "int32"), [-1, 1]), f = F(0, o, 1, "int32"), m = d(c, f), E = Gs(ls(m, i), Ls(m, C(l))), g = Cs([s, o], r.dtype);
   return O(Z(us(O(r, [-1, s, o])).map((y) => q(E, y, g))), t);
 }
 const Tn = /* @__PURE__ */ b({ bandPart_: In });
@@ -282,10 +282,10 @@ function cs(e, a = !1) {
     for (let c = 0; c < l; ++c) {
       const f = s, m = i, E = t;
       [i, s, t] = k.tidy(() => {
-        const g = T(s, [c, c], [n - c, 1]), y = ps(g), M = T(s, [c, c], [1, 1]), I = q(ts(M, 0), J([[-1]]), J([[1]])), A = d(M, h(I, y)), L = S(g, A);
-        L.shape[0] === 1 ? i = K(o) : i = X([
+        const g = T(s, [c, c], [n - c, 1]), y = ps(g), M = T(s, [c, c], [1, 1]), I = q(ts(M, 0), J([[-1]]), J([[1]])), A = d(M, h(I, y)), D = S(g, A);
+        D.shape[0] === 1 ? i = K(o) : i = X([
           o,
-          T(L, [1, 0], [L.shape[0] - 1, L.shape[1]])
+          T(D, [1, 0], [D.shape[0] - 1, D.shape[1]])
         ], 0);
         const P = C(S(w(I, A), y)), Y = T(s, [c, 0], [n - c, r]), U = h(P, i), es = as(i);
         if (c === 0)
@@ -323,7 +323,7 @@ function On(e, a, n = N.SUM_BY_NONZERO_WEIGHTS) {
     return _(s);
   if (n === N.MEAN) {
     if (t == null)
-      return Ds(s);
+      return Bs(s);
     {
       const o = r.size / t.size, i = S(_(s), _(t));
       return o > 1 ? S(i, $(o)) : i;
@@ -333,67 +333,67 @@ function On(e, a, n = N.SUM_BY_NONZERO_WEIGHTS) {
     if (t == null)
       return S(_(s), $(r.size));
     {
-      const o = h(t, Ys(r.shape)), i = W(_(Ls(o, $(0))), "float32");
+      const o = h(t, Ys(r.shape)), i = W(_(Ds(o, $(0))), "float32");
       return S(_(s), i);
     }
   }
   throw Error(`Unknown reduction: ${n}`);
 }
-const D = /* @__PURE__ */ b({ computeWeightedLoss_: On });
+const B = /* @__PURE__ */ b({ computeWeightedLoss_: On });
 function Gn(e, a, n, r = N.SUM_BY_NONZERO_WEIGHTS) {
   const t = u(e, "labels", "absoluteDifference"), s = u(a, "predictions", "absoluteDifference");
   let o = null;
-  n != null && (o = u(n, "weights", "absoluteDifference")), B(t.shape, s.shape, "Error in absoluteDifference: ");
+  n != null && (o = u(n, "weights", "absoluteDifference")), L(t.shape, s.shape, "Error in absoluteDifference: ");
   const i = ns(d(t, s));
-  return D(i, o, r);
+  return B(i, o, r);
 }
-const Bn = /* @__PURE__ */ b({ absoluteDifference_: Gn });
-function Dn(e, a, n, r, t = N.SUM_BY_NONZERO_WEIGHTS) {
+const Ln = /* @__PURE__ */ b({ absoluteDifference_: Gn });
+function Bn(e, a, n, r, t = N.SUM_BY_NONZERO_WEIGHTS) {
   const s = u(e, "labels", "cosineDistance"), o = u(a, "predictions", "cosineDistance");
   let i = null;
-  r != null && (i = u(r, "weights", "cosineDistance")), B(s.shape, o.shape, "Error in cosineDistance: ");
+  r != null && (i = u(r, "weights", "cosineDistance")), L(s.shape, o.shape, "Error in cosineDistance: ");
   const l = $(1), c = d(l, _(h(s, o), n, !0));
-  return D(c, i, t);
+  return B(c, i, t);
 }
-const Ln = /* @__PURE__ */ b({ cosineDistance_: Dn });
+const Dn = /* @__PURE__ */ b({ cosineDistance_: Bn });
 function Wn(e, a, n, r = N.SUM_BY_NONZERO_WEIGHTS) {
   let t = u(e, "labels", "hingeLoss");
   const s = u(a, "predictions", "hingeLoss");
   let o = null;
-  n != null && (o = u(n, "weights", "hingeLoss")), B(t.shape, s.shape, "Error in hingeLoss: ");
+  n != null && (o = u(n, "weights", "hingeLoss")), L(t.shape, s.shape, "Error in hingeLoss: ");
   const i = $(1);
   t = d(h($(2), t), i);
   const l = fs(d(i, h(t, s)));
-  return D(l, o, r);
+  return B(l, o, r);
 }
 const Rn = /* @__PURE__ */ b({ hingeLoss_: Wn });
 function qn(e, a, n, r = 1, t = N.SUM_BY_NONZERO_WEIGHTS) {
   const s = u(e, "labels", "huberLoss"), o = u(a, "predictions", "huberLoss");
   let i = null;
-  n != null && (i = u(n, "weights", "huberLoss")), B(s.shape, o.shape, "Error in huberLoss: ");
+  n != null && (i = u(n, "weights", "huberLoss")), L(s.shape, o.shape, "Error in huberLoss: ");
   const l = $(r), c = ns(d(o, s)), f = Q(c, l), m = d(c, f), E = G(h($(0.5), Ts(f)), h(l, m));
-  return D(E, i, t);
+  return B(E, i, t);
 }
 const zn = /* @__PURE__ */ b({ huberLoss_: qn });
 function Vn(e, a, n, r = 1e-7, t = N.SUM_BY_NONZERO_WEIGHTS) {
   const s = u(e, "labels", "logLoss"), o = u(a, "predictions", "logLoss");
   let i = null;
-  n != null && (i = u(n, "weights", "logLoss")), B(s.shape, o.shape, "Error in logLoss: ");
+  n != null && (i = u(n, "weights", "logLoss")), L(s.shape, o.shape, "Error in logLoss: ");
   const l = $(1), c = $(r), f = C(h(s, is(G(o, c)))), m = h(d(l, s), is(G(d(l, o), c))), E = d(f, m);
-  return D(E, i, t);
+  return B(E, i, t);
 }
 const Cn = /* @__PURE__ */ b({ logLoss_: Vn });
 function Pn(e, a, n, r = N.SUM_BY_NONZERO_WEIGHTS) {
   const t = u(e, "labels", "meanSquaredError"), s = u(a, "predictions", "meanSquaredError");
   let o = null;
-  n != null && (o = u(n, "weights", "meanSquaredError")), B(t.shape, s.shape, "Error in meanSquaredError: ");
+  n != null && (o = u(n, "weights", "meanSquaredError")), L(t.shape, s.shape, "Error in meanSquaredError: ");
   const i = Ws(t, s);
-  return D(i, o, r);
+  return B(i, o, r);
 }
 const Yn = /* @__PURE__ */ b({ meanSquaredError_: Pn });
 function vn(e, a) {
   const n = u(e, "labels", "sigmoidCrossEntropyWithLogits"), r = u(a, "logits", "sigmoidCrossEntropyWithLogits");
-  B(n.shape, r.shape, "Error in sigmoidCrossEntropyWithLogits: ");
+  L(n.shape, r.shape, "Error in sigmoidCrossEntropyWithLogits: ");
   const t = fs(r), s = h(r, n), o = Rs(ss(C(ns(r))));
   return G(d(t, s), o);
 }
@@ -401,12 +401,12 @@ function jn(e, a, n, r = 0, t = N.SUM_BY_NONZERO_WEIGHTS) {
   let s = u(e, "multiClassLabels", "sigmoidCrossEntropy");
   const o = u(a, "logits", "sigmoidCrossEntropy");
   let i = null;
-  if (n != null && (i = u(n, "weights", "sigmoidCrossEntropy")), B(s.shape, o.shape, "Error in sigmoidCrossEntropy: "), r > 0) {
+  if (n != null && (i = u(n, "weights", "sigmoidCrossEntropy")), L(s.shape, o.shape, "Error in sigmoidCrossEntropy: "), r > 0) {
     const c = $(r), f = $(1), m = $(0.5);
     s = G(h(s, d(f, c)), h(m, c));
   }
   const l = vn(s, o);
-  return D(l, i, t);
+  return B(l, i, t);
 }
 const Fn = /* @__PURE__ */ b({ sigmoidCrossEntropy_: jn });
 function Zn(e, a, n = -1) {
@@ -429,12 +429,12 @@ function Un(e, a, n, r = 0, t = N.SUM_BY_NONZERO_WEIGHTS) {
   let s = u(e, "onehotLabels", "softmaxCrossEntropy");
   const o = u(a, "logits", "softmaxCrossEntropy");
   let i = null;
-  if (n != null && (i = u(n, "weights", "softmaxCrossEntropy")), B(s.shape, o.shape, "Error in softmaxCrossEntropy: "), r > 0) {
+  if (n != null && (i = u(n, "weights", "softmaxCrossEntropy")), L(s.shape, o.shape, "Error in softmaxCrossEntropy: "), r > 0) {
     const c = $(r), f = $(1), m = $(s.shape[1]);
     s = G(h(s, d(f, c)), S(c, m));
   }
   const l = Zn(s, o);
-  return D(l, i, t);
+  return B(l, i, t);
 }
 const Hn = /* @__PURE__ */ b({ softmaxCrossEntropy_: Un });
 const $t = {
@@ -458,9 +458,9 @@ const $t = {
   gramSchmidt: xn,
   qr: wn
 }, _t = {
-  absoluteDifference: Bn,
-  computeWeightedLoss: D,
-  cosineDistance: Ln,
+  absoluteDifference: Ln,
+  computeWeightedLoss: B,
+  cosineDistance: Dn,
   hingeLoss: Rn,
   huberLoss: zn,
   logLoss: Cn,

package/dist/{pack16-nQ6JaLo-.js → pack16-k4jq6aMX.js}

@@ -1,6 +1,6 @@
-import { u as o, e as u } from "./index-DOvlwCh-.js";
-import { isPackedTensor as i } from "./utilities/packed.js";
-import { s as t } from "./slice-BfEGSH82.js";
+import { j as o, e as i } from "./index-D0RBWjq8.js";
+import { isPackedTensor as t } from "./utilities/packed.js";
+import { s as u } from "./slice-Aqy7KbJh.js";
 const s = {
   kernelName: "Unpack16",
   inputsToSave: [],
@@ -11,9 +11,9 @@ const s = {
 };
 o(s);
 function p(n, a = 1, e = !1) {
-  if (!i(n))
+  if (!t(n))
     return n;
-  const r = u().runKernel("Unpack16", { x: n }, { scaling: a });
+  const r = i().runKernel("Unpack16", { x: n }, { scaling: a });
   return e && n.dispose(), r;
 }
 const c = {
@@ -23,13 +23,13 @@ const c = {
   gradFunc: (n, a, e) => ({
     x: () => {
       const r = p(n);
-      return e.originalShape && e.padding && e.padding > 0 ? t(r, new Array(r.shape.length).fill(0), e.originalShape) : r;
+      return e.originalShape && e.padding && e.padding > 0 ? u(r, new Array(r.shape.length).fill(0), e.originalShape) : r;
     }
   })
 };
 o(c);
 function d(n, a = 1, e = 0) {
-  return u().runKernel("Pack16", { x: n }, { scaling: a, padding: e });
+  return i().runKernel("Pack16", { x: n }, { scaling: a, padding: e });
 }
 export {
   s as a,

package/dist/patches/webgpu_backend.js

@@ -1,8 +1,8 @@
-import { h as c, i as l, j as m, l as n } from "../index-DOvlwCh-.js";
-import "../webgpu_util-RxEF33Rj.js";
-import { W as P } from "../backend_webgpu-DE3ACOLx.js";
-import { compileProgram as y } from "./webgpu_program.js";
-import { m as M } from "../webgpu_program-DuOXPQol.js";
+import { V as c, ae as l, ab as m, y as n } from "../index-D0RBWjq8.js";
+import "../webgpu_util-BdovYhXr.js";
+import { W as y } from "../backend_webgpu-B0Vls736.js";
+import { compileProgram as P } from "./webgpu_program.js";
+import { m as M } from "../webgpu_program-CAE4RICo.js";
 const b = (o, s) => {
   const i = o.limits.maxComputeWorkgroupsPerDimension, t = s.dispatchLayout, a = s.dispatch;
   if (a.every((r) => r <= i))
@@ -17,7 +17,7 @@ const b = (o, s) => {
     () => "Total dispatch size exceeds WebGPU maximum."
   ), [e, e, e]) : [e, e, 1];
 };
-class z extends P {
+class z extends y {
   subgroupMaxSize;
   subgroupMinSize;
   constructor(s, i) {
@@ -42,7 +42,7 @@ class z extends P {
     });
     s.shaderKey = M(s, r, e);
     const d = m().getBool("WEBGPU_ENGINE_COMPILE_ONLY");
-    return s.shaderKey in this.pipelineCache || (this.pipelineCache[s.shaderKey] = y(
+    return s.shaderKey in this.pipelineCache || (this.pipelineCache[s.shaderKey] = P(
       this.device,
       s,
       r,

package/dist/patches/webgpu_base.js

@@ -1,4 +1,4 @@
-import { f as n } from "../index-DOvlwCh-.js";
+import { f as n } from "../index-D0RBWjq8.js";
 import p from "./webgpu_backend.js";
 function c(a) {
   n(

package/dist/patches/webgpu_program.js

@@ -1,8 +1,8 @@
-import { j as z, l as F, n as O, o as _ } from "../index-DOvlwCh-.js";
-import { g as A, d as b, a as x, b as g, s as L, t as l, c as y } from "../webgpu_program-DuOXPQol.js";
+import { ab as z, y as F, ac as O, ad as _ } from "../index-D0RBWjq8.js";
+import { g as A, d as b, a as x, b as g, s as L, t as l, c as y } from "../webgpu_program-CAE4RICo.js";
 var N = /* @__PURE__ */ ((s) => (s[s.FROM_PIXELS = 0] = "FROM_PIXELS", s[s.DRAW = 1] = "DRAW", s))(N || {});
 const H = (s, t, e, o, u) => {
-  const a = { dtype: o.dtype, shape: o.shape }, n = j(e, a, t), r = s.createShaderModule({ code: n, label: t.constructor.name });
+  const a = { dtype: o.dtype, shape: o.shape }, n = D(e, a, t), r = s.createShaderModule({ code: n, label: t.constructor.name });
   let d = z().get("WEBGPU_PRINT_SHADER");
   if (d !== "") {
     d = d.toLowerCase();
@@ -19,7 +19,7 @@ const H = (s, t, e, o, u) => {
     layout: "auto"
   });
 };
-function j(s, t, e) {
+function D(s, t, e) {
   const o = [], u = e.workgroupSize[0] * e.workgroupSize[1] * e.workgroupSize[2];
   if (e.outputComponent = e.outputComponent ? e.outputComponent : 1, o.push(`
 
@@ -93,7 +93,7 @@ function j(s, t, e) {
     e.subgroups ? "enable subgroups;" : "",
     C,
     o.join(`
-`) + D,
+`) + T,
     x(t.shape),
     f,
     W(t.shape.length)
@@ -165,12 +165,12 @@ const C = `
     let floatToUint: vec4<u32> = bitcast<vec4<u32>>(val);
     return (floatToUint & vec4<u32>(0x7fffffffu)) > vec4<u32>(0x7f800000u);
   }
-`, D = `
+`, T = `
   fn isinf(val: f32) -> bool {
     return abs(val) == uniforms.INFINITY;
   }
 `;
-function T(s, t) {
+function j(s, t) {
   const e = s.name, o = s.shape.length, u = g(o), a = "get" + e.charAt(0).toUpperCase() + e.slice(1), n = ["d0", "d1", "d2", "d3", "d4", "d5"].slice(0, o), r = n.map((v) => `${v} : i32`).join(", ");
   if (o < 1)
     return `
@@ -237,7 +237,7 @@ function E(s, t, e, o) {
 `;
 }
 function U(s, t, e, o) {
-  let u = T(s, e);
+  let u = j(s, e);
   return s.shape.length <= t.length && (u += E(s, t, e, o)), u;
 }
 function B(s, t) {

package/dist/{random_width-or-CEftb.js → random_width-UGQn4OWb.js}

@@ -1,36 +1,36 @@
-import { A as tt, B as R, l as S, E as X, cA as li, X as L, cB as ui, cC as Oa, cD as Ra, cE as ci, af as Yt, b0 as ut, c as U, C as _a, at as Ba, cF as Wa, x as z, L as Ga, a3 as Lt, a6 as An, cG as In, cH as Pa, cI as Ua, cJ as Va, cK as ja, cL as Ha, cM as qa, cN as Ka, cO as Za, cP as Ja, bR as Xa, m as f, c9 as Ya, y as Qt, b as Q, w as W, ca as Qa, bX as to, a4 as lt, cQ as eo, bq as so, a7 as Y, cb as no, cc as io, cd as ro, cf as ao, ce as oo, cg as lo, cR as uo, cS as co, br as ho, M as po, bt as fo, cT as mo, bU as go, b_ as bo, F as yo, cU as wo, H as ko, bv as No, bw as xo, cV as vo, bx as So, by as Ao, bA as Io, bB as Co, ci as Do, cW as zo, cX as To, aI as Eo, cY as Lo, bD as $o, an as Fo, I as Mo, b$ as Oo, J as Ro, c0 as _o, bu as Bo, K as Wo, b2 as Go, aQ as Po, cj as Uo, ck as Vo, cl as jo, aJ as Ho, b5 as qo, ao as Ko, cZ as Zo, c_ as Jo, cm as Xo, am as Yo, c1 as Qo, c$ as tl, d0 as el, bG as sl, b7 as nl, h as hi, a$ as il, b8 as rl, co as al, N as ol, c2 as ll, aq as ul, bH as cl, bI as hl, W as pl, bJ as dl, d1 as fl, z as Ws, aK as ml, c3 as gl, aT as bl, cp as yl, aL as wl, P as kl, R as Nl, bd as xl, d2 as vl, be as Sl, d3 as Al, bL as Il, ba as Cl, bM as Dl, a_ as zl, bN as Tl, aH as El, cq as Ll, bO as $l, bP as Fl, S as Ml, Q as Ol, bE as Rl, bK as _l, U as Bl, c5 as Wl, d4 as Gl, bb as Pl, aM as Ul, c7 as Vl, Y as jl, cu as Hl, bs as ql, T as Kl, as as Zl, bc as Jl, bQ as Xl, cz as Ne, d5 as Yl, u as Ql, d6 as v, t as y, d7 as Me, d8 as Oe, n as $t, d as q, j as tu, d9 as Cn, k as _t, aZ as ze, q as eu, D as su, a8 as we, O as nu, da as iu, a as Dn, db as ru, dc as zn, dd as au } from "./index-DOvlwCh-.js";
-import { c as te, k as Gs, m as Tn, b as ou, t as hs, a as Ps, s as Us, l as lu, p as uu, e as cu } from "./step-CINUs5QB.js";
-import { n as pt, t as P } from "./transpose-ClWiBS_b.js";
-import { r as C } from "./reshape-ByE68wS9.js";
-import { s as _ } from "./sum-DWAtNGez.js";
-import { m as Mt } from "./mat_mul-SjpJRLyL.js";
-import { j as Vs, o as Xe, G as pi, b as hu, s as pu, D as du, t as fu, A as mu, F as gu, e as Ht, q as En, r as Ye, l as bu, c as di, p as yu, z as fi, f as wu, x as ku, a as js, i as Qe, y as vt, B as Nu, E as Hs, w as xu, v as vu, n as Su, C as Au, k as Iu, u as Cu, h as Ln, g as Du, m as zu, d as Tu } from "./unsorted_segment_sum-BDDhB_E6.js";
-import { d as st, w as qt, b as Re, c as _e, l as qs, g as Wt, a as Eu, u as ts, f as Lu, m as mi, e as Ts, j as $u } from "./resize_nearest_neighbor-B19mCEg2.js";
-import { s as Vt } from "./split-CVLc0w--.js";
-import { e as Es, a as gi, g as Ks, c as Fu } from "./axis_util-BaG7mf5A.js";
-import { e as xe, a as le, m as Te, l as Mu } from "./log_sum_exp-ngO0-4pK.js";
-import { t as Ce } from "./tile-mbfagpsB.js";
-import { s as ps } from "./stack-Cf4n9h0N.js";
-import { o as he } from "./ones-D2rT0xk2.js";
-import { s as me } from "./slice-BfEGSH82.js";
-import { f as bi } from "./floor-D5QdR_le.js";
-import { z as ft } from "./zeros-KnWaWf-X.js";
-import { s as Ou, b as Ru, c as _u, g as Bu, a as Wu, S as Gu } from "./selu_util-kbhpTdYD.js";
-import { p as Pu } from "./slice_util-uTKwiEpW.js";
-import { c as Zs } from "./concat-BV8bt5H-.js";
-import { e as ue } from "./expand_dims-DT4tEPwA.js";
-import { g as Uu } from "./gather-D3JcZUaI.js";
-import { V as d, N as B, r as ds, c as Js, a as et, b as ge, e as Be, s as Xs, g as yi, f as wi, t as Ot, R as zt, h as ht, A as Gt, i as V, n as oe, j as nt, k as Vu, l as ju, m as We, o as Pt, p as Tt, q as ae, u as jt, v as qe, w as ce, x as Hu, y as es, z as fs, B as ki, C as St, D as $n, E as qu, F as Ku, G as Zu, H as Kt, I as Ju, J as Ys, K as It, L as Ke, M as Xu, O as ot, P as Ni, Q as mt, S as ee, T as Fn, U as ke, d as Et, W as Mn, X as Ge, Y as xi, Z as Qs, _ as Yu, $ as Qu, a0 as vi, a1 as tc } from "./tfjs_backend-y1cvNhLA.js";
-import { s as tn } from "./squeeze-C7Z2srUo.js";
-import { t as Ls } from "./tensor1d-D11P_7Dp.js";
-import { r as Pe } from "./relu-CP0ZcxWO.js";
-import { c as At } from "./clip_by_value-BnO7-a88.js";
-import { s as Si } from "./softmax-CA5jFsLR.js";
-import { M as ec, a as ms } from "./dropout-BcvN9JYi.js";
-import { e as sc, l as nc, i as Rt } from "./ops-B5yanEdW.js";
-import { t as ic } from "./tensor-DJoc7gJU.js";
-import { r as rc } from "./range-BklejeeW.js";
-import { v as ac } from "./variable-WawDEaAb.js";
+import { q as tt, u as R, y as S, E as X, cA as li, N as L, cB as ui, cC as Oa, cD as Ra, cE as ci, af as Yt, b0 as ut, c as U, w as _a, at as Ba, cF as Wa, n as z, D as Ga, $ as Lt, a2 as An, cG as In, cH as Pa, cI as Ua, cJ as Va, cK as ja, cL as Ha, cM as qa, cN as Ka, cO as Za, cP as Ja, bR as Xa, m as f, c9 as Ya, o as Qt, b as Q, l as W, ca as Qa, bX as to, a0 as lt, cQ as eo, bq as so, a3 as Y, cb as no, cc as io, cd as ro, cf as ao, ce as oo, cg as lo, cR as uo, cS as co, br as ho, H as po, bt as fo, cT as mo, bU as go, b_ as bo, C as yo, cU as wo, A as ko, bv as No, bw as xo, cV as vo, bx as So, by as Ao, bA as Io, bB as Co, ci as Do, cW as zo, cX as To, aI as Eo, cY as Lo, bD as $o, an as Fo, B as Mo, b$ as Oo, F as Ro, c0 as _o, bu as Bo, G as Wo, b2 as Go, aQ as Po, cj as Uo, ck as Vo, cl as jo, aJ as Ho, b5 as qo, ao as Ko, cZ as Zo, c_ as Jo, cm as Xo, am as Yo, c1 as Qo, c$ as tl, d0 as el, bG as sl, b7 as nl, V as hi, a$ as il, b8 as rl, co as al, M as ol, c2 as ll, aq as ul, bH as cl, bI as hl, P as pl, bJ as dl, d1 as fl, p as Ws, aK as ml, c3 as gl, aT as bl, cp as yl, aL as wl, I as kl, R as Nl, bd as xl, d2 as vl, be as Sl, d3 as Al, bL as Il, ba as Cl, bM as Dl, a_ as zl, bN as Tl, aH as El, cq as Ll, bO as $l, bP as Fl, S as Ml, J as Ol, bE as Rl, bK as _l, K as Bl, c5 as Wl, d4 as Gl, bb as Pl, aM as Ul, c7 as Vl, Q as jl, cu as Hl, bs as ql, T as Kl, as as Zl, bc as Jl, bQ as Xl, cz as Ne, d5 as Yl, j as Ql, d6 as v, t as y, d7 as Me, d8 as Oe, ac as $t, d as q, ab as tu, d9 as Cn, k as _t, aZ as ze, i as eu, x as su, a4 as we, O as nu, da as iu, a as Dn, db as ru, dc as zn, dd as au } from "./index-D0RBWjq8.js";
+import { c as te, k as Gs, m as Tn, b as ou, t as hs, a as Ps, s as Us, l as lu, p as uu, e as cu } from "./step-dXR33iOg.js";
+import { n as pt, t as P } from "./transpose-JawVKyZy.js";
+import { r as C } from "./reshape-CkjKPPqB.js";
+import { s as _ } from "./sum-BdplSvq_.js";
+import { m as Mt } from "./mat_mul-DxpNTCRz.js";
+import { j as Vs, o as Xe, G as pi, b as hu, s as pu, D as du, t as fu, A as mu, F as gu, e as Ht, q as En, r as Ye, l as bu, c as di, p as yu, z as fi, f as wu, x as ku, a as js, i as Qe, y as vt, B as Nu, E as Hs, w as xu, v as vu, n as Su, C as Au, k as Iu, u as Cu, h as Ln, g as Du, m as zu, d as Tu } from "./unsorted_segment_sum-LAbmE9G4.js";
+import { d as st, w as qt, b as Re, c as _e, l as qs, g as Wt, a as Eu, u as ts, f as Lu, m as mi, e as Ts, j as $u } from "./resize_nearest_neighbor-DB8k9KN_.js";
+import { s as Vt } from "./split-BNz5jcGc.js";
+import { e as Es, a as gi, g as Ks, c as Fu } from "./axis_util-DofAuy0p.js";
+import { e as xe, a as le, m as Te, l as Mu } from "./log_sum_exp-VLZgbFAH.js";
+import { t as Ce } from "./tile-CvN_LyVr.js";
+import { s as ps } from "./stack-WJK22CFn.js";
+import { o as he } from "./ones-BX_wEgzB.js";
+import { s as me } from "./slice-Aqy7KbJh.js";
+import { f as bi } from "./floor-DIb-lN_u.js";
+import { z as ft } from "./zeros-DeiE2zTa.js";
+import { s as Ou, b as Ru, c as _u, g as Bu, a as Wu, S as Gu } from "./selu_util-BuLbmbrl.js";
+import { p as Pu } from "./slice_util-D8CQRenR.js";
+import { c as Zs } from "./concat-C6X3AAlQ.js";
+import { e as ue } from "./expand_dims-BzfJK2uc.js";
+import { g as Uu } from "./gather-BcO5UQNJ.js";
+import { V as d, N as B, r as ds, c as Js, a as et, b as ge, e as Be, s as Xs, g as yi, f as wi, t as Ot, R as zt, h as ht, A as Gt, i as V, n as oe, j as nt, k as Vu, l as ju, m as We, o as Pt, p as Tt, q as ae, u as jt, v as qe, w as ce, x as Hu, y as es, z as fs, B as ki, C as St, D as $n, E as qu, F as Ku, G as Zu, H as Kt, I as Ju, J as Ys, K as It, L as Ke, M as Xu, O as ot, P as Ni, Q as mt, S as ee, T as Fn, U as ke, d as Et, W as Mn, X as Ge, Y as xi, Z as Qs, _ as Yu, $ as Qu, a0 as vi, a1 as tc } from "./tfjs_backend-Vi4JfLzT.js";
+import { s as tn } from "./squeeze--YMgaAAf.js";
+import { t as Ls } from "./tensor1d-LxP9asMm.js";
+import { r as Pe } from "./relu-Cf80uA2p.js";
+import { c as At } from "./clip_by_value-Dn5tzexi.js";
+import { s as Si } from "./softmax-faLoUZVT.js";
+import { M as ec, a as ms } from "./dropout-OxuaJz6z.js";
+import { e as sc, l as nc, i as Rt } from "./ops-FJapAPfm.js";
+import { t as ic } from "./tensor-BQqrDvpx.js";
+import { r as rc } from "./range-CuGvVN2c.js";
+import { v as ac } from "./variable-DQ9yYgEU.js";
 function oc(n, t, e, s, i, r = "NDHWC") {
   const a = R(n, "x", "avgPool3d", "float32");
   let o = a, l = !1;

package/dist/range-CuGvVN2c.js

@@ -0,0 +1,10 @@
+import { E as e, _ as f } from "./index-D0RBWjq8.js";
+function E(n, o, r = 1, a = "float32") {
+  if (r === 0)
+    throw new Error("Cannot have a step of zero");
+  const t = { start: n, stop: o, step: r, dtype: a };
+  return e.runKernel(f, {}, t);
+}
+export {
+  E as r
+};

package/dist/{relu-CP0ZcxWO.js → relu-Cf80uA2p.js}

@@ -1,4 +1,4 @@
-import { A as o, B as e, E as s, P as t } from "./index-DOvlwCh-.js";
+import { q as o, u as e, E as s, I as t } from "./index-D0RBWjq8.js";
 function u(r) {
   const n = { x: e(r, "x", "relu") };
   return s.runKernel(t, n);

package/dist/{reshape-ByE68wS9.js → reshape-CkjKPPqB.js}

@@ -1,4 +1,4 @@
-import { A as o, B as t, E as a, R as p } from "./index-DOvlwCh-.js";
+import { q as o, u as t, E as a, R as p } from "./index-D0RBWjq8.js";
 function c(r, s) {
   const e = { x: t(r, "x", "reshape", "string_or_numeric") }, n = { shape: s };
   return a.runKernel(p, e, n);