@genai-fi/nanogpt 0.19.1 → 0.20.0

package/dist/binary_op_util-pKXltfxI.js

@@ -1,192 +0,0 @@
-var t;
-(function(e) {
-  e[e.ADD = 0] = "ADD", e[e.ATAN2 = 1] = "ATAN2", e[e.COMPLEX_MULTIPLY_IMAG = 2] = "COMPLEX_MULTIPLY_IMAG", e[e.COMPLEX_MULTIPLY_REAL = 3] = "COMPLEX_MULTIPLY_REAL", e[e.DIV = 4] = "DIV", e[e.ELU_DER = 5] = "ELU_DER", e[e.EQUAL = 6] = "EQUAL", e[e.FLOOR_DIV = 7] = "FLOOR_DIV", e[e.GREATER = 8] = "GREATER", e[e.GREATER_EQUAL = 9] = "GREATER_EQUAL", e[e.LESS = 10] = "LESS", e[e.LESS_EQUAL = 11] = "LESS_EQUAL", e[e.LOGICAL_AND = 12] = "LOGICAL_AND", e[e.LOGICAL_OR = 13] = "LOGICAL_OR", e[e.MAX = 14] = "MAX", e[e.MIN = 15] = "MIN", e[e.MOD = 16] = "MOD", e[e.MUL = 17] = "MUL", e[e.NOT_EQUAL = 18] = "NOT_EQUAL", e[e.POW = 19] = "POW", e[e.PRELU = 20] = "PRELU", e[e.SQUARED_DIFFERENCE = 21] = "SQUARED_DIFFERENCE", e[e.SUB = 22] = "SUB";
-})(t || (t = {}));
-const r = "let resultTemp = a + b;", L = "let resultTemp = atan2(a, b);", E = "let resultTemp = areal * breal - aimag * bimag;", c = "let resultTemp = areal * bimag + aimag * breal;", u = "let resultTemp = a / b;", n = "let resultTemp = select(a * (b + 1.0), a, b >= b - b);", A = `
-  let zero = sign(a) * 0 + 0;
-  let one = sign(b) * 0 + 1;
-  let resultTemp = select(zero, one, a == b);
-`, N = `
-  let remainder =
-      select(a % b, round(a % b), (round(a) == a) & (round(b) == b));
-  let quotient = (a - remainder) / b;
-  let resultTemp =
-      round(select(quotient, quotient - 1, sign(remainder) == -sign(b)));
-`, _ = `
-  let zero = sign(a) * 0 + 0;
-  let one = sign(b) * 0 + 1;
-  let resultTemp = select(zero, one, a > b);
-`, m = `
-  let zero = sign(a) * 0 + 0;
-  let one = sign(b) * 0 + 1;
-  let resultTemp = select(zero, one, a >= b);
-`, R = `
-  let zero = sign(a) * 0 + 0;
-  let one = sign(b) * 0 + 1;
-  let resultTemp = select(zero, one, a < b);
-`, U = `
-  let zero = sign(a) * 0 + 0;
-  let one = sign(b) * 0 + 1;
-  let resultTemp = select(zero, one, a <= b);
-`, I = "return f32(a >= 1.0 && b >= 1.0);", f = `return (vec4<f32>(a >= vec4<f32>(1.0)) *
-  vec4<f32>(b >= vec4<f32>(1.0)));`, M = "return f32(a >= 1.0 || b >= 1.0);", i = `return min(vec4<f32>(a >= vec4<f32>(1.0)) +
-  vec4<f32>(b >= vec4<f32>(1.0)), vec4<f32>(1.0));`, D = "let resultTemp = max(a, b);", v = "let resultTemp = min(a, b);", P = `
-  let isNaN = b == 0.;
-  var resultTemp = a % b;
-  resultTemp = select((resultTemp + b) % b, resultTemp,
-      (a < 0. && b < 0.) || (a >= 0. && b > 0.));
-`, g = `
-  let isNaN = !vec4<bool>(b);
-  var resultTemp = vec4<f32>(a % b);
-  if (!((a[0] < 0. && b[0] < 0.) || (a[0] >= 0. && b[0] > 0.))) {
-    resultTemp[0] = (resultTemp[0] + b[0]) % b[0];
-  }
-  if (!((a[1] < 0. && b[1] < 0.) || (a[1] >= 0. && b[1] > 0.))) {
-    resultTemp[1] = (resultTemp[1] + b[1]) % b[1];
-  }
-  if (!((a[2] < 0. && b[2] < 0.) || (a[2] >= 0. && b[2] > 0.))) {
-    resultTemp[2] = (resultTemp[2] + b[2]) % b[2];
-  }
-  if (!((a[3] < 0. && b[3] < 0.) || (a[3] >= 0. && b[3] > 0.))) {
-    resultTemp[3] = (resultTemp[3] + b[3]) % b[3];
-  }
-`, C = "let resultTemp = a * b;", S = `
-  var resultTemp = f32(a != b);
-  let valueForNaN = 1.0;
-`, d = `
-  var resultTemp = vec4<f32>(a != b);
-  let valueForNaN = 1.0;
-`, T = `
-  let isNaN = a < 0.0 && floor(b) < b;
-  if (b == 0.0) {
-    return 1.0;
-  }
-  var resultTemp = select(sign(a) * pow(abs(a), b), pow(abs(a), b),
-      round(abs(b) % 2.0) != 1.0);
-`, G = `
-  let isModRound1Bool = vec4<i32>(round(abs(b) % vec4<f32>(2.0))) == vec4<i32>(1);
-  let isModRound1 = vec4<f32>(isModRound1Bool);
-  let multiplier = sign(a) * isModRound1 + (vec4<f32>(1.0) - isModRound1);
-  var resultTemp = multiplier * pow(abs(a), b);
-
-  // Ensure that a^0 = 1, including 0^0 = 1 as this correspond to TF and JS
-  let isExpZero = b == vec4<f32>(0.0);
-  if (isExpZero.r) {
-    resultTemp.r = 1.0;
-  }
-  if (isExpZero.g) {
-    resultTemp.g = 1.0;
-  }
-  if (isExpZero.b) {
-    resultTemp.b = 1.0;
-  }
-  if (isExpZero.a) {
-    resultTemp.a = 1.0;
-  }
-  let isNaN = (a < vec4<f32>(0.0)) & (floor(b) < b);
-`, Q = "if (a < 0.0) { return b * a; }  return a;", k = `
-  let aLessThanZero = vec4<f32>(a < vec4<f32>(0.0));
-  return (aLessThanZero * (b * a)) + ((vec4<f32>(1.0) - aLessThanZero) * a);
-`, F = "let resultTemp = (a - b) * (a - b);", z = "let resultTemp = a - b;";
-function X(e, a) {
-  let s;
-  do {
-    switch (e) {
-      case t.ATAN2:
-        s = L;
-        break;
-      case t.MAX:
-        s = D;
-        break;
-      case t.MIN:
-        s = v;
-        break;
-      case t.MOD:
-        s = a ? g : P;
-        break;
-      case t.NOT_EQUAL:
-        s = a ? d : S;
-        break;
-      case t.POW:
-        s = a ? G : T;
-        break;
-      default:
-        continue;
-    }
-    let o, l, b;
-    return a ? (o = "isnanVec4", l = "vec4<f32>", b = "vec4<bool>") : (o = "isnan", l = "f32", b = "bool"), `
-      let aIsNaN = ${o}(a);
-      let aPostLegalization = select(a, ${l}(42), aIsNaN);
-      let bIsNaN = ${o}(b);
-      let bPostLegalization = select(b, ${l}(42), bIsNaN);
-      let isNaN = false;
-      let valueForNaN = uniforms.NAN;
-      {
-        let a = aPostLegalization;
-        let b = bPostLegalization;
-        ${s}
-        return select(
-            resultTemp, ${l}(valueForNaN),
-            ${b}(isNaN) | aIsNaN | bIsNaN);
-      }
-    `;
-  } while (!1);
-  switch (e) {
-    case t.ADD:
-      s = r;
-      break;
-    case t.COMPLEX_MULTIPLY_IMAG:
-      s = c;
-      break;
-    case t.COMPLEX_MULTIPLY_REAL:
-      s = E;
-      break;
-    case t.DIV:
-      s = u;
-      break;
-    case t.ELU_DER:
-      s = n;
-      break;
-    case t.EQUAL:
-      s = A;
-      break;
-    case t.FLOOR_DIV:
-      s = N;
-      break;
-    case t.GREATER:
-      s = _;
-      break;
-    case t.GREATER_EQUAL:
-      s = m;
-      break;
-    case t.LESS:
-      s = R;
-      break;
-    case t.LESS_EQUAL:
-      s = U;
-      break;
-    case t.LOGICAL_AND:
-      return a ? f : I;
-    case t.LOGICAL_OR:
-      return a ? i : M;
-    case t.MUL:
-      s = C;
-      break;
-    case t.PRELU:
-      return a ? k : Q;
-    case t.SQUARED_DIFFERENCE:
-      s = F;
-      break;
-    case t.SUB:
-      s = z;
-      break;
-  }
-  return `
-    ${s}
-    return resultTemp;
-  `;
-}
-export {
-  t as B,
-  X as g
-};

