@genai-fi/nanogpt 0.10.2 → 0.10.3

package/dist/{RealDiv-zz7FpkKX.js → RealDiv-KAPDe8zB.js}

@@ -1,12 +1,10 @@
-import "./index-D6Q1lPZO.js";
-import { r as $ } from "./Reshape-CHdUjC72.js";
-import { _ as T, g as E, y as B, $ as F } from "./tensor_util-DfwaWayG.js";
-import { G as _, e as G, p as O, s as V } from "./tensor-CzmOBsdf.js";
-import { a as A, b as k, d as C, c as N, e as R } from "./axis_util-BsIr9ZNu.js";
-import { t as K, m as U } from "./shared-CHhxz-O5.js";
-import { c as W } from "./backend_util-B1XRLuq9.js";
-import { f as y } from "./gpgpu_math-DsCcikas.js";
-import { g as j, b as L } from "./kernel_funcs_utils-CWfOAPGO.js";
+import { aG as T, j as E, af as O, h as V, aS as B, Y as F, am as j, aT as G } from "./index-DOvlwCh-.js";
+import { r as $ } from "./Reshape-Zt6eb7yh.js";
+import { a as A, b as k, d as C, c as N, e as R } from "./axis_util-BaG7mf5A.js";
+import { t as K, m as U } from "./shared-DT1TkE6w.js";
+import { c as W } from "./backend_util-RCe-rHaj.js";
+import { f as y } from "./gpgpu_math-DAOmgtXR.js";
+import { g as _, b as L } from "./kernel_funcs_utils-CCzYdUZg.js";
 class w {
   constructor(s, e) {
     this.variableNames = ["x"];
@@ -16,7 +14,7 @@ class w {
     let o = "sumValue += dot(values, ones);";
     if (e != null) {
       const p = 1 / e;
-      o = `sumValue += dot(values * ${_(p) ? p.toPrecision(2) : p}, ones);`;
+      o = `sumValue += dot(values * ${T(p) ? p.toPrecision(2) : p}, ones);`;
     }
     let u = "";
     l % t > 0 && (u = `
@@ -183,7 +181,7 @@ class X {
     `;
   }
 }
-function q(a) {
+function Y(a) {
   const s = [];
   for (; s.length === 0 || s[s.length - 1].outSize !== 1; ) {
     const e = s.length ? s[s.length - 1].outSize : a[1], t = W(e);
@@ -196,7 +194,7 @@ function q(a) {
   return s;
 }
 function P(a, s, e, t) {
-  const n = q(a.shape);
+  const n = Y(a.shape);
   let l = a;
   for (let r = 0; r < n.length; r++) {
     const { inSize: i, windowSize: c, outSize: o } = n[r];
@@ -205,7 +203,7 @@ function P(a, s, e, t) {
   }
   return l;
 }
-class Y {
+class q {
   constructor(s, e) {
     this.variableNames = ["A"];
     const t = new Array(s.length);
@@ -238,7 +236,7 @@ class J {
       t[u] = s[e[u]];
     if (this.outputShape = t, this.rank = t.length, this.rank > 6)
       throw Error(`Packed transpose for rank ${this.rank} is not yet supported.`);
-    const n = y(this.rank), l = j("rc", this.rank), r = new Array(this.rank);
+    const n = y(this.rank), l = _("rc", this.rank), r = new Array(this.rank);
     for (let u = 0; u < e.length; u++)
       r[e[u]] = l[u];
     const i = `vec2(${r.slice(-2).join()})`, c = `++${l[this.rank - 1]} < ${t[this.rank - 1]}`, o = `getChannel(getA(${r.join()}), ${i})`;
@@ -263,7 +261,7 @@ class J {
   }
 }
 function D(a, s, e) {
-  const t = G().getBool("WEBGL_PACK_ARRAY_OPERATIONS") ? new J(a.shape, s) : new Y(a.shape, s);
+  const t = E().getBool("WEBGL_PACK_ARRAY_OPERATIONS") ? new J(a.shape, s) : new q(a.shape, s);
   return e.runWebGLProgram(t, [a], a.dtype);
 }
 function Q(a, s, e, t) {
@@ -275,15 +273,15 @@ function Q(a, s, e, t) {
   const [p, h] = N(u.shape, i);
   let d = p;
   e && (d = R(p, r));
-  const f = V(h), g = V(a.shape) / f, x = $({ inputs: { x: u }, attrs: { shape: [g, f] }, backend: t }), b = T(a.dtype), I = P(x, b, "sum", t), m = $({ inputs: { x: I }, attrs: { shape: d }, backend: t });
+  const f = V(h), g = V(a.shape) / f, x = $({ inputs: { x: u }, attrs: { shape: [g, f] }, backend: t }), b = B(a.dtype), I = P(x, b, "sum", t), m = $({ inputs: { x: I }, attrs: { shape: d }, backend: t });
   return t.disposeIntermediateTensorInfo(x), t.disposeIntermediateTensorInfo(I), o && t.disposeIntermediateTensorInfo(u), m;
 }
 function Z(a) {
   const { inputs: s, backend: e, attrs: t } = a, { x: n } = s, { axis: l, keepDims: r } = t;
   return Q(n, l, r, e);
 }
-const fe = {
-  kernelName: E,
+const pe = {
+  kernelName: F,
   backendName: "webgl",
   kernelFunc: Z
 };
@@ -323,8 +321,8 @@ function te(a) {
     x = ee(d, S, g, e);
   return p && e.disposeIntermediateTensorInfo(d), x;
 }
-const me = {
-  kernelName: B,
+const he = {
+  kernelName: j,
   backendName: "webgl",
   kernelFunc: te
 };
@@ -350,16 +348,16 @@ return a / b;`, se = `
   }
 
   return result;
-`, ne = L({ opSnippet: ae, packedOpSnippet: se, checkOutOfBounds: !0 }), xe = {
-  kernelName: F,
+`, ne = L({ opSnippet: ae, packedOpSnippet: se, checkOutOfBounds: !0 }), fe = {
+  kernelName: G,
   backendName: "webgl",
   kernelFunc: ne
 };
 export {
   P as a,
-  me as b,
-  xe as c,
-  fe as d,
+  he as b,
+  fe as c,
+  pe as d,
   te as m,
   ne as r,
   Z as s,

package/dist/Reshape-BYkmUnAv.js

@@ -0,0 +1,14 @@
+import { h, ah as d, l as c, R as m } from "./index-DOvlwCh-.js";
+function i(n) {
+  const { inputs: p, attrs: o } = n, { x: e } = p, { shape: r } = o, a = h(e.shape), s = d(r, a), t = h(s);
+  return c(a === t, () => `The new shape (${s}) has ${t} elements and the old shape (${e.shape}) has ${a} elements. The new shape and old shape must have the same number of elements.`), n.backend.incRef(e.dataId), { dataId: e.dataId, shape: s, dtype: e.dtype };
+}
+const u = {
+  kernelName: m,
+  backendName: "webgpu",
+  kernelFunc: i
+};
+export {
+  u as a,
+  i as r
+};

package/dist/{Reshape-CHdUjC72.js → Reshape-Zt6eb7yh.js}

@@ -1,10 +1,8 @@
-import "./index-D6Q1lPZO.js";
-import { u as C, g as f, a as R, b as g, c as I, d as c, e as u, i as m } from "./gpgpu_math-DsCcikas.js";
-import { b as x } from "./tensor_util-DfwaWayG.js";
-import { s as l, n as F, a as $ } from "./tensor-CzmOBsdf.js";
+import { R as C, h as c, ah as R, l as f } from "./index-DOvlwCh-.js";
+import { u as g, g as I, a as x, b as F, c as $, d as u, e as l, i as m } from "./gpgpu_math-DAOmgtXR.js";
 class S {
   constructor(t, i) {
-    this.variableNames = ["A"], this.packedInputs = !0, this.packedOutput = !0, this.customUniforms = [{ name: "inputShape", type: "ivec3" }], this.outputShape = t, this.enableShapeUniforms = C(this.outputShape.length);
+    this.variableNames = ["A"], this.packedInputs = !0, this.packedOutput = !0, this.customUniforms = [{ name: "inputShape", type: "ivec3" }], this.outputShape = t, this.enableShapeUniforms = g(this.outputShape.length);
     let a = "";
     for (let e = 0; e < 4; e++) {
       let o = "thisRC = rc;";
@@ -22,8 +20,8 @@ class S {
       `;
     }
     this.userCode = `
-      ${b(i, this.enableShapeUniforms)}
-      ${this.enableShapeUniforms ? f() : R(t)}
+      ${v(i, this.enableShapeUniforms)}
+      ${this.enableShapeUniforms ? I() : x(t)}
 
       void main() {
         ivec3 rc = getOutputCoords();
@@ -41,41 +39,41 @@ class S {
     `;
   }
 }
-function b(s, t) {
+function v(s, t) {
   return `
     ivec3 inputCoordsFromReshapedOutCoords(int index) {
-      ${t ? g(["r", "c", "d"], "inputShape") : I(["r", "c", "d"], s)}
+      ${t ? F(["r", "c", "d"], "inputShape") : $(["r", "c", "d"], s)}
       return ivec3(r, c, d);
     }
   `;
 }
-function v(s, t, i) {
+function b(s, t, i) {
   const a = [
-    c(s.shape),
-    ...u(s.shape)
+    u(s.shape),
+    ...l(s.shape)
   ], e = {
     dtype: s.dtype,
     shape: a,
     dataId: s.dataId
   }, o = [
-    c(t),
-    ...u(t)
+    u(t),
+    ...l(t)
   ], r = new S(o, a), p = !0, n = [a], h = i.runWebGLProgram(r, [e], s.dtype, n, p);
   return { dataId: h.dataId, shape: t, dtype: h.dtype };
 }
 function y(s) {
-  const { inputs: t, backend: i, attrs: a } = s, { x: e } = t, { shape: o } = a, r = i, p = l(e.shape), n = F(o, p), h = l(n);
-  $(p === h, () => `The new shape (${n}) has ${h} elements and the old shape (${e.shape}) has ${p} elements. The new shape and old shape must have the same number of elements.`);
+  const { inputs: t, backend: i, attrs: a } = s, { x: e } = t, { shape: o } = a, r = i, p = c(e.shape), n = R(o, p), h = c(n);
+  f(p === h, () => `The new shape (${n}) has ${h} elements and the old shape (${e.shape}) has ${p} elements. The new shape and old shape must have the same number of elements.`);
   const d = r.texData.get(e.dataId);
-  return d.isPacked && !m(e.shape, n) && !(d.texture !== null && m(d.shape, n)) ? v(e, n, r) : (r.incRef(e.dataId), { dataId: e.dataId, shape: n, dtype: e.dtype });
+  return d.isPacked && !m(e.shape, n) && !(d.texture !== null && m(d.shape, n)) ? b(e, n, r) : (r.incRef(e.dataId), { dataId: e.dataId, shape: n, dtype: e.dtype });
 }
-const O = {
-  kernelName: x,
+const U = {
+  kernelName: C,
   backendName: "webgl",
   kernelFunc: y
 };
 export {
   S as R,
-  O as a,
+  U as a,
   y as r
 };

package/dist/TeachableLLM.js

@@ -5,25 +5,24 @@ import u from "./Generator.js";
 import p from "./Trainer.js";
 import { E as f } from "./index-DvYrXKkX.js";
 import { dummyPassTrainAsync as m } from "./utilities/dummy.js";
-import "./utilities/packed.js";
-import "./index-D6Q1lPZO.js";
+import "./index-DOvlwCh-.js";
+import "./random_width-or-CEftb.js";
+import "./zeros_like-DvE73F4e.js";
+import "./index-Cp39cXWe.js";
+import "./dataset-BcwmTGYc.js";
 import "./ops/cpu/attentionMask.js";
 import "./ops/webgl/attentionMask.js";
 import "./ops/grads/attentionMask.js";
-import "./random_width-BVV9HveY.js";
-import "./register_all_kernels-nvj2k7OC.js";
-import "./index-Cp39cXWe.js";
-import "./dataset-D2P7rHAw.js";
 import "./ops/cpu/rope.js";
 import "./ops/webgl/rope.js";
-import "./rope-s4W2XO9B.js";
+import "./rope-Ir4mTyD1.js";
 import "./ops/cpu/appendCache.js";
 import "./ops/webgl/appendCache.js";
 import "./ops/grads/softmax16.js";
-import "./matMul16-fEAJ4smh.js";
+import "./matMul16-BWRSOCWB.js";
 import "./ops/webgl/matMul16.js";
 import "./ops/cpu/matMul16.js";
-import "./pack16-CmVZs6af.js";
+import "./pack16-nQ6JaLo-.js";
 import "./ops/transpose16.js";
 import "./ops/reshape16.js";
 import "./ops/cpu/qkv.js";
@@ -42,11 +41,11 @@ import g from "./tokeniser/bpe.js";
 import "./papaparse.min-C0cScC2i.js";
 import "./jszip.min-Bz5-11Bk.js";
 import "./ops/cpu/matMulGelu.js";
-import "./ops/webgl/matMulGelu.js";
+import "./matMulGelu-CzfgT6Wq.js";
 import "./ops/grads/matMulGelu.js";
 import "./ops/cpu/gelu.js";
 import "./ops/webgl/gelu.js";
-import "./gelu-Bmhopi0J.js";
+import "./gelu-CjNPL4OH.js";
 import "./ops/webgl/log.js";
 import "./ops/cpu/adamMoments.js";
 import "./ops/webgl/adamMoments.js";

package/dist/{axis_util-BsIr9ZNu.js → axis_util-BaG7mf5A.js}

@@ -1,4 +1,4 @@
-import { a as c } from "./tensor-CzmOBsdf.js";
+import { l as c } from "./index-DOvlwCh-.js";
 function i(e, n) {
   for (let t = 0; t < e.length; ++t)
     if (e[e.length - t - 1] !== n - 1 - t)
@@ -12,7 +12,7 @@ function a(e, n, t) {
     t.indexOf(u) === -1 ? s.push(e[o++]) : s.push(n[f++]);
   return s;
 }
-function l(e, n) {
+function p(e, n) {
   const t = [], r = e.length;
   for (let o = 0; o < r; o++)
     n.indexOf(o) === -1 && t.push(e[o]);
@@ -46,7 +46,7 @@ function x(e, n) {
 export {
   d as a,
   x as b,
-  l as c,
+  p as c,
   m as d,
   h as e,
   i as f,

package/dist/backend.js

@@ -1,9 +1,9 @@
-import { g as o, s as e, r as s } from "./index-D6Q1lPZO.js";
+import { g as o, s as e, r as s } from "./index-DOvlwCh-.js";
 async function c(t, a) {
   if (o() !== t) {
     if (t === "webgpu") {
       const { registerWebGPUBackend: i } = await import("./patches/webgpu_base.js");
-      i(a), await import("./index-DRyE072i.js"), await import("./ops/webgpu/index.js");
+      i(a), await import("./index-BwexR4lA.js"), await import("./ops/webgpu/index.js");
     }
     await e(t), await s(), console.log(`Backend set to ${t}`);
   }

package/dist/{backend_util-B1XRLuq9.js → backend_util-RCe-rHaj.js}

@@ -1,11 +1,10 @@
-import { s as m, i as w, K as I, a as d, z as A, q as _ } from "./tensor-CzmOBsdf.js";
-import { d as y, f as M, h as T, c as b, e as D, a as L, b as x, g as W } from "./axis_util-BsIr9ZNu.js";
-import { j as v, h as F, q as N } from "./index-D6Q1lPZO.js";
-import { a as C, c as P } from "./concat_util-iBYIyuQe.js";
-import { S as z, a as q, b as B, c as U, d as j, e as G, f as H, g as V, h as Z, i as k, j as K, k as J, l as X, m as Y, s as Q, n as ee, t as te } from "./selu_util-BGPXmd4B.js";
-import { s as se } from "./slice_util-DyjSAD0u.js";
-import { a0 as ne, u as re, w as oe } from "./tensor_util-DfwaWayG.js";
-import { c as ae, v as ie, a as ue } from "./scatter_nd_util-C7zXRT_h.js";
+import { h as m, ad as w, aU as I, l as d, ax as A, aB as _, a3 as y, o as M, a4 as T, aV as b, ak as D, aW as x } from "./index-DOvlwCh-.js";
+import { d as L, f as W, h as v, c as F, e as N, a as C, b as P, g as z } from "./axis_util-BaG7mf5A.js";
+import { a as B, c as U } from "./concat_util-DpW8mL_l.js";
+import { c as V, b as G, d as H, f as j, g as q, h as Z, i as k, j as J, k as K, m as X, t as Y } from "./step-CINUs5QB.js";
+import { S as Q, a as ee, b as te, g as se, c as ne, s as re } from "./selu_util-kbhpTdYD.js";
+import { s as oe } from "./slice_util-uTKwiEpW.js";
+import { c as ae, v as ie, a as ue } from "./scatter_nd_util-lvSiX8q4.js";
 import { a as le, c as pe, b as ce, e as he, d as fe, g as ge, m as de, s as me } from "./complex_util-Yc1A_gV1.js";
 function Ee(e, t) {
   const r = e.shape.length, s = t.shape.length;
@@ -147,7 +146,7 @@ function Te(e, t, r) {
     s.push(e[n + 1] - t[n][0] - t[n][1]);
   return s;
 }
-const be = 0.3275911, De = 0.254829592, Le = -0.284496736, xe = 1.421413741, We = -1.453152027, ve = 1.061405429;
+const be = 0.3275911, De = 0.254829592, xe = -0.284496736, Le = 1.421413741, We = -1.453152027, ve = 1.061405429;
 const E = "->", Fe = /->/g, S = ",", R = "...";
 function Ne(e, t) {
   e = e.replace(/\s/g, "");
@@ -213,22 +212,22 @@ function ze(e, t) {
     s.push([]);
   const o = [];
   for (let a = 0; a < r.length; ++a) {
-    const u = r[a], p = Be(t, u);
+    const u = r[a], p = Ue(t, u);
     for (const c of p)
       o.indexOf(c) === -1 && (s[a].push(c), o.push(c));
   }
   return { path: r, steps: s };
 }
-function qe(e) {
+function Be(e) {
   return e.every((t, r) => t === r);
 }
-function Be(e, t) {
+function Ue(e, t) {
   const r = [];
   for (let s = 0; s < e.length; ++s)
     (e[s].length === 0 || e[s].indexOf(t) !== -1 || t === -1) && r.push(s);
   return r;
 }
-function Ue(e, t, r = 0) {
+function Ve(e, t, r = 0) {
   let s = [];
   if (typeof t == "number")
     d(e.shape[r] % t === 0, () => "Number of splits must evenly divide the axis."), s = new Array(t).fill(e.shape[r] / t);
@@ -244,17 +243,17 @@ function Ue(e, t, r = 0) {
   }
   return s;
 }
-function je(e) {
+function Ge(e) {
   return `Received SparseTensor with denseShape[0] = 0 but
   indices.shape[0] = ${e}`;
 }
-function Ge(e, t) {
+function He(e, t) {
   return `indices(${e}, 0) is invalid: ${t} < 0`;
 }
-function He(e, t, r) {
+function je(e, t, r) {
   return `indices(${e}, 0) is invalid: ${t} >= ${r}`;
 }
-function Ve(e, t) {
+function qe(e, t) {
   return `only one output dimension may be -1, not both ${e} and ${t}`;
 }
 function Ze(e, t) {
@@ -263,12 +262,12 @@ function Ze(e, t) {
 function ke() {
   return "reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero";
 }
-function Ke(e, t) {
+function Je(e, t) {
   const r = m(e), s = m(t);
   return `Input to reshape is a SparseTensor with ${r}
   dense values, but the requested shape requires a multiple of ${s}. inputShape=${e} outputShape= ${t}`;
 }
-function Je(e, t) {
+function Ke(e, t) {
   const r = m(e), s = m(t);
   return `Input to reshape is a tensor with ${r} dense values, but the requested shape has ${s}. inputShape=${e} outputShape=${t}`;
 }
@@ -336,11 +335,11 @@ function ot(e) {
 function at(e) {
   return e.map((t) => _(t));
 }
-const mt = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
+const dt = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
   __proto__: null,
   ERF_A1: De,
-  ERF_A2: Le,
-  ERF_A3: xe,
+  ERF_A2: xe,
+  ERF_A3: Le,
   ERF_A4: We,
   ERF_A5: ve,
   ERF_P: be,
@@ -348,84 +347,84 @@ const mt = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
   get RowPartitionType() {
     return f;
   },
-  SELU_SCALE: z,
-  SELU_SCALEALPHA: q,
-  applyActivation: B,
-  assertAndGetBroadcastShape: v,
-  assertAxesAreInnerMostDims: y,
-  assertParamsConsistent: C,
+  SELU_SCALE: Q,
+  SELU_SCALEALPHA: ee,
+  applyActivation: te,
+  assertAndGetBroadcastShape: y,
+  assertAxesAreInnerMostDims: L,
+  assertParamsConsistent: B,
   assignToTypedArray: le,
-  axesAreInnerMostDims: M,
+  axesAreInnerMostDims: W,
   calculateShapes: ae,
   checkEinsumDimSizes: Pe,
-  checkPadOnDimRoundingMode: U,
-  combineLocations: T,
+  checkPadOnDimRoundingMode: V,
+  combineLocations: v,
   combineRaggedTensorToTensorShapes: Ie,
   complexWithEvenIndex: pe,
   complexWithOddIndex: ce,
-  computeConv2DInfo: j,
-  computeConv3DInfo: G,
-  computeDefaultPad: H,
-  computeDilation2DInfo: V,
+  computeConv2DInfo: G,
+  computeConv3DInfo: H,
+  computeDefaultPad: j,
+  computeDilation2DInfo: q,
   computeOptimalWindowSize: Oe,
-  computeOutAndReduceShapes: b,
-  computeOutShape: P,
+  computeOutAndReduceShapes: F,
+  computeOutShape: U,
   computePool2DInfo: Z,
   computePool3DInfo: k,
-  convertConv2DDataFormat: K,
+  convertConv2DDataFormat: J,
   decodeEinsumEquation: Ne,
-  eitherStridesOrDilationsAreOne: J,
-  expandShapeToKeepDim: D,
+  eitherStridesOrDilationsAreOne: K,
+  expandShapeToKeepDim: N,
   exponent: he,
   exponents: fe,
   fromStringArrayToUint8: at,
   fromUint8ToStringArray: ot,
-  getAxesPermutation: L,
-  getBroadcastDims: F,
+  getAxesPermutation: C,
+  getBroadcastDims: M,
   getComplexWithIndex: ge,
   getEinsumComputePath: ze,
   getEinsumPermutation: Ce,
-  getFusedBiasGradient: X,
-  getFusedDyActivation: Y,
+  getFusedBiasGradient: se,
+  getFusedDyActivation: ne,
   getImageCenter: we,
-  getInnerMostAxes: x,
+  getInnerMostAxes: P,
   getPermuted: _e,
   getRaggedRank: Se,
-  getReductionAxes: N,
+  getReductionAxes: T,
   getReshaped: Ae,
   getReshapedPermuted: ye,
   getRowPartitionTypesHelper: $e,
   getSliceBeginCoords: Me,
   getSliceSize: Te,
-  getSparseFillEmptyRowsIndicesDenseShapeMismatch: je,
-  getSparseFillEmptyRowsNegativeIndexErrorMessage: Ge,
-  getSparseFillEmptyRowsOutOfRangeIndexErrorMessage: He,
+  getSparseFillEmptyRowsIndicesDenseShapeMismatch: Ge,
+  getSparseFillEmptyRowsNegativeIndexErrorMessage: He,
+  getSparseFillEmptyRowsOutOfRangeIndexErrorMessage: je,
   getSparseReshapeEmptyTensorZeroOutputDimErrorMessage: ke,
-  getSparseReshapeInputOutputMismatchErrorMessage: Je,
-  getSparseReshapeInputOutputMultipleErrorMessage: Ke,
-  getSparseReshapeMultipleNegativeOneOutputDimErrorMessage: Ve,
+  getSparseReshapeInputOutputMismatchErrorMessage: Ke,
+  getSparseReshapeInputOutputMultipleErrorMessage: Je,
+  getSparseReshapeMultipleNegativeOneOutputDimErrorMessage: qe,
   getSparseReshapeNegativeOutputDimErrorMessage: Ze,
   getSparseSegmentReductionIndicesOutOfRangeErrorMessage: et,
   getSparseSegmentReductionNegativeSegmentIdsErrorMessage: Xe,
   getSparseSegmentReductionNonIncreasingSegmentIdsErrorMessage: Ye,
   getSparseSegmentReductionSegmentIdOutOfRangeErrorMessage: Qe,
-  getUndoAxesPermutation: W,
-  isIdentityPermutation: qe,
-  log: ne,
+  getUndoAxesPermutation: z,
+  isIdentityPermutation: Be,
+  log: b,
   mergeRealAndImagArrays: de,
   prepareAndValidate: Ee,
-  prepareSplitSize: Ue,
+  prepareSplitSize: Ve,
   segment_util: rt,
-  shouldFuse: Q,
-  slice_util: se,
+  shouldFuse: re,
+  slice_util: oe,
   splitRealAndImagArrays: me,
-  stridesOrDilationsArePositive: ee,
-  tupleValuesAreOne: te,
-  upcastType: re,
+  stridesOrDilationsArePositive: X,
+  tupleValuesAreOne: Y,
+  upcastType: D,
   validateDefaultValueShape: Re,
   validateInput: ie,
   validateUpdateShape: ue,
-  warn: oe
+  warn: x
 }, Symbol.toStringTag, { value: "Module" }));
 export {
   _e as A,
@@ -437,37 +436,37 @@ export {
   Pe as G,
   ze as H,
   Ce as I,
-  qe as J,
+  Be as J,
   Ee as K,
   nt as L,
   we as M,
-  Ue as N,
+  Ve as N,
   st as O,
   tt as P,
   f as R,
   Se as a,
-  mt as b,
+  dt as b,
   Oe as c,
   Ie as d,
   at as e,
   ot as f,
   $e as g,
-  je as h,
-  Ge as i,
-  He as j,
-  Ve as k,
+  Ge as h,
+  He as i,
+  je as j,
+  qe as k,
   Ze as l,
   ke as m,
-  Ke as n,
-  Je as o,
+  Je as n,
+  Ke as o,
   Xe as p,
   Ye as q,
   Qe as r,
   et as s,
   De as t,
-  Le as u,
+  xe as u,
   Re as v,
-  xe as w,
+  Le as w,
   We as x,
   ve as y,
   Ae as z