@genai-fi/nanogpt 0.8.5 → 0.9.0

package/dist/ops/webgpu/gelu.js

@@ -1,8 +1,8 @@
-import { f as i } from "../../index-DdmHGZjq.js";
-import { g as o } from "../../webgpu_program-Dhk9R5aG.js";
-import { f as s, c as p } from "../../webgpu_util-BqGnZg8t.js";
-const a = 0.7978845608028654, u = 0.044715;
-class h {
+import { f as s } from "../../index-BzFyqcy-.js";
+import { g as a } from "../../webgpu_program-DkQJOJSd.js";
+import { f as o, c as p } from "../../webgpu_util-pLEV9tks.js";
+const u = 0.7978845608028654, i = 0.044715;
+class d {
   outputShape;
   shaderKey;
   dispatchLayout;
@@ -11,32 +11,33 @@ class h {
   workgroupSize;
   size = !0;
   constructor(e) {
-    this.workgroupSize = [128, 1, 1], this.outputShape = e, this.dispatchLayout = s(this.outputShape), this.dispatch = p(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "unary_gelu";
+    this.workgroupSize = [128, 1, 1], this.outputShape = e, this.dispatchLayout = o(this.outputShape), this.dispatch = p(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "unary_gelu";
   }
   getUserCode() {
     return `
-      fn polyTanh(x: f32) -> f32 {
-         return select(tanh(x), sign(x), abs(x) > 15.0);
-      }
-      fn unaryOperation(x : f32) -> f32 {
-        let x3 = x * x * x;
-        var inner = fma(${u}, x3, x);
-        inner = ${a} * inner;
-        inner = polyTanh(inner);
-        inner = 0.5 * (1.0 + inner);
-        return x * inner;
-      }
-      ${o("index")} {
-        if (index < uniforms.size) {
-          let a = getAByOutputIndex(index);
-          setOutputAtIndex(index, unaryOperation(a));
-        }
-      }
+            // TODO: revisit after https://github.com/gpuweb/gpuweb/issues/4458 is resolved
+            fn tanhComplete(x: f32) -> f32 {
+                return select(tanh(x), sign(x), abs(x) > 15.0);
+            }
+            fn unaryOperation(x : f32) -> f32 {
+                let x3 = x * x * x;
+                var inner = fma(${i}, x3, x);
+                inner = ${u} * inner;
+                inner = tanhComplete(inner);
+                inner = 0.5 * (1.0 + inner);
+                return x * inner;
+            }
+            ${a("index")} {
+                if (index < uniforms.size) {
+                let a = getAByOutputIndex(index);
+                setOutputAtIndex(index, unaryOperation(a));
+                }
+            }
       `;
   }
 }
 function c(t) {
-  const { x: e } = t.inputs, n = t.backend, r = new h(e.shape);
+  const { x: e } = t.inputs, n = t.backend, r = new d(e.shape);
   return n.runWebGPUProgram(r, [e], "float32");
 }
 const l = {
@@ -44,7 +45,7 @@ const l = {
   backendName: "webgpu",
   kernelFunc: c
 };
-i(l);
+s(l);
 class x {
   // Inputs: dy, x
   variableNames = ["dy", "x"];
@@ -55,22 +56,23 @@ class x {
   workgroupSize = [128, 1, 1];
   size = !0;
   constructor(e) {
-    this.outputShape = e, this.dispatchLayout = s(this.outputShape), this.dispatch = p(this.dispatchLayout, this.outputShape, this.workgroupSize);
+    this.outputShape = e, this.dispatchLayout = o(this.outputShape), this.dispatch = p(this.dispatchLayout, this.outputShape, this.workgroupSize);
   }
   getUserCode() {
     return `
-            fn polyTanh(x: f32) -> f32 {
+            // TODO: revisit after https://github.com/gpuweb/gpuweb/issues/4458 is resolved
+            fn tanhComplete(x: f32) -> f32 {
                 return select(tanh(x), sign(x), abs(x) > 15.0);
             }
-            ${o("index")} {
+            ${a("index")} {
                 if (index < uniforms.size) {
                     let X  = getXByOutputIndex(index);
                     let x2 = X * X;
                     let x3 = x2 * X;
-                    let u  = ${a} * (X + ${u} * x3);
-                    let t  = polyTanh(u);
+                    let u  = ${u} * (X + ${i} * x3);
+                    let t  = tanhComplete(u);
                     let sech2 = 1.0 - t * t;
-                    let du_dx = ${a} * (1.0 + 3.0 * ${u} * x2);
+                    let du_dx = ${u} * (1.0 + 3.0 * ${i} * x2);
                     let dgelu = 0.5 * (1.0 + t) + 0.5 * X * sech2 * du_dx;
                     let DY = getDyByOutputIndex(index);
                     setOutputAtIndex(index, DY * dgelu);
@@ -79,15 +81,15 @@ class x {
   }
 }
 function g(t) {
-  const { dy: e, x: n } = t.inputs, r = t.backend, d = new x(n.shape);
-  return r.runWebGPUProgram(d, [e, n], "float32");
+  const { dy: e, x: n } = t.inputs, r = t.backend, h = new x(n.shape);
+  return r.runWebGPUProgram(h, [e, n], "float32");
 }
 const f = {
   kernelName: "GeluGrad",
   backendName: "webgpu",
   kernelFunc: g
 };
-i(f);
+s(f);
 export {
-  h as GeluProgram
+  d as GeluProgram
 };

package/dist/ops/webgpu/normRMS.js

@@ -1,5 +1,5 @@
-import { f as n } from "../../webgpu_util-BqGnZg8t.js";
-import { f as p, a4 as h } from "../../index-DdmHGZjq.js";
+import { f as n } from "../../webgpu_util-pLEV9tks.js";
+import { f as p, a4 as h } from "../../index-BzFyqcy-.js";
 import { createReduceInfo as u, reduce as c, createReductionShader as m } from "./utils/reductions.js";
 class d {
   outputShape;

package/dist/ops/webgpu/normRMSGrad.js

@@ -1,9 +1,9 @@
-import { f, a4 as c, e as g } from "../../index-DdmHGZjq.js";
+import { f, a4 as c, e as g } from "../../index-BzFyqcy-.js";
 import { createReduceInfo as k } from "./utils/reductions.js";
-import { f as x } from "../../webgpu_util-BqGnZg8t.js";
-import { g as z } from "../../webgpu_program-Dhk9R5aG.js";
-import { s as d } from "../../slice-B7jXtPnp.js";
-import { s as w } from "../../sum-ByFINZgi.js";
+import { f as x } from "../../webgpu_util-pLEV9tks.js";
+import { g as z } from "../../webgpu_program-DkQJOJSd.js";
+import { s as d } from "../../slice-d0Vo9XTN.js";
+import { s as w } from "../../sum-CJ0ULhmt.js";
 class y {
   outputShape;
   shaderKey = "RMSNormGrad";

package/dist/ops/webgpu/qkv.js

@@ -1,6 +1,6 @@
-import { g as h } from "../../webgpu_program-Dhk9R5aG.js";
-import { f as c, c as d } from "../../webgpu_util-BqGnZg8t.js";
-import { f as p, a4 as m } from "../../index-DdmHGZjq.js";
+import { g as h } from "../../webgpu_program-DkQJOJSd.js";
+import { f as c, c as d } from "../../webgpu_util-pLEV9tks.js";
+import { f as p, a4 as m } from "../../index-BzFyqcy-.js";
 class l {
   variableNames = ["x", "kernel"];
   outputShape;

package/dist/ops/webgpu/rope.js

@@ -1,6 +1,6 @@
-import { g as c } from "../../webgpu_program-Dhk9R5aG.js";
-import { f as m, c as l } from "../../webgpu_util-BqGnZg8t.js";
-import { f as x, a4 as f } from "../../index-DdmHGZjq.js";
+import { g as c } from "../../webgpu_program-DkQJOJSd.js";
+import { f as m, c as l } from "../../webgpu_util-pLEV9tks.js";
+import { f as x, a4 as f } from "../../index-BzFyqcy-.js";
 class S {
   variableNames = ["x", "sin", "cos"];
   outputShape;

package/dist/ops/webgpu/scatterSub.js

@@ -1,6 +1,6 @@
-import { g as p } from "../../webgpu_program-Dhk9R5aG.js";
-import { f as u, c as d } from "../../webgpu_util-BqGnZg8t.js";
-import { f as h, a4 as o } from "../../index-DdmHGZjq.js";
+import { g as p } from "../../webgpu_program-DkQJOJSd.js";
+import { f as u, c as d } from "../../webgpu_util-pLEV9tks.js";
+import { f as h, a4 as o } from "../../index-BzFyqcy-.js";
 class b {
   variableNames = ["labels", "softmaxProbs", "dy"];
   outputShape;

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

@@ -1,7 +1,7 @@
-import { p as l, j as d } from "../../../index-DdmHGZjq.js";
-import { g as p } from "../../../webgpu_program-Dhk9R5aG.js";
-import { r as f } from "../../../Reshape-41YpQqEo.js";
-import { c as x } from "../../../axis_util-Did9235A.js";
+import { p as l, j as d } from "../../../index-BzFyqcy-.js";
+import { g as p } from "../../../webgpu_program-DkQJOJSd.js";
+import { r as f } from "../../../Reshape-DUqYftGC.js";
+import { c as x } from "../../../axis_util-TbGYJ208.js";
 function I(e, r, t, s, u) {
   return `
            fn DIV_CEIL(a : u32, b : u32) -> u32 {

package/dist/{ops-542ai2vG.js → ops-LuCMAnmM.js}

@@ -1,22 +1,22 @@
-import { C as f, D as c, n as p, E as h, aS as Is, l as D, aT as Ss, aU as Ts, aV as xs, p as Ms, aW as M, w as ls, x as As, b as k, aX as gs, aY as ws, X as qs, ap as Ds, Q as K, aZ as Bs, a_ as Gs, a$ as Os, b0 as zs, b1 as Ls, b2 as Rs, b3 as Ws, b4 as Ks, b5 as vs, a0 as q, b6 as Cs, b7 as Vs, b8 as Ps, b9 as Ys, ba as js, aJ as Fs, bb as Us, t as Zs, bc as Q, bd as Hs, be as Xs, bf as Js, bg as Qs, bh as sn, bi as nn, bj as en, bk as tn, a as g, q as O, o as S, h as rn, c as d, bl as on, H as ss, d as an, a4 as z, J as cn } from "./index-DdmHGZjq.js";
-import { t as w } from "./tensor1d-CtJq5BOv.js";
+import { E as f, F as c, n as p, G as h, aS as Is, l as D, aT as Ss, aU as Ts, aV as xs, p as Ms, aW as M, w as ls, x as As, b as k, aX as gs, aY as ws, X as qs, ap as Ds, Q as K, aZ as Bs, a_ as Gs, a$ as zs, b0 as Os, b1 as Ls, b2 as Rs, b3 as Ws, b4 as Ks, b5 as vs, a0 as q, b6 as Vs, b7 as Cs, b8 as Ps, b9 as Ys, ba as js, aJ as Fs, bb as Us, t as Zs, bc as Q, bd as Hs, be as Xs, bf as Js, bg as Qs, bh as sn, bi as nn, bj as en, bk as tn, a as g, q as z, o as S, h as rn, c as d, bl as on, J as ss, d as an, a4 as O, z as cn } from "./index-BzFyqcy-.js";
+import { t as w } from "./tensor1d-vML0r3q6.js";
 import { n as un, a as ln, b as pn } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { r as y } from "./reshape-WeJkT3ja.js";
-import { s as ds } from "./split-BPxr8_8m.js";
-import { s as E } from "./sum-ByFINZgi.js";
-import { b as ns } from "./broadcast_to-CUvOdOT5.js";
-import { s as x } from "./slice-B7jXtPnp.js";
-import { r as Z } from "./range-BcUvLuf5.js";
-import { t as fn } from "./tensor-DbqgIV9B.js";
-import { s as H } from "./stack-BNwLzE43.js";
-import { c as mn, z as hn } from "./zeros-Dnwix0p4.js";
-import { e as $s } from "./axis_util-Did9235A.js";
-import { m as es, a as ps, e as os, l as bn } from "./log_sum_exp-C8yFJfZz.js";
-import { c as ts } from "./concat-pHiVqR3L.js";
-import { m as G } from "./mat_mul-Dpy2mMRu.js";
-import { t as rs } from "./tensor2d-CObBWBkW.js";
-import { o as gn } from "./ones-BAqVh-eA.js";
-import { r as Es } from "./relu-B1AXs7p5.js";
+import { r as y } from "./reshape-CnIwVG1c.js";
+import { s as ds } from "./split-DK2k5eHf.js";
+import { s as E } from "./sum-CJ0ULhmt.js";
+import { b as ns } from "./broadcast_to-DzlNweb8.js";
+import { s as x } from "./slice-d0Vo9XTN.js";
+import { r as Z } from "./range-CWcz7xFA.js";
+import { t as fn } from "./tensor-CZr4dh61.js";
+import { s as H } from "./stack-DFatutCx.js";
+import { c as mn, z as hn } from "./zeros-Bj5rMYA7.js";
+import { e as $s } from "./axis_util-TbGYJ208.js";
+import { m as es, a as ps, e as os, l as bn } from "./log_sum_exp-DO6z8tSE.js";
+import { c as ts } from "./concat-B912vBbo.js";
+import { m as G } from "./mat_mul-DzjTFx-u.js";
+import { t as rs } from "./tensor2d-D76QGjF3.js";
+import { o as gn } from "./ones-tIJeHlq-.js";
+import { r as Es } from "./relu-BjCh_SYb.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -312,7 +312,7 @@ const An = /* @__PURE__ */ f({ greaterEqual_: Mn });
  */
 function wn(e) {
   const s = { input: c(e, "input", "imag") };
-  return h.runKernel(Os, s);
+  return h.runKernel(zs, s);
 }
 const qn = /* @__PURE__ */ f({ imag_: wn });
 /**
@@ -335,7 +335,7 @@ function Dn(e, r) {
   let s = c(e, "a", "less", "string_or_numeric"), t = c(r, "b", "less", "string_or_numeric");
   [s, t] = K(s, t), D(s.shape, t.shape);
   const n = { a: s, b: t };
-  return h.runKernel(zs, n);
+  return h.runKernel(Os, n);
 }
 const ms = /* @__PURE__ */ f({ less_: Dn });
 /**
@@ -381,7 +381,7 @@ function Gn(e) {
   const s = { x: c(e, "x", "log1p") };
   return h.runKernel(Rs, s);
 }
-const On = /* @__PURE__ */ f({ log1p_: Gn });
+const zn = /* @__PURE__ */ f({ log1p_: Gn });
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -398,11 +398,11 @@ const On = /* @__PURE__ */ f({ log1p_: Gn });
  * limitations under the License.
  * =============================================================================
  */
-function zn(e) {
+function On(e) {
   const s = { x: c(e, "x", "neg") };
   return h.runKernel(Ws, s);
 }
-const C = /* @__PURE__ */ f({ neg_: zn });
+const V = /* @__PURE__ */ f({ neg_: On });
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -467,7 +467,7 @@ function vn(e, r) {
   let s = c(e, "a", "minimum"), t = c(r, "b", "minimum");
   [s, t] = K(s, t), s.dtype === "bool" && (s = q(s, "int32"), t = q(t, "int32")), D(s.shape, t.shape);
   const n = { a: s, b: t };
-  return h.runKernel(Cs, n);
+  return h.runKernel(Vs, n);
 }
 const as = /* @__PURE__ */ f({ minimum_: vn });
 /**
@@ -486,13 +486,13 @@ const as = /* @__PURE__ */ f({ minimum_: vn });
  * limitations under the License.
  * =============================================================================
  */
-function Cn(e, r) {
+function Vn(e, r) {
   let s = c(e, "a", "notEqual", "string_or_numeric"), t = c(r, "b", "notEqual", "string_or_numeric");
   [s, t] = K(s, t), D(s.shape, t.shape);
   const n = { a: s, b: t };
-  return h.runKernel(Vs, n);
+  return h.runKernel(Cs, n);
 }
-const Vn = /* @__PURE__ */ f({ notEqual_: Cn });
+const Cn = /* @__PURE__ */ f({ notEqual_: Vn });
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -627,7 +627,7 @@ function Qn(e, r, s) {
   const n = { x: t }, o = { perm: r };
   return t.dtype === "complex64" ? Zs(() => {
     let a = Yn(t), i = qn(t);
-    return a = h.runKernel(Q, { x: a }, o), i = h.runKernel(Q, { x: i }, o), s && (i = C(i)), mn(a, i);
+    return a = h.runKernel(Q, { x: a }, o), i = h.runKernel(Q, { x: i }, o), s && (i = V(i)), mn(a, i);
   }) : h.runKernel(Q, n, o);
 }
 const hs = /* @__PURE__ */ f({ transpose_: Qn });
@@ -782,7 +782,7 @@ const ue = /* @__PURE__ */ f({ rotateWithOffset_: ce });
  * limitations under the License.
  * =============================================================================
  */
-function V(e, r, s, t, n, o) {
+function C(e, r, s, t, n, o) {
   t == null && (t = 0.5), n == null && (n = Number.NEGATIVE_INFINITY), o == null && (o = 0);
   const a = e.shape[0];
   return s = Math.min(s, a), p(0 <= t && t <= 1, () => `iouThreshold must be in [0, 1], but was '${t}'`), p(e.rank === 2, () => `boxes must be a 2D tensor, but was of rank '${e.rank}'`), p(e.shape[1] === 4, () => `boxes must have 4 columns, but 2nd dimension was ${e.shape[1]}`), p(r.rank === 1, () => "scores must be a 1D tensor"), p(r.shape[0] === a, () => `scores has incompatible shape with boxes. Expected ${a}, but was ${r.shape[0]}`), p(0 <= o && o <= 1, () => `softNmsSigma must be in [0, 1], but was '${o}'`), { maxOutputSize: s, iouThreshold: t, scoreThreshold: n, softNmsSigma: o };
@@ -804,7 +804,7 @@ function V(e, r, s, t, n, o) {
  * =============================================================================
  */
 function le(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY) {
-  const o = c(e, "boxes", "nonMaxSuppression", "float32"), a = c(r, "scores", "nonMaxSuppression", "float32"), i = V(o, a, s, t, n);
+  const o = c(e, "boxes", "nonMaxSuppression", "float32"), a = c(r, "scores", "nonMaxSuppression", "float32"), i = C(o, a, s, t, n);
   s = i.maxOutputSize, t = i.iouThreshold, n = i.scoreThreshold;
   const l = { maxOutputSize: s, iouThreshold: t, scoreThreshold: n };
   return h.runKernel(Qs, { boxes: o, scores: a }, l);
@@ -827,7 +827,7 @@ const pe = /* @__PURE__ */ f({ nonMaxSuppression_: le });
  * =============================================================================
  */
 async function fe(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY) {
-  const o = c(e, "boxes", "nonMaxSuppressionAsync"), a = c(r, "scores", "nonMaxSuppressionAsync"), i = V(o, a, s, t, n);
+  const o = c(e, "boxes", "nonMaxSuppressionAsync"), a = c(r, "scores", "nonMaxSuppressionAsync"), i = C(o, a, s, t, n);
   s = i.maxOutputSize, t = i.iouThreshold, n = i.scoreThreshold;
   const l = await Promise.all([o.data(), a.data()]), u = l[0], m = l[1], { selectedIndices: b } = un(u, m, s, t, n);
   return o !== e && o.dispose(), a !== r && a.dispose(), w(b, "int32");
@@ -850,7 +850,7 @@ const me = fe;
  * =============================================================================
  */
 function he(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = 0) {
-  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = V(a, i, s, t, n, o);
+  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = C(a, i, s, t, n, o);
   s = l.maxOutputSize, t = l.iouThreshold, n = l.scoreThreshold, o = l.softNmsSigma;
   const u = { boxes: a, scores: i }, m = { maxOutputSize: s, iouThreshold: t, scoreThreshold: n, softNmsSigma: o }, b = h.runKernel(sn, u, m);
   return { selectedIndices: b[0], selectedScores: b[1] };
@@ -873,7 +873,7 @@ const be = /* @__PURE__ */ f({ nonMaxSuppressionWithScore_: he });
  * =============================================================================
  */
 async function ge(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = 0) {
-  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = V(a, i, s, t, n, o);
+  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = C(a, i, s, t, n, o);
   s = l.maxOutputSize, t = l.iouThreshold, n = l.scoreThreshold, o = l.softNmsSigma;
   const u = await Promise.all([a.data(), i.data()]), m = u[0], b = u[1], { selectedIndices: _, selectedScores: $ } = ln(m, b, s, t, n, o);
   return a !== e && a.dispose(), i !== r && i.dispose(), {
@@ -899,7 +899,7 @@ const de = ge;
  * =============================================================================
  */
 function $e(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = !1) {
-  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = V(
+  const a = c(e, "boxes", "nonMaxSuppression"), i = c(r, "scores", "nonMaxSuppression"), l = C(
     a,
     i,
     s,
@@ -933,7 +933,7 @@ const Ee = /* @__PURE__ */ f({ nonMaxSuppressionPadded_: $e });
  * =============================================================================
  */
 async function _e(e, r, s, t = 0.5, n = Number.NEGATIVE_INFINITY, o = !1) {
-  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = V(
+  const a = c(e, "boxes", "nonMaxSuppressionAsync"), i = c(r, "scores", "nonMaxSuppressionAsync"), l = C(
     a,
     i,
     s,
@@ -1020,7 +1020,7 @@ function Te(e, r = "binary", s = !1, t = 0.5) {
   if (p(n.rank === 3, () => `Error in threshold: image must be rank 3,but got rank ${n.rank}.`), p(n.shape[2] === 3 || n.shape[2] === 1, () => `Error in threshold: image color channel must be equal to 3 or 1but got ${n.shape[2]}.`), p(n.dtype === "int32" || n.dtype === "float32", () => `Error in dtype: image dtype must be int32 or float32,but got dtype ${n.dtype}.`), p(r === "otsu" || r === "binary", () => `Method must be binary or otsu, but was ${r}`), n.shape[2] === 3) {
     [m, b, _] = ds(n, [1, 1, 1], -1);
     const T = g(m, o), B = g(b, a), R = g(_, i);
-    $ = O(O(T, B), R);
+    $ = z(z(T, B), R);
   } else
     $ = e;
   if (r === "otsu") {
@@ -1036,7 +1036,7 @@ function xe(e, r) {
     o = x(e, 0, b + 1), a = x(e, b + 1), u = S(E(o), r), m = S(E(a), r);
     const _ = E(g(o, Z(0, o.size)));
     i = S(_, E(o));
-    const $ = rn(a.shape, o.size), N = O(Z(0, a.size), $), A = g(a, N);
+    const $ = rn(a.shape, o.size), N = z(Z(0, a.size), $), A = g(a, N);
     l = S(E(A), E(a));
     const T = d(i, l), B = d(i, l), R = g(u, m);
     n = g(g(R, T), B);
@@ -1091,7 +1091,7 @@ function qe(e, r, s) {
   const n = t.shape, [o, a] = t.shape.slice(-2);
   let i, l;
   typeof r == "number" ? (p(r % 1 === 0, () => `bandPart(): numLower must be an integer, got ${r}.`), p(r <= o, () => `bandPart(): numLower (${r}) must not be greater than the number of rows (${o}).`), i = c(r < 0 ? o : r, "numLower", "bandPart")) : (p(r.dtype === "int32", () => "bandPart(): numLower's dtype must be an int32."), i = v(ms(r, 0), o, as(r, o))), typeof s == "number" ? (p(s % 1 === 0, () => `bandPart(): numUpper must be an integer, got ${s}.`), p(s <= a, () => `bandPart(): numUpper (${s}) must not be greater than the number of columns (${a}).`), l = c(s < 0 ? a : s, "numUpper", "bandPart")) : (p(s.dtype === "int32", () => "bandPart(): numUpper's dtype must be an int32."), l = v(ms(s, 0), a, as(s, a)));
-  const u = y(Z(0, o, 1, "int32"), [-1, 1]), m = Z(0, a, 1, "int32"), b = d(u, m), _ = Rn(ys(b, i), An(b, C(l))), $ = hn([o, a], t.dtype);
+  const u = y(Z(0, o, 1, "int32"), [-1, 1]), m = Z(0, a, 1, "int32"), b = d(u, m), _ = Rn(ys(b, i), An(b, V(l))), $ = hn([o, a], t.dtype);
   return y(H(Ns(y(t, [-1, o, a])).map((N) => v(_, N, $))), n);
 }
 const De = /* @__PURE__ */ f({ bandPart_: qe });
@@ -1151,7 +1151,7 @@ const Ge = /* @__PURE__ */ f({ gramSchmidt_: Be });
  * limitations under the License.
  * =============================================================================
  */
-function Oe(e, r = !1) {
+function ze(e, r = !1) {
   if (p(e.rank >= 2, () => `qr() requires input tensor to have a rank >= 2, but got rank ${e.rank}`), e.rank === 2)
     return bs(e, r);
   {
@@ -1184,7 +1184,7 @@ function bs(e, r = !1) {
           a,
           x(R, [1, 0], [R.shape[0] - 1, R.shape[1]])
         ], 0);
-        const Y = C(S(G(T, B), N)), j = x(o, [u, 0], [s - u, t]), X = g(Y, i), cs = hs(i);
+        const Y = V(S(G(T, B), N)), j = x(o, [u, 0], [s - u, t]), X = g(Y, i), cs = hs(i);
         if (u === 0)
           o = d(j, G(X, G(cs, j)));
         else {
@@ -1204,7 +1204,7 @@ function bs(e, r = !1) {
     return !r && s > t && (n = x(n, [0, 0], [s, t]), o = x(o, [0, 0], [t, t])), [n, o];
   });
 }
-const ze = /* @__PURE__ */ f({ qr_: Oe });
+const Oe = /* @__PURE__ */ f({ qr_: ze });
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1246,7 +1246,7 @@ function Le(e, r, s = I.SUM_BY_NONZERO_WEIGHTS) {
     if (n == null)
       return S(E(o), k(t.size));
     {
-      const a = g(n, gn(t.shape)), i = q(E(Vn(a, k(0))), "float32");
+      const a = g(n, gn(t.shape)), i = q(E(Cn(a, k(0))), "float32");
       return S(E(o), i);
     }
   }
@@ -1272,7 +1272,7 @@ const L = /* @__PURE__ */ f({ computeWeightedLoss_: Le });
 function Re(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
   const n = c(e, "labels", "absoluteDifference"), o = c(r, "predictions", "absoluteDifference");
   let a = null;
-  s != null && (a = c(s, "weights", "absoluteDifference")), z(n.shape, o.shape, "Error in absoluteDifference: ");
+  s != null && (a = c(s, "weights", "absoluteDifference")), O(n.shape, o.shape, "Error in absoluteDifference: ");
   const i = M(d(n, o));
   return L(i, a, t);
 }
@@ -1280,22 +1280,22 @@ const We = /* @__PURE__ */ f({ absoluteDifference_: Re });
 function Ke(e, r, s, t, n = I.SUM_BY_NONZERO_WEIGHTS) {
   const o = c(e, "labels", "cosineDistance"), a = c(r, "predictions", "cosineDistance");
   let i = null;
-  t != null && (i = c(t, "weights", "cosineDistance")), z(o.shape, a.shape, "Error in cosineDistance: ");
+  t != null && (i = c(t, "weights", "cosineDistance")), O(o.shape, a.shape, "Error in cosineDistance: ");
   const l = k(1), u = d(l, E(g(o, a), s, !0));
   return L(u, i, n);
 }
 const ve = /* @__PURE__ */ f({ cosineDistance_: Ke });
-function Ce(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
+function Ve(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
   let n = c(e, "labels", "hingeLoss");
   const o = c(r, "predictions", "hingeLoss");
   let a = null;
-  s != null && (a = c(s, "weights", "hingeLoss")), z(n.shape, o.shape, "Error in hingeLoss: ");
+  s != null && (a = c(s, "weights", "hingeLoss")), O(n.shape, o.shape, "Error in hingeLoss: ");
   const i = k(1);
   n = d(g(k(2), n), i);
   const l = Es(d(i, g(n, o)));
   return L(l, a, t);
 }
-const Ve = /* @__PURE__ */ f({ hingeLoss_: Ce });
+const Ce = /* @__PURE__ */ f({ hingeLoss_: Ve });
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1315,8 +1315,8 @@ const Ve = /* @__PURE__ */ f({ hingeLoss_: Ce });
 function Pe(e, r, s, t = 1, n = I.SUM_BY_NONZERO_WEIGHTS) {
   const o = c(e, "labels", "huberLoss"), a = c(r, "predictions", "huberLoss");
   let i = null;
-  s != null && (i = c(s, "weights", "huberLoss")), z(o.shape, a.shape, "Error in huberLoss: ");
-  const l = k(t), u = M(d(a, o)), m = as(u, l), b = d(u, m), _ = O(g(k(0.5), gs(m)), g(l, b));
+  s != null && (i = c(s, "weights", "huberLoss")), O(o.shape, a.shape, "Error in huberLoss: ");
+  const l = k(t), u = M(d(a, o)), m = as(u, l), b = d(u, m), _ = z(g(k(0.5), gs(m)), g(l, b));
   return L(_, i, n);
 }
 const Ye = /* @__PURE__ */ f({ huberLoss_: Pe });
@@ -1339,8 +1339,8 @@ const Ye = /* @__PURE__ */ f({ huberLoss_: Pe });
 function je(e, r, s, t = 1e-7, n = I.SUM_BY_NONZERO_WEIGHTS) {
   const o = c(e, "labels", "logLoss"), a = c(r, "predictions", "logLoss");
   let i = null;
-  s != null && (i = c(s, "weights", "logLoss")), z(o.shape, a.shape, "Error in logLoss: ");
-  const l = k(1), u = k(t), m = C(g(o, ps(O(a, u)))), b = g(d(l, o), ps(O(d(l, a), u))), _ = d(m, b);
+  s != null && (i = c(s, "weights", "logLoss")), O(o.shape, a.shape, "Error in logLoss: ");
+  const l = k(1), u = k(t), m = V(g(o, ps(z(a, u)))), b = g(d(l, o), ps(z(d(l, a), u))), _ = d(m, b);
   return L(_, i, n);
 }
 const Fe = /* @__PURE__ */ f({ logLoss_: je });
@@ -1363,7 +1363,7 @@ const Fe = /* @__PURE__ */ f({ logLoss_: je });
 function Ue(e, r, s, t = I.SUM_BY_NONZERO_WEIGHTS) {
   const n = c(e, "labels", "meanSquaredError"), o = c(r, "predictions", "meanSquaredError");
   let a = null;
-  s != null && (a = c(s, "weights", "meanSquaredError")), z(n.shape, o.shape, "Error in meanSquaredError: ");
+  s != null && (a = c(s, "weights", "meanSquaredError")), O(n.shape, o.shape, "Error in meanSquaredError: ");
   const i = Zn(n, o);
   return L(i, a, t);
 }
@@ -1386,17 +1386,17 @@ const Ze = /* @__PURE__ */ f({ meanSquaredError_: Ue });
  */
 function He(e, r) {
   const s = c(e, "labels", "sigmoidCrossEntropyWithLogits"), t = c(r, "logits", "sigmoidCrossEntropyWithLogits");
-  z(s.shape, t.shape, "Error in sigmoidCrossEntropyWithLogits: ");
-  const n = Es(t), o = g(t, s), a = On(os(C(M(t))));
-  return O(d(n, o), a);
+  O(s.shape, t.shape, "Error in sigmoidCrossEntropyWithLogits: ");
+  const n = Es(t), o = g(t, s), a = zn(os(V(M(t))));
+  return z(d(n, o), a);
 }
 function Xe(e, r, s, t = 0, n = I.SUM_BY_NONZERO_WEIGHTS) {
   let o = c(e, "multiClassLabels", "sigmoidCrossEntropy");
   const a = c(r, "logits", "sigmoidCrossEntropy");
   let i = null;
-  if (s != null && (i = c(s, "weights", "sigmoidCrossEntropy")), z(o.shape, a.shape, "Error in sigmoidCrossEntropy: "), t > 0) {
+  if (s != null && (i = c(s, "weights", "sigmoidCrossEntropy")), O(o.shape, a.shape, "Error in sigmoidCrossEntropy: "), t > 0) {
     const u = k(t), m = k(1), b = k(0.5);
-    o = O(g(o, d(m, u)), g(b, u));
+    o = z(g(o, d(m, u)), g(b, u));
   }
   const l = He(o, a);
   return L(l, i, n);
@@ -1424,7 +1424,7 @@ function Qe(e, r, s = -1) {
   return cn((n, o, a) => {
     const l = bn(o, [s], !0), u = d(q(o, "float32"), l);
     a([n, u]);
-    const m = C(g(u, n));
+    const m = V(g(u, n));
     return { value: E(m, [s]), gradFunc: ($, N) => {
       const [A, T] = N, B = $s($.shape, [s]);
       return [
@@ -1438,9 +1438,9 @@ function st(e, r, s, t = 0, n = I.SUM_BY_NONZERO_WEIGHTS) {
   let o = c(e, "onehotLabels", "softmaxCrossEntropy");
   const a = c(r, "logits", "softmaxCrossEntropy");
   let i = null;
-  if (s != null && (i = c(s, "weights", "softmaxCrossEntropy")), z(o.shape, a.shape, "Error in softmaxCrossEntropy: "), t > 0) {
+  if (s != null && (i = c(s, "weights", "softmaxCrossEntropy")), O(o.shape, a.shape, "Error in softmaxCrossEntropy: "), t > 0) {
     const u = k(t), m = k(1), b = k(o.shape[1]);
-    o = O(g(o, d(m, u)), S(u, b));
+    o = z(g(o, d(m, u)), S(u, b));
   }
   const l = Qe(o, a);
   return L(l, i, n);
@@ -1481,12 +1481,12 @@ const kt = {
 }, yt = {
   bandPart: De,
   gramSchmidt: Ge,
-  qr: ze
+  qr: Oe
 }, Nt = {
   absoluteDifference: We,
   computeWeightedLoss: L,
   cosineDistance: ve,
-  hingeLoss: Ve,
+  hingeLoss: Ce,
   huberLoss: Ye,
   logLoss: Fe,
   meanSquaredError: Ze,
@@ -1505,13 +1505,13 @@ export {
   An as g,
   W as h,
   Tn as i,
-  On as j,
+  zn as j,
   as as k,
   yt as l,
   Kn as m,
-  C as n,
+  V as n,
   kt as o,
-  Vn as p,
+  Cn as p,
   fs as q,
   ks as r,
   Xn as s,

package/dist/{random_width-DKGeiFuR.js → random_width-CXVRloNK.js}

@@ -1,26 +1,26 @@
-import { C as z, D, E as O, bJ as La, bK as Fa, bL as Ci, n as b, bM as Ii, a0 as L, bN as Di, bO as $i, bP as Ti, bQ as zi, h as Ei, bR as Li, bS as Fi, bT as Oi, bU as Mi, bV as Oa, bW as Ri, bX as Ma, bY as _i, bZ as Ra, b_ as Bi, Q as hn, l as vt, bx as _a, b$ as Wi, c0 as Gi, J as Ge, c as V, a as w, c1 as Ba, c2 as Pi, c3 as Ui, p as ce, aX as pt, c4 as Vi, c5 as ji, c6 as Ki, c7 as Hi, c8 as qi, bD as Ji, c9 as Zi, ca as Yi, M as Wa, ap as Ga, aq as Pa, cb as Xi, cc as Ua, q as T, cd as Ps, ce as Va, cf as ja, j as pn, cg as Us, ch as Ka, ci as Ha, cj as qa, ck as Ja, cl as Za, cm as Ya, cn as Xa, bm as Qa, co as tl, w as he, b as et, o as U, cp as el, bt as nl, ax as ht, cq as sl, L as Q, cr as il, cs as rl, ct as ol, cu as al, cv as ll, cw as ul, cx as cl, cy as hl, U as pl, cz as dl, bq as fl, bw as ml, cA as gl, I as bl, cB as yl, ac as wl, cC as kl, cD as xl, cE as Nl, as as vl, cF as Al, ai as Sl, aY as Cl, by as Il, ao as Dl, bz as $l, G as Tl, a_ as zl, az as El, cG as Ll, cH as Fl, cI as Ol, at as Ml, b2 as Rl, aj as _l, cJ as Bl, cK as Wl, cL as Gl, ah as Pl, bA as Ul, cM as Vl, cN as jl, b5 as Kl, aV as Hl, b6 as ql, cO as Jl, Y as Zl, bB as Yl, b3 as Xl, P as Ql, cP as tu, x as rs, au as eu, bC as nu, aC as su, Z as iu, av as ru, _ as ou, V as au, bj as lu, cQ as uu, bk as cu, cR as hu, b9 as pu, aT as du, ar as fu, cS as mu, ad as gu, $ as bu, S as yu, W as wu, bF as ku, cT as xu, ba as Nu, aw as vu, bH as Au, a1 as Su, cU as Cu, X as Iu, bc as Du, bb as $u, cV as Oe, cW as Tu, i as zu, an as Vs, cX as Eu, t as x, aW as $e, cY as S, cZ as Ke, c_ as He, af as Vt, d as J, ag as Lu, c$ as js, k as Jt, H as Fu, T as Te, O as Ou, d0 as Mu, m as Ks, d1 as Ru, d2 as Hs, d3 as _u } from "./index-DdmHGZjq.js";
-import { k as ke, c as xt, o as ze, s as Qi, b as tr, d as Bu, m as er, t as In, l as nr, v as os, a as Wu, S as Gu, p as Pu, w as as, x as sr, y as Uu, z as Vu } from "./selu_util-BgUO9gHY.js";
-import { m as nt, n as wt, w as ne, b as qe, g as Je, c as ls, t as K, d as Ce, e as Yt, f as ju, u as dn, h as ye, i as Ku, l as Hu, j as qu, s as us, k as ir, o as qt, p as Hn, q as Ju } from "./ops-542ai2vG.js";
-import { r as N } from "./reshape-WeJkT3ja.js";
-import { s as W } from "./sum-ByFINZgi.js";
-import { m as ct } from "./mat_mul-Dpy2mMRu.js";
-import { s as Qt } from "./split-BPxr8_8m.js";
-import { s as Zu, c as rr } from "./sin-CPxad7Am.js";
-import { e as qn, g as or, h as cs, c as Yu } from "./axis_util-Did9235A.js";
-import { m as Ee, a as se, e as ie, l as Xu } from "./log_sum_exp-C8yFJfZz.js";
-import { s as Dn } from "./stack-BNwLzE43.js";
-import { o as xe } from "./ones-BAqVh-eA.js";
-import { s as Dt } from "./slice-B7jXtPnp.js";
-import { M as Qu, f as ar, r as tc, d as ec, a as $n } from "./dropout-CcKSfOYE.js";
-import { z as Nt } from "./zeros-Dnwix0p4.js";
-import { c as pe } from "./concat-pHiVqR3L.js";
-import { g as lr } from "./gather-CPg6ZlQA.js";
-import { t as fn } from "./tensor1d-CtJq5BOv.js";
-import { r as Ze } from "./relu-B1AXs7p5.js";
-import { s as ur } from "./softmax-BfsyI4As.js";
-import { t as nc } from "./tensor-DbqgIV9B.js";
-import { r as sc } from "./range-BcUvLuf5.js";
-import { v as ic } from "./variable-DPFOJyRG.js";
+import { E as z, F as D, G as O, bJ as La, bK as Fa, bL as Ci, n as b, bM as Ii, a0 as L, bN as Di, bO as $i, bP as Ti, bQ as zi, h as Ei, bR as Li, bS as Fi, bT as Oi, bU as Mi, bV as Oa, bW as Ri, bX as Ma, bY as _i, bZ as Ra, b_ as Bi, Q as hn, l as vt, bx as _a, b$ as Wi, c0 as Gi, z as Ge, c as V, a as w, c1 as Ba, c2 as Pi, c3 as Ui, p as ce, aX as pt, c4 as Vi, c5 as ji, c6 as Ki, c7 as Hi, c8 as qi, bD as Ji, c9 as Zi, ca as Yi, M as Wa, ap as Ga, aq as Pa, cb as Xi, cc as Ua, q as T, cd as Ps, ce as Va, cf as ja, j as pn, cg as Us, ch as Ka, ci as Ha, cj as qa, ck as Ja, cl as Za, cm as Ya, cn as Xa, bm as Qa, co as tl, w as he, b as et, o as U, cp as el, bt as nl, ax as ht, cq as sl, A as Q, cr as il, cs as rl, ct as ol, cu as al, cv as ll, cw as ul, cx as cl, cy as hl, U as pl, cz as dl, bq as fl, bw as ml, cA as gl, K as bl, cB as yl, ac as wl, cC as kl, cD as xl, cE as Nl, as as vl, cF as Al, ai as Sl, aY as Cl, by as Il, ao as Dl, bz as $l, H as Tl, a_ as zl, az as El, cG as Ll, cH as Fl, cI as Ol, at as Ml, b2 as Rl, aj as _l, cJ as Bl, cK as Wl, cL as Gl, ah as Pl, bA as Ul, cM as Vl, cN as jl, b5 as Kl, aV as Hl, b6 as ql, cO as Jl, Y as Zl, bB as Yl, b3 as Xl, P as Ql, cP as tu, x as rs, au as eu, bC as nu, aC as su, Z as iu, av as ru, _ as ou, V as au, bj as lu, cQ as uu, bk as cu, cR as hu, b9 as pu, aT as du, ar as fu, cS as mu, ad as gu, $ as bu, S as yu, W as wu, bF as ku, cT as xu, ba as Nu, aw as vu, bH as Au, a1 as Su, cU as Cu, X as Iu, bc as Du, bb as $u, cV as Oe, cW as Tu, i as zu, an as Vs, cX as Eu, t as x, aW as $e, cY as S, cZ as Ke, c_ as He, af as Vt, d as J, ag as Lu, c$ as js, k as Jt, J as Fu, T as Te, O as Ou, d0 as Mu, m as Ks, d1 as Ru, d2 as Hs, d3 as _u } from "./index-BzFyqcy-.js";
+import { k as ke, c as xt, o as ze, s as Qi, b as tr, d as Bu, m as er, t as In, l as nr, v as os, a as Wu, S as Gu, p as Pu, w as as, x as sr, y as Uu, z as Vu } from "./selu_util-OtRzVwW5.js";
+import { m as nt, n as wt, w as ne, b as qe, g as Je, c as ls, t as K, d as Ce, e as Yt, f as ju, u as dn, h as ye, i as Ku, l as Hu, j as qu, s as us, k as ir, o as qt, p as Hn, q as Ju } from "./ops-LuCMAnmM.js";
+import { r as N } from "./reshape-CnIwVG1c.js";
+import { s as W } from "./sum-CJ0ULhmt.js";
+import { m as ct } from "./mat_mul-DzjTFx-u.js";
+import { s as Qt } from "./split-DK2k5eHf.js";
+import { s as Zu, c as rr } from "./sin-gpDNRxE0.js";
+import { e as qn, g as or, h as cs, c as Yu } from "./axis_util-TbGYJ208.js";
+import { m as Ee, a as se, e as ie, l as Xu } from "./log_sum_exp-DO6z8tSE.js";
+import { s as Dn } from "./stack-DFatutCx.js";
+import { o as xe } from "./ones-tIJeHlq-.js";
+import { s as Dt } from "./slice-d0Vo9XTN.js";
+import { M as Qu, f as ar, r as tc, d as ec, a as $n } from "./dropout-C-csYCLj.js";
+import { z as Nt } from "./zeros-Bj5rMYA7.js";
+import { c as pe } from "./concat-B912vBbo.js";
+import { g as lr } from "./gather-Dnpgw-YQ.js";
+import { t as fn } from "./tensor1d-vML0r3q6.js";
+import { r as Ze } from "./relu-BjCh_SYb.js";
+import { s as ur } from "./softmax-D7Jj3p_P.js";
+import { t as nc } from "./tensor-CZr4dh61.js";
+import { r as sc } from "./range-CWcz7xFA.js";
+import { v as ic } from "./variable-Bm2OFwGI.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.