package/dist/broadcast_to-eS93CCN_.js

@@ -1,28 +0,0 @@
-import { o as h, q as f, u as p, w as g, E as u, T } from "./index-CUXkjxiT.js";
-import { r as b } from "./reshape-BDOuCSNW.js";
-function m(e, r) {
-  let n = f(e, "broadcastTo", "x");
-  const a = n.shape;
-  if (p(r), r.length < n.rank)
-    throw new Error(`broadcastTo(): shape.length=${r.length} < input.rank=${n.rank}.`);
-  if (r.length > n.rank) {
-    const t = n.shape.slice();
-    for (; t.length < r.length; )
-      t.unshift(1);
-    n = b(n, t);
-  }
-  const s = n.shape, o = Array.from(r);
-  for (let t = r.length - 1; t >= 0; t--)
-    if (s[t] === r[t])
-      o[t] = 1;
-    else if (n.shape[t] !== 1)
-      throw new Error(`broadcastTo(): [${a}] cannot be broadcast to [${r}].`);
-  if (o.map((t, l) => t > 1 ? l : -1).filter((t) => t >= 0).length === 0)
-    return g(n);
-  const i = { x: n }, c = { reps: o };
-  return u.runKernel(T, i, c);
-}
-const E = /* @__PURE__ */ h({ broadcastTo_: m });
-export {
-  E as b
-};

