@genai-fi/nanogpt 0.7.1 → 0.7.3

package/dist/ops/webgpu/qkv.js

@@ -1,7 +1,7 @@
-import { g as c } from "../../webgpu_program-WaoMq-WD.js";
-import { f as d, c as h } from "../../webgpu_util-DhSeP4b6.js";
-import { f as p } from "../../index-UdZhlibC.js";
-class m {
+import { g as h } from "../../webgpu_program-PFzf1hAQ.js";
+import { f as c, c as d } from "../../webgpu_util-D____QpY.js";
+import { f as p, a2 as m } from "../../index-BoWRt-10.js";
+class l {
   variableNames = ["x", "kernel"];
   outputShape;
   shaderKey = "QKV";
@@ -11,13 +11,13 @@ class m {
   workgroupSize = [64, 1, 1];
   size = !0;
   constructor(t, e, o, s) {
-    const r = s / e;
-    this.shaderKey = `QKV_${e}_${r}`, this.outputShape = [t, e, o, r], this.dispatchLayout = d(this.outputShape), this.dispatch = h(this.dispatchLayout, this.outputShape, this.workgroupSize);
+    const n = s / e;
+    this.shaderKey = `QKV_${e}_${n}`, this.outputShape = [t, e, o, n], this.dispatchLayout = c(this.outputShape), this.dispatch = d(this.dispatchLayout, this.outputShape, this.workgroupSize);
   }
   getUserCode() {
     const t = this.outputShape[1], e = this.outputShape[3], o = t * e;
     return `
-        ${c("index")} {
+        ${h("index")} {
             if (index < uniforms.size) {
                 let coords = getCoordsFromIndex(index); // [b, h, t, d]
                 let b = coords[0];
@@ -26,13 +26,14 @@ class m {
                 let d = coords[3];
 
                 // Compute output channel index in fused kernel
-                let out_offset = uniforms.mode * ${t} * ${e} + h * ${e} + d;
+                let out_offset = uniforms.mode * ${o} + h * ${e} + d;
 
                 var sum = 0.0;
+                let baseX = b * uniforms.xShape[1] * uniforms.xShape[2] + t * uniforms.xShape[2];
                 for (var c = 0; c < ${o}; c += 1) {
-                    let xval = getX(b, t, c); // fetch from x
-                    let kval = getKernel(c, out_offset); // fetch from kernel
-                    sum += xval * kval;
+                    let xval = x[baseX + c];
+                    let kval = getKernel(c, out_offset);
+                    sum = fma(xval, kval, sum);
                 }
 
                 setOutputAtIndex(index, sum);
@@ -41,17 +42,20 @@ class m {
         `;
   }
 }
-function l(a) {
-  const { x: t, kernel: e } = a.inputs, { heads: o } = a.attrs, s = a.backend, r = t.shape[0], i = t.shape[1], u = t.shape[2], n = new m(r, o, i, u);
+function f(r) {
+  const { x: t, kernel: e } = r.inputs, { heads: o } = r.attrs, s = r.backend, n = t.shape[0], u = t.shape[1], a = t.shape[2];
+  if (m(e.shape, [a, 3 * a], "Error in QKV: "), a % o !== 0)
+    throw new Error(`Channel dimension ${a} must be divisible by number of heads ${o} in QKV.`);
+  const i = new l(n, o, u, a);
   return [
-    s.runWebGPUProgram(n, [t, e], "float32", [{ type: "int32", data: [0] }]),
-    s.runWebGPUProgram(n, [t, e], "float32", [{ type: "int32", data: [1] }]),
-    s.runWebGPUProgram(n, [t, e], "float32", [{ type: "int32", data: [2] }])
+    s.runWebGPUProgram(i, [t, e], "float32", [{ type: "int32", data: [0] }]),
+    s.runWebGPUProgram(i, [t, e], "float32", [{ type: "int32", data: [1] }]),
+    s.runWebGPUProgram(i, [t, e], "float32", [{ type: "int32", data: [2] }])
   ];
 }
-const f = {
+const b = {
   kernelName: "QKV",
   backendName: "webgpu",
-  kernelFunc: l
+  kernelFunc: f
 };
-p(f);
+p(b);

package/dist/ops/webgpu/rope.js

@@ -1,7 +1,7 @@
-import { g as h } from "../../webgpu_program-WaoMq-WD.js";
-import { f as l, c as m } from "../../webgpu_util-DhSeP4b6.js";
-import { f as g } from "../../index-UdZhlibC.js";
-class f {
+import { g as c } from "../../webgpu_program-PFzf1hAQ.js";
+import { f as m, c as l } from "../../webgpu_util-D____QpY.js";
+import { f as x, a2 as f } from "../../index-BoWRt-10.js";
+class S {
   variableNames = ["x", "sin", "cos"];
   outputShape;
   shaderKey = "Rope";
@@ -10,13 +10,13 @@ class f {
   workgroupSize = [64, 1, 1];
   size = !0;
   uniforms = "pastLen: i32";
-  constructor(t, s, n, o) {
-    this.shaderKey = `Rope_${o}`, this.outputShape = [t, s, n, o], this.dispatchLayout = l(this.outputShape), this.dispatch = m(this.dispatchLayout, this.outputShape, this.workgroupSize);
+  constructor(t, e, a, s) {
+    this.shaderKey = `Rope_${s}`, this.outputShape = [t, e, a, s], this.dispatchLayout = m(this.outputShape), this.dispatch = l(this.dispatchLayout, this.outputShape, this.workgroupSize);
   }
   getUserCode() {
     const t = this.outputShape[3];
     return `
-        ${h("index")} {
+        ${c("index")} {
             if (index < uniforms.size) {
                 let coords = getCoordsFromIndex(index); // [b, h, t, d]
                 let b = coords[0];
@@ -28,27 +28,31 @@ class f {
 
                 var outVal = 0.0;
 
+                let xIdx = b * uniforms.outShapeStrides[0] +
+                    h * uniforms.outShapeStrides[1] +
+                    t * uniforms.outShapeStrides[2] +
+                    d;
+
                 if (d < rotaryDim) {
-                    let pairIdx = d / 2;
-                    let cos = getCos(t + uniforms.pastLen, pairIdx, 0);
-                    let sin = getSin(t + uniforms.pastLen, pairIdx, 0);
+                    let idx = (t + uniforms.pastLen) * uniforms.cosShape[1] + d / 2;
+                    let cos = cos[idx];
+                    let sin = sin[idx];
 
-                    let ownX = getX(b, h, t, d) * cos;
+                    let ownX = x[xIdx] * cos;
+                    var evenOdd = 0.0;
 
                     if (d % 2 == 0) {
                         // even index
-                        let even = ownX;
-                        let odd = getX(b, h, t, d + 1);
-                        outVal = even - odd * sin;
+                        evenOdd = -x[xIdx + 1];
                     } else {
                         // odd index
-                        let even = getX(b, h, t, d - 1);
-                        let odd = ownX;
-                        outVal = even * sin + odd;
+                        evenOdd = x[xIdx - 1];
                     }
+
+                    outVal = fma(evenOdd, sin, ownX);
                 } else {
                     // pass through for non-rotary dims
-                    outVal = getX(b, h, t, d);
+                    outVal = x[xIdx];
                 }
 
                 setOutputAtIndex(index, outVal);
@@ -57,13 +61,22 @@ class f {
         `;
   }
 }
-function b(e) {
-  const { x: t, sin: s, cos: n } = e.inputs, { pastLen: o } = e.attrs, a = e.backend, r = t.shape[0], i = t.shape[1], d = t.shape[2], p = t.shape[3], u = new f(r, i, d, p), c = [{ type: "int32", data: [o] }];
-  return a.runWebGPUProgram(u, [t, s, n], "float32", c);
+function g(o) {
+  const { x: t, sin: e, cos: a } = o.inputs, { pastLen: s } = o.attrs, i = o.backend, p = t.shape[0], d = t.shape[1], n = t.shape[2], r = t.shape[3];
+  if (f(e.shape, a.shape, "Error in Rope: "), e.shape[0] < n + s)
+    throw new Error(
+      `Sin tensor shape ${e.shape} is not compatible with seqLength ${n} and pastLen ${s}.`
+    );
+  if (e.shape[1] * 2 < r)
+    throw new Error(`Sin tensor shape ${e.shape} is not compatible with feature dimension ${r}.`);
+  if (e.shape.length !== 3)
+    throw new Error(`Sin tensor must be 3-dimensional, but got shape ${e.shape}.`);
+  const h = new S(p, d, n, r), u = [{ type: "int32", data: [s] }];
+  return i.runWebGPUProgram(h, [t, e, a], "float32", u);
 }
-const x = {
+const b = {
   kernelName: "Rope",
   backendName: "webgpu",
-  kernelFunc: b
+  kernelFunc: g
 };
-g(x);
+x(b);

package/dist/ops/webgpu/scatterSub.js

@@ -1,7 +1,7 @@
-import { g as c } from "../../webgpu_program-WaoMq-WD.js";
-import { f as p, c as u } from "../../webgpu_util-DhSeP4b6.js";
-import { f as d } from "../../index-UdZhlibC.js";
-class h {
+import { g as p } from "../../webgpu_program-PFzf1hAQ.js";
+import { f as u, c as d } from "../../webgpu_util-D____QpY.js";
+import { f as h, a2 as o } from "../../index-BoWRt-10.js";
+class b {
   variableNames = ["labels", "softmaxProbs", "dy"];
   outputShape;
   shaderKey = "ScatterSub";
@@ -10,29 +10,31 @@ class h {
   workgroupSize = [64, 1, 1];
   size = !0;
   constructor(t, e) {
-    this.outputShape = [t, e], this.dispatchLayout = p(this.outputShape), this.dispatch = u(this.dispatchLayout, this.outputShape, this.workgroupSize);
+    this.outputShape = [t, e], this.dispatchLayout = u(this.outputShape), this.dispatch = d(this.dispatchLayout, this.outputShape, this.workgroupSize);
   }
   getUserCode() {
     return `
-            ${c("index")} {
+            ${p("index")} {
                 if (index < uniforms.size) {
                     let coords = getCoordsFromIndex(index); // [batch, depth]
-                    let idx = i32(getLabels(coords[0]));
-                    let prob = getSoftmaxProbsByOutputIndex(index);
-                    let dy = getDy(coords[0]);
+                    let idx = i32(labels[coords[0]]);
+                    let prob = softmaxProbs[index];
+                    let dy = dy[coords[0]];
                     setOutputAtIndex(index, select(prob, prob - 1.0, idx == coords[1]) * dy);
                 }
             }
             `;
   }
 }
-function b(o) {
-  const { logits: t, labels: e, dy: s } = o.inputs, r = o.backend, a = e.shape[0], i = t.shape[1], n = new h(a, i);
-  return r.runWebGPUProgram(n, [e, t, s], "float32");
+function f(a) {
+  const { logits: t, labels: e, dy: s } = a.inputs, c = a.backend, r = e.shape[0], i = t.shape[1];
+  o(s.shape, [r], "Error in EfficientScatterSub dy: "), o(t.shape, [r, i], "Error in EfficientScatterSub logits: "), o(e.shape, [r], "Error in EfficientScatterSub labels: ");
+  const n = new b(r, i);
+  return c.runWebGPUProgram(n, [e, t, s], "float32");
 }
 const l = {
   kernelName: "EfficientScatterSub",
   backendName: "webgpu",
-  kernelFunc: b
+  kernelFunc: f
 };
-d(l);
+h(l);

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

@@ -1,7 +1,7 @@
-import { p as l, j as d } from "../../../index-UdZhlibC.js";
-import { g as p } from "../../../webgpu_program-WaoMq-WD.js";
-import { r as f } from "../../../Reshape-Do18N3gO.js";
-import { c as x } from "../../../axis_util-5DTW2tFV.js";
+import { p as l, j as d } from "../../../index-BoWRt-10.js";
+import { g as p } from "../../../webgpu_program-PFzf1hAQ.js";
+import { r as f } from "../../../Reshape-DvudQDvJ.js";
+import { c as x } from "../../../axis_util-BzbKo31C.js";
 function I(e, r, t, s, u) {
   return `
            fn DIV_CEIL(a : u32, b : u32) -> u32 {

package/dist/{ops-aRTXR2Sr.js → ops-BFGCx8Ri.js}

@@ -1,19 +1,19 @@
-import { B as m, C as u, l, E as f, aP as gn, k as S, aQ as $n, aR as kn, aS as yn, p as En, aT as w, q as rn, x as _n, b as pn, aU as In, aV as Nn, N as An, al as Tn, I as v, aW as xn, aX as wn, aY as Sn, aZ as Mn, a_ as zn, a$ as qn, b0 as Bn, X as K, b1 as Gn, b2 as Kn, b3 as Rn, aG as Dn, $ as vn, Z as Vn, _ as Pn, b4 as jn, t as Fn, b5 as L, b6 as Wn, b7 as Yn, b8 as On, b9 as Cn, ba as Ln, bb as Xn, bc as Zn, bd as Jn, a as $, o as U, n as q, h as Qn, c as A, be as Hn, M as X, d as Un } from "./index-UdZhlibC.js";
+import { B as m, C as u, l, E as f, aP as gn, k as S, aQ as $n, aR as kn, aS as yn, p as En, aT as w, q as rn, x as _n, b as pn, aU as In, aV as Nn, N as An, al as Tn, I as v, aW as xn, aX as wn, aY as Sn, aZ as Mn, a_ as zn, a$ as qn, b0 as Bn, X as K, b1 as Gn, b2 as Kn, b3 as Rn, aG as Dn, $ as vn, Z as Vn, _ as Pn, b4 as jn, t as Fn, b5 as L, b6 as Wn, b7 as Yn, b8 as On, b9 as Cn, ba as Ln, bb as Xn, bc as Zn, bd as Jn, a as $, o as U, n as q, h as Qn, c as A, be as Hn, M as X, d as Un } from "./index-BoWRt-10.js";
 import { n as ne, a as ee, b as se } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { r as k } from "./reshape-BR0eoLYN.js";
-import { s as mn } from "./split-CRU0PjVV.js";
-import { s as E } from "./sum-NLYbiDag.js";
-import { b as Z } from "./broadcast_to-Ba9h_8DO.js";
-import { s as _ } from "./slice-BHbDHObE.js";
-import { r as W } from "./range-D9CZhVlR.js";
-import { t as te } from "./tensor-Do9PKbIE.js";
-import { s as Y } from "./stack-ikk2Y8_P.js";
-import { c as re, z as ae } from "./zeros-DnPT2nD4.js";
-import { e as oe } from "./axis_util-5DTW2tFV.js";
-import { m as J } from "./max-ByjEGoFx.js";
-import { c as Q } from "./concat-CbXTetof.js";
-import { m as N } from "./mat_mul-dwmZz69e.js";
-import { t as H } from "./tensor2d-CWHxHpLh.js";
+import { r as k } from "./reshape-CdBq1WJ6.js";
+import { s as mn } from "./split-B-dikLRw.js";
+import { s as E } from "./sum-66ew2byf.js";
+import { b as Z } from "./broadcast_to-CdbwV-Dj.js";
+import { s as _ } from "./slice-BiOsknYS.js";
+import { r as W } from "./range-CRuAh-gd.js";
+import { t as te } from "./tensor-JwS7ZYY6.js";
+import { s as Y } from "./stack-B17UN2nn.js";
+import { c as re, z as ae } from "./zeros--BdLQ3oG.js";
+import { e as oe } from "./axis_util-BzbKo31C.js";
+import { m as J } from "./max-Ddnnb5xe.js";
+import { c as Q } from "./concat-CsxrgovM.js";
+import { m as N } from "./mat_mul-8m8pfdcx.js";
+import { t as H } from "./tensor2d-wxPAnDQy.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{random_width-DbSpgl4o.js → random_width-sZORGo5k.js}

@@ -1,25 +1,25 @@
-import { B as T, C as I, E as O, bE as Oa, bF as Ma, bG as Ci, l as b, bH as Ii, X as L, bI as Di, bJ as $i, bK as Ti, bL as zi, h as Ei, bM as Li, bN as Fi, bO as Oi, bP as Mi, bQ as _a, bR as _i, bS as Ra, bT as Ri, bU as Ba, bV as Bi, I as Ge, k as kt, bq as Wa, bW as Wi, bX as Gi, bY as Pi, y as Pe, c as V, a as w, bZ as Ga, b_ as Ui, b$ as Vi, c0 as ji, p as ce, aU as pt, bv as Pa, c1 as Ki, c2 as Hi, c3 as qi, c4 as Ji, c5 as Zi, bx as Xi, c6 as Yi, c7 as Qi, F as Ua, al as Va, aq as ja, c8 as tr, c9 as Ka, o as z, ca as Ps, cb as Ha, cc as qa, j as pn, cd as Us, ce as Ja, cf as Za, cg as Xa, ch as Ya, ci as Qa, cj as tl, ck as el, bf as nl, cl as sl, q as he, b as et, n as U, cm as il, bm as rl, aw as ht, cn as ol, z as Q, co as al, cp as ll, cq as ul, cr as cl, cs as hl, ct as pl, cu as dl, cv as fl, J as ml, cw as gl, bj as bl, bp as yl, cx as wl, Q as kl, cy as xl, a8 as Nl, cz as vl, cA as Al, cB as Sl, as as Cl, cC as Il, ae as Dl, aV as $l, br as Tl, ak as zl, bs as El, G as Ll, aX as Fl, am as Ol, cD as Ml, cE as _l, cF as Rl, an as Bl, af as Wl, cG as Gl, cH as Pl, cI as Ul, U as Vl, bt as jl, cJ as Kl, cK as Hl, aS as ql, b1 as Jl, cL as Zl, cM as Xl, bu as Yl, a$ as Ql, P as tu, cN as eu, x as rs, ap as nu, bw as su, az as iu, V as ru, au as ou, at as au, K as lu, bc as uu, cO as cu, bd as hu, cP as pu, b3 as du, aQ as fu, ar as mu, cQ as gu, a9 as bu, W as yu, S as wu, L as ku, bz as xu, cR as Nu, bA as vu, av as Au, bC as Su, Y as Cu, cS as Iu, N as Du, b5 as $u, b4 as Tu, cT as Oe, cU as zu, i as Eu, aj as Vs, cV as Lu, t as x, aT as $e, cW as S, cX as He, cY as qe, ab as Vt, d as J, ac as Fu, cZ as js, w as Jt, M as Ou, T as Te, O as Mu, c_ as _u, m as Ks, c$ as Ru, d0 as Hs, d1 as Bu } from "./index-UdZhlibC.js";
-import { k as ke, c as Nt, o as ze, s as er, b as nr, d as Wu, m as sr, t as In, l as ir, v as os, a as Gu, S as Pu, p as Uu, w as as, x as rr, y as Je, z as Vu, A as ju } from "./selu_util-Ce6pu9IM.js";
-import { n as wt, w as ne, a as Ze, g as Xe, b as ls, t as K, c as Ce, d as Xt, e as Ku, u as dn, f as ye, h as fn, i as Hu, l as qu, s as us, m as or, j as qt, k as Ju } from "./ops-aRTXR2Sr.js";
-import { r as N } from "./reshape-BR0eoLYN.js";
-import { s as W } from "./sum-NLYbiDag.js";
-import { m as ct } from "./mat_mul-dwmZz69e.js";
-import { s as Qt } from "./split-CRU0PjVV.js";
-import { s as Zu, c as ar } from "./sin-CiBxrDqX.js";
-import { e as Hn, g as lr, h as cs, c as Xu } from "./axis_util-5DTW2tFV.js";
-import { a as se, e as ie, l as Yu } from "./log_sum_exp-BnmCkHWl.js";
-import { s as Dn } from "./stack-ikk2Y8_P.js";
-import { o as xe } from "./ones-C8Mfln6-.js";
-import { s as Dt } from "./slice-BHbDHObE.js";
-import { M as Qu, f as ur, r as tc, d as ec, a as $n } from "./dropout-DPfPgWWe.js";
-import { z as vt } from "./zeros-DnPT2nD4.js";
-import { c as pe } from "./concat-CbXTetof.js";
-import { g as cr } from "./gather-Bbh8DHhM.js";
-import { s as hr } from "./softmax-JMEIUo2J.js";
-import { m as Ee } from "./max-ByjEGoFx.js";
-import { t as nc } from "./tensor-Do9PKbIE.js";
-import { r as sc } from "./range-D9CZhVlR.js";
-import { v as ic } from "./variable-BTBkayv_.js";
+import { B as T, C as I, E as O, bE as Oa, bF as Ma, bG as Ci, l as b, bH as Ii, X as L, bI as Di, bJ as $i, bK as Ti, bL as zi, h as Ei, bM as Li, bN as Fi, bO as Oi, bP as Mi, bQ as _a, bR as _i, bS as Ra, bT as Ri, bU as Ba, bV as Bi, I as Ge, k as kt, bq as Wa, bW as Wi, bX as Gi, bY as Pi, y as Pe, c as V, a as w, bZ as Ga, b_ as Ui, b$ as Vi, c0 as ji, p as ce, aU as pt, bv as Pa, c1 as Ki, c2 as Hi, c3 as qi, c4 as Ji, c5 as Zi, bx as Xi, c6 as Yi, c7 as Qi, F as Ua, al as Va, aq as ja, c8 as tr, c9 as Ka, o as z, ca as Ps, cb as Ha, cc as qa, j as pn, cd as Us, ce as Ja, cf as Za, cg as Xa, ch as Ya, ci as Qa, cj as tl, ck as el, bf as nl, cl as sl, q as he, b as et, n as U, cm as il, bm as rl, aw as ht, cn as ol, z as Q, co as al, cp as ll, cq as ul, cr as cl, cs as hl, ct as pl, cu as dl, cv as fl, J as ml, cw as gl, bj as bl, bp as yl, cx as wl, Q as kl, cy as xl, a8 as Nl, cz as vl, cA as Al, cB as Sl, as as Cl, cC as Il, ae as Dl, aV as $l, br as Tl, ak as zl, bs as El, G as Ll, aX as Fl, am as Ol, cD as Ml, cE as _l, cF as Rl, an as Bl, af as Wl, cG as Gl, cH as Pl, cI as Ul, U as Vl, bt as jl, cJ as Kl, cK as Hl, aS as ql, b1 as Jl, cL as Zl, cM as Xl, bu as Yl, a$ as Ql, P as tu, cN as eu, x as rs, ap as nu, bw as su, az as iu, V as ru, au as ou, at as au, K as lu, bc as uu, cO as cu, bd as hu, cP as pu, b3 as du, aQ as fu, ar as mu, cQ as gu, a9 as bu, W as yu, S as wu, L as ku, bz as xu, cR as Nu, bA as vu, av as Au, bC as Su, Y as Cu, cS as Iu, N as Du, b5 as $u, b4 as Tu, cT as Oe, cU as zu, i as Eu, aj as Vs, cV as Lu, t as x, aT as $e, cW as S, cX as He, cY as qe, ab as Vt, d as J, ac as Fu, cZ as js, w as Jt, M as Ou, T as Te, O as Mu, c_ as _u, m as Ks, c$ as Ru, d0 as Hs, d1 as Bu } from "./index-BoWRt-10.js";
+import { k as ke, c as Nt, o as ze, s as er, b as nr, d as Wu, m as sr, t as In, l as ir, v as os, a as Gu, S as Pu, p as Uu, w as as, x as rr, y as Je, z as Vu, A as ju } from "./selu_util-BJEXVvjX.js";
+import { n as wt, w as ne, a as Ze, g as Xe, b as ls, t as K, c as Ce, d as Xt, e as Ku, u as dn, f as ye, h as fn, i as Hu, l as qu, s as us, m as or, j as qt, k as Ju } from "./ops-BFGCx8Ri.js";
+import { r as N } from "./reshape-CdBq1WJ6.js";
+import { s as W } from "./sum-66ew2byf.js";
+import { m as ct } from "./mat_mul-8m8pfdcx.js";
+import { s as Qt } from "./split-B-dikLRw.js";
+import { s as Zu, c as ar } from "./sin-BeA3tsEd.js";
+import { e as Hn, g as lr, h as cs, c as Xu } from "./axis_util-BzbKo31C.js";
+import { a as se, e as ie, l as Yu } from "./log_sum_exp-DbjkV734.js";
+import { s as Dn } from "./stack-B17UN2nn.js";
+import { o as xe } from "./ones-Dj0SDhHf.js";
+import { s as Dt } from "./slice-BiOsknYS.js";
+import { M as Qu, f as ur, r as tc, d as ec, a as $n } from "./dropout-DYs5QFGQ.js";
+import { z as vt } from "./zeros--BdLQ3oG.js";
+import { c as pe } from "./concat-CsxrgovM.js";
+import { g as cr } from "./gather-CMMy2KEG.js";
+import { s as hr } from "./softmax-Bv_6lyMX.js";
+import { m as Ee } from "./max-Ddnnb5xe.js";
+import { t as nc } from "./tensor-JwS7ZYY6.js";
+import { r as sc } from "./range-CRuAh-gd.js";
+import { v as ic } from "./variable-BuddVFLa.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{range-D9CZhVlR.js → range-CRuAh-gd.js}

@@ -1,4 +1,4 @@
-import { E as e, R as f } from "./index-UdZhlibC.js";
+import { E as e, R as f } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{reciprocal-CGB48wZB.js → reciprocal-BvGAyKyu.js}

@@ -1,4 +1,4 @@
-import { B as c, C as n, E as e, V as s } from "./index-UdZhlibC.js";
+import { B as c, C as n, E as e, V as s } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{register_all_kernels-DnbAyBXt.js → register_all_kernels-BwDSRN-f.js}

@@ -1,35 +1,35 @@
-import { B as oe, C as se, E as le, cl as ps, cm as hs, co as en, cp as fs, cq as ms, cs as gs, I as xs, cr as Cs, ct as bs, bp as To, l as Y, dd as tn, cz as sn, n as Do, z as nn, br as Eo, cD as Is, cE as ys, cF as $s, aq as Nc, cI as on, y as vc, a as an, k as Ne, dl as ks, cL as rn, cM as ws, X as It, bw as Fo, cQ as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cS as vs, dv as dn, dw as pn, al as xe, dx as S, aT as Sc, o as Rc, T as Ao, dy as Tc, cT as Dc, x as Ec, q as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aO as Vt, d7 as Vo, aM as Wo, d4 as Ac, aL as Wt, aj as Ke, d8 as Mo, as as hn, an as _c, j as z, bg as Ge, ap as Lc, at as fn, au as mn, K as Vc, a2 as Wc, J as Bo, a1 as ne, cd as Go, cn as zo, bE as Uo, p as me, a7 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, cu as jo, cv as Yo, bL as Qo, bK as Zo, aP as Jo, dc as ea, bM as gn, cx as ta, aY as sa, Q as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cy as ca, bQ as la, a8 as xn, bR as Cn, b6 as ua, bS as da, u as nt, H as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cA as Kn, dz as mt, dA as Ca, cB as qn, df as Bc, Y as Gc, aR as ba, cC as Ia, bW as In, aV as ya, az as $a, ao as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, a4 as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cH as Ea, U as zc, b_ as Fa, b$ as Pa, cJ as Oa, cK as Aa, dm as _a, c0 as La, aS as Va, bh as Wa, S as Ma, ad as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cU as Xa, c2 as Ka, P as qa, a5 as ja, c3 as Ya, cN as wn, dB as Qa, dC as Za, dD as Ja, R as er, V as Nn, bc as tr, cO as sr, bd as nr, cP as or, c5 as ar, b8 as rr, b3 as vn, a3 as ir, dn as cr, aQ as lr, c6 as Sn, a9 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, L as gr, cR as xr, av as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, N as $r, be as kr, ax as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, W as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aI as Rr, bq as ol, ae as al, ak as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, af as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, ay as Tr, bx as Cl, ar as bl, bz as Il, bB as yl } from "./index-UdZhlibC.js";
-import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-DbSpgl4o.js";
-import { r as re } from "./reshape-BR0eoLYN.js";
-import { b as jl } from "./broadcast_to-Ba9h_8DO.js";
-import { c as os } from "./concat-CbXTetof.js";
-import { c as Yl, s as Ql } from "./sin-CiBxrDqX.js";
-import { w as Lr, o as Vr, n as ss, a as zs, r as Us, p as Wr, f as Zl, g as Jl, d as eu, b as tu, e as su, k as nu, m as ou, q as au, v as ru, x as iu, s as cu, c as lu, t as uu, u as du } from "./ops-aRTXR2Sr.js";
-import { m as Tt } from "./mat_mul-dwmZz69e.js";
-import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-Ce6pu9IM.js";
-import { e as $u, l as ku, a as wu } from "./log_sum_exp-BnmCkHWl.js";
-import { f as Nu } from "./dropout-DPfPgWWe.js";
-import { g as vu } from "./gather-Bbh8DHhM.js";
-import { c as Hs, z as Hr } from "./zeros-DnPT2nD4.js";
-import { s as as } from "./slice-BHbDHObE.js";
-import { m as Su } from "./max-ByjEGoFx.js";
-import { r as Ru } from "./reciprocal-CGB48wZB.js";
-import { s as Xs } from "./split-CRU0PjVV.js";
-import { s as Tu } from "./softmax-JMEIUo2J.js";
-import { s as Du } from "./stack-ikk2Y8_P.js";
-import { s as Eu } from "./sum-NLYbiDag.js";
-import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-Czipaeb6.js";
-import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-C9Ut8n0Q.js";
-import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-5DTW2tFV.js";
+import { B as oe, C as se, E as le, cl as ps, cm as hs, co as en, cp as fs, cq as ms, cs as gs, I as xs, cr as Cs, ct as bs, bp as To, l as Y, dd as tn, cz as sn, n as Do, z as nn, br as Eo, cD as Is, cE as ys, cF as $s, aq as Nc, cI as on, y as vc, a as an, k as Ne, dl as ks, cL as rn, cM as ws, X as It, bw as Fo, cQ as Ns, dg as cn, dj as ln, b as Po, bA as Oo, ds as un, cS as vs, dv as dn, dw as pn, al as xe, dx as S, aT as Sc, o as Rc, T as Ao, dy as Tc, cT as Dc, x as Ec, q as Fc, aU as Pc, c as Oc, ac as L, d2 as _o, d3 as Lo, e as Me, aO as Vt, d7 as Vo, aM as Wo, d4 as Ac, aL as Wt, aj as Ke, d8 as Mo, as as hn, an as _c, j as z, bg as Ge, ap as Lc, at as fn, au as mn, K as Vc, a3 as Wc, J as Bo, a1 as ne, cd as Go, cn as zo, bE as Uo, p as me, a7 as st, bF as Ho, bG as Xo, bH as Ko, ab as ct, bI as qo, cu as jo, cv as Yo, bL as Qo, bK as Zo, aP as Jo, dc as ea, bM as gn, cx as ta, aY as sa, Q as na, bN as oa, by as Te, c9 as aa, bO as ra, bP as ia, cy as ca, bQ as la, a8 as xn, bR as Cn, b6 as ua, bS as da, u as nt, H as Mc, bT as pa, bU as ha, bV as fa, cb as ma, cc as ga, de as xa, bo as Ss, bi as Ft, bl as bn, cA as Kn, dz as mt, dA as Ca, cB as qn, df as Bc, Y as Gc, aR as ba, cC as Ia, bW as In, aV as ya, az as $a, ao as yt, d9 as ka, da as wa, b7 as Na, ca as va, di as Sa, a5 as Ra, G as Ta, dk as Da, bX as yn, b0 as $n, bZ as kn, cH as Ea, U as zc, b_ as Fa, b$ as Pa, cJ as Oa, cK as Aa, dm as _a, c0 as La, aS as Va, bh as Wa, S as Ma, ad as Ba, b9 as Ga, bb as za, ba as Ua, c1 as Ha, cU as Xa, c2 as Ka, P as qa, a2 as ja, c3 as Ya, cN as wn, dB as Qa, dC as Za, dD as Ja, R as er, V as Nn, bc as tr, cO as sr, bd as nr, cP as or, c5 as ar, b8 as rr, b3 as vn, a4 as ir, dn as cr, aQ as lr, c6 as Sn, a9 as Rn, c7 as Tn, bY as Dn, c4 as ur, dE as dr, dF as pr, dp as hr, dq as fr, dr as mr, L as gr, cR as xr, av as En, dt as Cr, dG as br, dH as Ir, bJ as Fn, du as yr, N as $r, be as kr, ax as Pn, b4 as wr, c8 as Nr, f as vr, dI as Sr, dJ as jn, d5 as Yn, d6 as Uc, t as Hc, dK as Xc, dL as Kc, bu as qc, b5 as jc, bf as Yc, bm as Qc, W as Zc, bn as Jc, bv as el, b2 as tl, bj as sl, bk as nl, aI as Rr, bq as ol, ae as al, ak as rl, bs as il, dh as cl, aW as ll, aX as ul, aZ as dl, a_ as pl, af as hl, bt as fl, b1 as ml, bC as gl, a$ as xl, ay as Tr, bx as Cl, ar as bl, bz as Il, bB as yl } from "./index-BoWRt-10.js";
+import { e as Dr, s as Er, l as Fr, a as Pr, b as Or, m as Ar, c as _r, r as Gs, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Ll, A as Vl, B as Wl, C as Ml, p as Bl, F as Gl, G as zl, H as Ul, I as Hl, J as Xl, K as Kl, L as ql } from "./random_width-sZORGo5k.js";
+import { r as re } from "./reshape-CdBq1WJ6.js";
+import { b as jl } from "./broadcast_to-CdbwV-Dj.js";
+import { c as os } from "./concat-CsxrgovM.js";
+import { c as Yl, s as Ql } from "./sin-BeA3tsEd.js";
+import { w as Lr, o as Vr, n as ss, a as zs, r as Us, p as Wr, f as Zl, g as Jl, d as eu, b as tu, e as su, k as nu, m as ou, q as au, v as ru, x as iu, s as cu, c as lu, t as uu, u as du } from "./ops-BFGCx8Ri.js";
+import { m as Tt } from "./mat_mul-8m8pfdcx.js";
+import { w as Mr, k as Ye, h as Ve, x as pu, z as hu, A as fu, y as mu, D as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as Rs, S as Br, a as Gr, B as zr, C as Ur, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-BJEXVvjX.js";
+import { e as $u, l as ku, a as wu } from "./log_sum_exp-DbjkV734.js";
+import { f as Nu } from "./dropout-DYs5QFGQ.js";
+import { g as vu } from "./gather-CMMy2KEG.js";
+import { c as Hs, z as Hr } from "./zeros--BdLQ3oG.js";
+import { s as as } from "./slice-BiOsknYS.js";
+import { m as Su } from "./max-Ddnnb5xe.js";
+import { r as Ru } from "./reciprocal-BvGAyKyu.js";
+import { s as Xs } from "./split-B-dikLRw.js";
+import { s as Tu } from "./softmax-Bv_6lyMX.js";
+import { s as Du } from "./stack-B17UN2nn.js";
+import { s as Eu } from "./sum-66ew2byf.js";
+import { a as j, u as ue, c as ut, i as dt, b as Fu, d as Pt, t as Pe, e as kt, f as gt, g as Xr, r as Ot, h as Be, j as Pu, k as Ou, l as On, z as Au, m as An, n as Kr, o as _u, p as Lu, q as Vu, v as Wu, w as Mu, x as Bu, y as Gu, A as zu, B as Uu, C as Hu, D as ft, E as Xu, F as Ku, G as qr, H as qu, I as ju, J as Yu, K as Qu, L as Zu, M as Ju, N as ed, O as td, P as sd, Q as nd, R as od, S as ad, T as rd, U as id, V as cd, W as ld, X as ud, Y as dd, Z as pd, _ as hd, $ as fd, a0 as md, a1 as gd, a2 as xd, a3 as Cd, a4 as bd, a5 as Id, a6 as yd, a7 as $d, a8 as kd, a9 as wd, aa as Nd, ab as vd, ac as Sd, ad as Rd, ae as Td, af as Dd, ag as Ed, ah as Fd, ai as Pd } from "./shared-B8ztnyEk.js";
+import { m as xt, A as Ts, B as Ds, C as Es, D as jr, F as Yr, H as Qr, G as je, I as Zr, J as Jr, K as ei, L as ti, M as si, z as ni, y as oi, x as ai, w as ri, u as ii, E as ci, T as li, U as Od, V as Ad, W as _d, X as Ld, Y as Vd, Z as Wd, N as ui, O as di, P as pi, Q as hi, c as fi, f as Md, S as Bd, _ as Gd } from "./backend_util-TE7aTPhZ.js";
+import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Fs } from "./axis_util-BzbKo31C.js";
 import { s as zd } from "./index-C4L8Cm77.js";
 import { n as mi, b as gi, a as xi } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { c as wt } from "./scatter_nd_util-OjyAxku2.js";
-import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DffelNS-.js";
-import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-DS5waSIY.js";
-import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-CXDy3EN7.js";
-import { R as hh, r as H, a as fh } from "./Reshape-CEsEp0AI.js";
-import { M as Ri } from "./mulmat_packed_gpu-IGPBp6h9.js";
-import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-CVYNbZxu.js";
+import { c as wt } from "./scatter_nd_util-DUstGbU1.js";
+import { h as Qn, j as Ud, k as Hd, l as Xd, m as Kd, n as qd, o as jd, P as _n, p as ze, u as We, q as Ci, c as bi, T as _e, E as Ii, g as yi, a as $i, r as Yd, s as Qd, t as Z, v as Mt, w as Zd, x as Zn, y as Jd, z as ep, A as Bt, B as tp, C as sp, D as Ks, F as jt, G as Yt, H as np, I as op, J as Jn, K as ap, L as rp, M as Vs, N as ip, O as cp, Q as lp, R as Qt, S as Ws, U as up, f as Ce, V as we, W as Zt, X as Jt, Y as dp, d as eo, e as to, i as ki, Z as pp, _ as hp, $ as fp, a0 as mp, a1 as gp, a2 as xp, a3 as Gt } from "./gpgpu_math-DGNLNL4I.js";
+import { s as wi, a as Cp, t as Ni, b as bp, c as Ip, d as vi, e as yp, n as $p, f as kp, g as wp, h as Np, i as vp, j as Sp, k as Rp, l as Tp, o as Dp, p as Ep, q as Fp, r as Pp, u as Op, v as Ap, w as _p, x as Lp, y as Vp, z as Wp, A as Mp, B as Bp, C as Gp, D as zp, E as Up, F as Hp, G as Xp, H as Kp, I as qp, J as jp, K as Yp, L as Qp, M as Si, N as Zp, O as Jp, P as eh, Q as th, R as sh, S as nh, T as oh, U as ah, V as rh, W as ih } from "./shared-wS99K7_n.js";
+import { a as Se, c as ch, U as it, d as tt, e as qe, A as so, f as Nt, B as Ln, h as Vn, m as At, u as ae, C as Ue, b as ke, i as Le, j as Wn, k as pt, l as vt, n as lh, o as uh, p as dh, q as ph } from "./kernel_funcs_utils-BYKWV8Aa.js";
+import { R as hh, r as H, a as fh } from "./Reshape-DH5srBP0.js";
+import { M as Ri } from "./mulmat_packed_gpu-VSekgsNv.js";
+import { t as Ti, s as Mn, a as zt, m as mh, r as gh, b as xh, c as Ch, d as bh } from "./RealDiv-Dy0p8Bvo.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{reshape-BR0eoLYN.js → reshape-CdBq1WJ6.js}

@@ -1,4 +1,4 @@
-import { B as o, C as t, E as a, K as p } from "./index-UdZhlibC.js";
+import { B as o, C as t, E as a, K as p } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{scatter_nd_util-OjyAxku2.js → scatter_nd_util-DUstGbU1.js}

@@ -1,4 +1,4 @@
-import { j as p, a1 as w } from "./index-UdZhlibC.js";
+import { j as p, a1 as w } from "./index-BoWRt-10.js";
 function k(o, t, r) {
   const n = t.rank > 1 ? t.shape[t.rank - 1] : 1, e = t.rank > 1 ? t.rank - 1 : 1, h = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${r.shape}, indices.shape: ${t.shape}, shape: ${o}, sliceDim: ${n}, and batchDim: ${e}.`;
   if (r.rank < e)

package/dist/{selu_util-Ce6pu9IM.js → selu_util-BJEXVvjX.js}

@@ -1,6 +1,6 @@
-import { l as N, aq as H, B as _, C as S, E as L, ar as te, as as ne, an as se, ap as re, at as ie, au as oe, av as le, a as ue, aw as fe, ax as O } from "./index-UdZhlibC.js";
-import { r as ce } from "./reshape-BR0eoLYN.js";
-import { s as ae } from "./sum-NLYbiDag.js";
+import { l as N, aq as H, B as _, C as S, E as L, ar as te, as as ne, an as se, ap as re, at as ie, au as oe, av as le, a as ue, aw as fe, ax as O } from "./index-BoWRt-10.js";
+import { r as ce } from "./reshape-CdBq1WJ6.js";
+import { s as ae } from "./sum-66ew2byf.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{shared-Czipaeb6.js → shared-B8ztnyEk.js}

@@ -1,9 +1,9 @@
-import { l as Lt, bf as Ce, j as V, k as Pt, a1 as H, bg as et, aa as ht, bh as dt, bi as at, a6 as De, a7 as st, am as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, al as K, bn as At, bo as W, bp as _t, bq as Vt, ae as Ct, br as Dt, ak as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, af as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, ao as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ar as Yt, W as Je, bz as te, bA as ee, bB as ne, aM as pt, bC as se, bD as ot } from "./index-UdZhlibC.js";
-import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-C9Ut8n0Q.js";
-import { g as pn, a as wn, e as In, c as bn } from "./axis_util-5DTW2tFV.js";
-import { b as xn } from "./broadcast_to-Ba9h_8DO.js";
-import { r as En } from "./reshape-BR0eoLYN.js";
-import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-Ce6pu9IM.js";
+import { l as Lt, bf as Ce, j as V, k as Pt, a1 as H, bg as et, aa as ht, bh as dt, bi as at, a6 as De, a7 as st, am as We, b2 as $e, bj as Ue, bk as ze, bl as Be, bm as qt, al as K, bn as At, bo as W, bp as _t, bq as Vt, ae as Ct, br as Dt, ak as Wt, bs as $t, aW as Ut, aX as zt, aZ as Bt, a_ as jt, af as Gt, bt as Zt, b1 as Kt, bu as Ht, a$ as je, ao as Ge, bv as Xt, b5 as Ze, bw as Ke, p as Jt, u as He, t as Xe, bx as Qt, by as It, ar as Yt, W as Je, bz as te, bA as ee, bB as ne, aM as pt, bC as se, bD as ot } from "./index-BoWRt-10.js";
+import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-TE7aTPhZ.js";
+import { g as pn, a as wn, e as In, c as bn } from "./axis_util-BzbKo31C.js";
+import { b as xn } from "./broadcast_to-CdbwV-Dj.js";
+import { r as En } from "./reshape-CdBq1WJ6.js";
+import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-BJEXVvjX.js";
 import { g as vn } from "./_commonjsHelpers-ByX85dGu.js";
 function Sn(e, t) {
   for (var n = 0; n < t.length; n++) {

package/dist/{shared-DS5waSIY.js → shared-wS99K7_n.js}

@@ -1,4 +1,4 @@
-import { s as l } from "./shared-Czipaeb6.js";
+import { s as l } from "./shared-B8ztnyEk.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sin-CiBxrDqX.js → sin-BeA3tsEd.js}

@@ -1,4 +1,4 @@
-import { B as o, C as t, E as c, a8 as a, a9 as e } from "./index-UdZhlibC.js";
+import { B as o, C as t, E as c, a8 as a, a9 as e } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{slice-BHbDHObE.js → slice-BiOsknYS.js}

@@ -1,4 +1,4 @@
-import { B as c, C as e, E as a, W as l } from "./index-UdZhlibC.js";
+import { B as c, C as e, E as a, W as l } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{softmax-JMEIUo2J.js → softmax-Bv_6lyMX.js}

@@ -1,4 +1,4 @@
-import { B as r, C as f, E as e, S as i } from "./index-UdZhlibC.js";
+import { B as r, C as f, E as e, S as i } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{split-CRU0PjVV.js → split-B-dikLRw.js}

@@ -1,4 +1,4 @@
-import { B as p, C as i, E as a, L as c } from "./index-UdZhlibC.js";
+import { B as p, C as i, E as a, L as c } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{stack-ikk2Y8_P.js → stack-B17UN2nn.js}

@@ -1,4 +1,4 @@
-import { B as e, D as c, l as n, E as k, P as i } from "./index-UdZhlibC.js";
+import { B as e, D as c, l as n, E as k, P as i } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sum-NLYbiDag.js → sum-66ew2byf.js}

@@ -1,4 +1,4 @@
-import { B as e, C as u, X as c, E as l, Y as m } from "./index-UdZhlibC.js";
+import { B as e, C as u, X as c, E as l, Y as m } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor-Do9PKbIE.js → tensor-JwS7ZYY6.js}

@@ -1,4 +1,4 @@
-import { Z as t, _ as a } from "./index-UdZhlibC.js";
+import { Z as t, _ as a } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor2d-CWHxHpLh.js → tensor2d-wxPAnDQy.js}

@@ -1,4 +1,4 @@
-import { $ as t, Z as s, _ as a } from "./index-UdZhlibC.js";
+import { $ as t, Z as s, _ as a } from "./index-BoWRt-10.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/Adam.js

@@ -1,7 +1,7 @@
 import { adamAdjust as b } from "../ops/adamAdjust.js";
 import { adamMoments as d } from "../ops/adamMoments.js";
-import { O as g, e as h, t as o, d as B } from "../index-UdZhlibC.js";
-import { z as M } from "../zeros-DnPT2nD4.js";
+import { O as g, e as h, t as o, d as B } from "../index-BoWRt-10.js";
+import { z as M } from "../zeros--BdLQ3oG.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/AdamExt.js

@@ -1,4 +1,4 @@
-import { a as r, b as c, c as h, e as o } from "../index-UdZhlibC.js";
+import { a as r, b as c, c as h, e as o } from "../index-BoWRt-10.js";
 import { AdamOptimizer as g } from "./Adam.js";
 class y extends g {
   constructor(t, e, s, i, a) {