package/dist/checks/appendCache.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>;

package/dist/checks/appendCache.js

@@ -1,22 +0,0 @@
-import { s, h as a } from "../index-CUXkjxiT.js";
-import { t } from "../tensor4d-C6UCG_u8.js";
-async function u(e) {
-  await s(e);
-  const n = t(
-    [
-      [
-        [
-          [0.1, 0.2, 0, 0],
-          [0.1, 0.2, 0, 0],
-          [0, 0, 0, 0],
-          [0, 0, 0, 0]
-        ]
-      ]
-    ],
-    [1, 1, 4, 4]
-  ), r = t([[[[0.1, 0.2, 0.3, 0.4]]]], [1, 1, 1, 4]);
-  return await a().runKernel("AppendCache", { cache: n, item: r }, { maxSize: 4, pastLen: 2 }).array();
-}
-export {
-  u as execute
-};

package/dist/checks/attentionMask.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>;

package/dist/checks/attentionMask.js

@@ -1,37 +0,0 @@
-import { s as i, h as a } from "../index-CUXkjxiT.js";
-import { t } from "../tensor4d-C6UCG_u8.js";
-import { t as e } from "../tensor2d-BnXMKScO.js";
-async function k(n) {
-  await i(n);
-  const s = t(
-    [
-      [
-        [
-          [0.1, 0.2, 0.3, 0.4],
-          [0.3, 0.4, 0.5, 0.6]
-        ]
-      ]
-    ],
-    [1, 1, 2, 4]
-  ), o = t(
-    [
-      [
-        [
-          [0.5, 0.6, 0.5, 0.6],
-          [0.7, 0.8, 0.7, 0.8]
-        ]
-      ]
-    ],
-    [1, 1, 2, 4]
-  ), r = e(
-    [
-      [0, -1 / 0, -1 / 0, -1 / 0],
-      [0, 0, 0, -1 / 0]
-    ],
-    [2, 4]
-  );
-  return await a().runKernel("AttentionMask", { q: s, k: o, mask: r }, { divisor: 0.5, pastLen: 0 }).array();
-}
-export {
-  k as execute
-};

package/dist/checks/check.d.ts

@@ -1,9 +0,0 @@
-interface Result {
-    backend: string;
-    result: unknown;
-    error?: string;
-    passed: boolean;
-    maxError?: number;
-}
-export default function runCheck(check: (backend: string) => Promise<unknown>, epsilon?: number, backends?: string[]): Promise<Result[]>;
-export {};

package/dist/checks/check.js

@@ -1,20 +0,0 @@
-import { arraysClose as p } from "../utilities/arrayClose.js";
-async function d(c, o, n) {
-  const u = n ?? ["cpu", "webgl", "webgpu"], t = [];
-  for (const e of u)
-    try {
-      const r = await c(e);
-      t.push({ backend: e, result: r, passed: !0 });
-    } catch (r) {
-      t.push({ backend: e, error: r.message, result: [], passed: !1 });
-    }
-  const s = await Promise.all(t), l = s[0].result;
-  for (let e = 1; e < s.length; e++) {
-    const r = s[e].result, a = p(l, r);
-    s[e].passed = a <= (o ?? 1e-6), s[e].maxError = a;
-  }
-  return s;
-}
-export {
-  d as default
-};

package/dist/checks/gelu.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>;

package/dist/checks/gelu.js

@@ -1,18 +0,0 @@
-import { s as e, h as o } from "../index-CUXkjxiT.js";
-import { t as s } from "../tensor2d-BnXMKScO.js";
-async function m(t) {
-  await e(t);
-  const r = s(
-    [
-      [0.1, 0.2, 0, 0, 1230, 1232331234, -12234234],
-      [0.1, 0.2, 0, 0, -1230, -1232331234, 12234234],
-      [0, 0, 0, 0, -1, 0, 0],
-      [0, 0, 0, 0, -0.1, 1e-3, 0]
-    ],
-    [4, 7]
-  );
-  return await o().runKernel("Gelu", { x: r }).array();
-}
-export {
-  m as execute
-};

package/dist/checks/index.d.ts

@@ -1,26 +0,0 @@
-import { execute as rope } from './rope';
-import { execute as normRMS } from './normRMS';
-import { execute as qkv } from './qkv';
-import { execute as gelu } from './gelu';
-import { execute as normRMSGrad } from './normRMSGrad';
-import { execute as appendCache } from './appendCache';
-import { execute as attentionMask } from './attentionMask';
-import { execute as matMulGelu } from './matMulGelu';
-import { execute as packUnpack } from './packUnpack';
-import { default as runCheck } from './check';
-import { createWeightStatistics, createTensorStatistics } from './weights';
-declare const checks: {
-    rope: typeof rope;
-    qkv: typeof qkv;
-    gelu: typeof gelu;
-    normRMS: typeof normRMS;
-    normRMSGrad: typeof normRMSGrad;
-    appendCache: typeof appendCache;
-    attentionMask: typeof attentionMask;
-    matMulGelu: typeof matMulGelu;
-    packUnpack: typeof packUnpack;
-    runCheck: typeof runCheck;
-    createLayerWeightStatistics: typeof createWeightStatistics;
-    createWeightStatistics: typeof createTensorStatistics;
-};
-export default checks;

package/dist/checks/index.js

@@ -1,28 +0,0 @@
-import { execute as e } from "./rope.js";
-import { execute as t } from "./normRMS.js";
-import { execute as r } from "./qkv.js";
-import { execute as c } from "./gelu.js";
-import { execute as o } from "./normRMSGrad.js";
-import { execute as a } from "./appendCache.js";
-import { execute as m } from "./attentionMask.js";
-import { execute as i } from "./matMulGelu.js";
-import { execute as u } from "./packUnpack.js";
-import s from "./check.js";
-import { createTensorStatistics as x, createWeightStatistics as p } from "./weights.js";
-const C = {
-  rope: e,
-  qkv: r,
-  gelu: c,
-  normRMS: t,
-  normRMSGrad: o,
-  appendCache: a,
-  attentionMask: m,
-  matMulGelu: i,
-  packUnpack: u,
-  runCheck: s,
-  createLayerWeightStatistics: p,
-  createWeightStatistics: x
-};
-export {
-  C as default
-};

package/dist/checks/matMulGelu.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>;

package/dist/checks/matMulGelu.js

@@ -1,28 +0,0 @@
-import { s as o, h as s } from "../index-CUXkjxiT.js";
-import { t as e } from "../tensor2d-BnXMKScO.js";
-async function i(t) {
-  await o(t);
-  const r = e(
-    [
-      [0.1, 0.2, 9, 10, 11],
-      [0.3, 0.4, -9, -10, -11],
-      [0.3, 0.4, -9, -10, -11],
-      [0.3, 0.4, -9, -10, -11],
-      [0.3, 0.4, -9, -10, -11]
-    ],
-    [5, 5]
-  ), n = e(
-    [
-      [0.5, 0.6, 7e4, -8e3, 0],
-      [0.7, 0.8, -7e4, 8e4, 0],
-      [0.7, 0.8, -7e4, 8e4, 0],
-      [0.7, 0.8, -7e4, 8e4, 0],
-      [0.7, 0.8, -7e4, 8e4, 0]
-    ],
-    [5, 5]
-  );
-  return await s().runKernel("MatMulGelu", { x: n, kernel: r }).array();
-}
-export {
-  i as execute
-};

package/dist/checks/normRMS.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<(number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][])[]>;

package/dist/checks/normRMS.js

@@ -1,16 +0,0 @@
-import { s as u, a0 as A, h } from "../index-CUXkjxiT.js";
-import { a as y } from "../ops-CsXeTq1P.js";
-import { t as p } from "../tensor1d-LMGMIUlr.js";
-import { t as r } from "../tensor-BWFldCso.js";
-const w = Array.from({ length: 2048 * 192 }, () => Math.random()), x = Array.from({ length: 192 }, () => Math.random()), M = Array.from({ length: 2048 * 192 }, () => Math.random());
-async function k(t) {
-  await u(t);
-  const o = p(x, "float32"), n = r(w, [16, 128, 192], "float32"), s = r(M, [16, 128, 192], "float32"), e = (d, g) => {
-    const i = h().runKernel("RMSNorm", { x: d, gamma: g });
-    return y.meanSquaredError(i, s);
-  }, { value: m, grads: a } = A(e)([n, o]), c = await m.array(), f = await a[0].array(), l = await a[1].array();
-  return [c, f, l];
-}
-export {
-  k as execute
-};

package/dist/checks/normRMSGrad.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<(number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][])[]>;

package/dist/checks/normRMSGrad.js

@@ -1,12 +0,0 @@
-import { s as c, h as d } from "../index-CUXkjxiT.js";
-import { t as f } from "../tensor1d-LMGMIUlr.js";
-import { t as r } from "../tensor-BWFldCso.js";
-const y = Array.from({ length: 2048 * 192 }, () => Math.random()), i = Array.from({ length: 192 }, () => Math.random()), l = Array.from({ length: 2048 * 192 }, () => Math.random());
-async function x(t) {
-  await c(t);
-  const o = f(i, "float32"), n = r(y, [16, 128, 192], "float32"), m = r(l, [16, 128, 192], "float32"), a = d().runKernel("RMSNormGrad", { x: n, gamma: o, dy: m }), s = await a[0].array(), e = await a[1].array();
-  return [s, e];
-}
-export {
-  x as execute
-};

package/dist/checks/packUnpack.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>;

package/dist/checks/packUnpack.js

@@ -1,18 +0,0 @@
-import { s as a, h as n } from "../index-CUXkjxiT.js";
-import { t as c } from "../tensor2d-BnXMKScO.js";
-async function i(e) {
-  await a(e);
-  const r = c(
-    [
-      [0.1, 0.2, 0, 0, 1230, 1232331234, -12234234],
-      [0.1, 0.2, 0, 0, -1230, -1232331234, 12234234],
-      [0, 0, 0, 0, -1, 0, 0],
-      [0, 0, 0, 0, -0.1, 1e-3, 0]
-    ],
-    [4, 7]
-  ), t = n().runKernel("Pack16", { x: r });
-  return await n().runKernel("Unpack16", { x: t }).array();
-}
-export {
-  i as execute
-};

package/dist/checks/qkv.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<(number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][])[]>;

package/dist/checks/qkv.js

@@ -1,34 +0,0 @@
-import { W as i, X as u, Y as c, s as l, h } from "../index-CUXkjxiT.js";
-import { t as f } from "../tensor2d-BnXMKScO.js";
-function m(t, e, n) {
-  if (i(t), e != null && e.length !== 3)
-    throw new Error("tensor3d() requires shape to have three numbers");
-  const r = u(t, n);
-  if (r.length !== 3 && r.length !== 1)
-    throw new Error("tensor3d() requires values to be number[][][] or flat/TypedArray");
-  if (r.length === 1 && e == null)
-    throw new Error("tensor3d() requires shape to be provided when `values` are a flat array");
-  return c(t, e, r, n);
-}
-async function y(t) {
-  await l(t);
-  const e = m(
-    [
-      [
-        [0.1, 0.2],
-        [0.3, 0.4]
-      ]
-    ],
-    [1, 2, 2]
-  ), n = f(
-    [
-      [0.5, 0.6, 0.9, 1, 1.3, 1.4],
-      [0.7, 0.8, 1.1, 1.2, 1.5, 1.6]
-    ],
-    [2, 6]
-  ), r = h().runKernel("QKV", { x: e, kernel: n }, { heads: 1 }), o = await r[0].array(), a = await r[1].array(), s = await r[2].array();
-  return [o, a, s];
-}
-export {
-  y as execute
-};

package/dist/checks/rope.d.ts

@@ -1 +0,0 @@
-export declare function execute(backend: string): Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][] | Promise<number | number[] | number[][] | number[][][] | number[][][][] | number[][][][][] | number[][][][][][]>[]>;

package/dist/checks/rope.js

@@ -1,36 +0,0 @@
-import s from "../layers/RoPECache.js";
-import { s as c, h as i } from "../index-CUXkjxiT.js";
-import { t as p } from "../tensor4d-C6UCG_u8.js";
-async function f(r) {
-  await c(r);
-  const n = p(
-    [
-      [
-        [
-          [0.1, 0.2],
-          [0.3, 0.4]
-        ]
-      ]
-    ],
-    [1, 1, 2, 2]
-  ), a = {
-    modelType: "GenAI_NanoGPT_v1",
-    vocabSize: 20,
-    nEmbed: 16,
-    nHead: 2,
-    nLayer: 1,
-    blockSize: 128,
-    mlpFactor: 4,
-    useRope: !0
-  }, e = new s(a);
-  e.ensureRopeCache(120);
-  const o = i().runKernel(
-    "Rope",
-    { x: n, sin: e.getSin(), cos: e.getCos() },
-    { pastLen: 20 }
-  );
-  return Array.isArray(o) ? o.map((t) => t.array()) : o.array();
-}
-export {
-  f as execute
-};

package/dist/checks/weights.d.ts

@@ -1,14 +0,0 @@
-import { default as BaseLayer } from '../layers/BaseLayer';
-import { Tensor } from '@tensorflow/tfjs-core';
-export interface TensorStatistics {
-    mean: number;
-    std: number;
-    min: number;
-    max: number;
-    sparsity: number;
-    isFinite: boolean;
-    hasNaN: boolean;
-    closeToZeroCount: number;
-}
-export declare function createTensorStatistics(weight: Tensor | number[]): Promise<TensorStatistics>;
-export declare function createWeightStatistics(layer: BaseLayer): Promise<Record<string, TensorStatistics>>;

package/dist/checks/weights.js

@@ -1,31 +0,0 @@
-async function b(s) {
-  if (!Array.isArray(s) && s.dtype !== "float32")
-    throw new Error(`Unsupported dtype ${s.dtype} for weight statistics.`);
-  const e = Array.isArray(s) ? s : await s.data(), a = e.length;
-  let r = 0, l = 0, n = e[0], i = e[0], f = 0, d = !0, u = !1, y = 0;
-  for (let c = 0; c < a; c++) {
-    const t = e[c];
-    r += t, l += t * t, t < n && (n = t), t > i && (i = t), t === 0 && f++, Math.abs(t) < 1e-8 && y++, Number.isNaN(t) && (u = !0);
-  }
-  const o = r / a, m = l / a - o * o, p = Math.sqrt(m), N = f / a;
-  return {
-    mean: o,
-    std: p,
-    min: n,
-    max: i,
-    sparsity: N,
-    isFinite: d,
-    hasNaN: u,
-    closeToZeroCount: y
-  };
-}
-async function h(s) {
-  const e = s.trainableVariables, a = {};
-  for (const r of e)
-    a[r.name] = await b(r);
-  return a;
-}
-export {
-  b as createTensorStatistics,
-  h as createWeightStatistics
-};

package/dist/clip_by_value-DDA7rrcT.js

@@ -1,12 +0,0 @@
-import { o as a, q as f, x as m, f as y, E as c, C as p } from "./index-CUXkjxiT.js";
-function u(o, s, t) {
-  const r = f(o, "x", "clipByValue");
-  if (m(s <= t, () => `Error in clip: min (${s}) must be less than or equal to max (${t}).`), s === t)
-    return y(r.shape, s, r.dtype);
-  const n = { x: r }, e = { clipValueMin: s, clipValueMax: t };
-  return c.runKernel(p, n, e);
-}
-const E = /* @__PURE__ */ a({ clipByValue_: u });
-export {
-  E as c
-};

package/dist/complex-DI35Q-gW.js

@@ -1,11 +0,0 @@
-import { o as c, q as s, a6 as n, E as m, a7 as r } from "./index-CUXkjxiT.js";
-function l(o, p) {
-  const a = s(o, "real", "complex"), e = s(p, "imag", "complex");
-  n(a.shape, e.shape, `real and imag shapes, ${a.shape} and ${e.shape}, must match in call to tf.complex().`);
-  const t = { real: a, imag: e };
-  return m.runKernel(r, t);
-}
-const i = /* @__PURE__ */ c({ complex_: l });
-export {
-  i as c
-};