@genai-fi/nanogpt 0.4.1 → 0.4.2

package/dist/ops/webgl/gatherSub.js

@@ -1,4 +1,4 @@
-import { r as l } from "../../index-XjBAhiFO.js";
+import { r as l } from "../../index-C4JCoBvj.js";
 class u {
   variableNames = ["labels", "logits", "values"];
   outputShape;

package/dist/ops/webgl/gelu.d.ts

@@ -0,0 +1,2 @@
+import { KernelFunc } from '@tensorflow/tfjs-core';
+export declare const gelu: KernelFunc;

package/dist/ops/webgl/gelu.js

@@ -0,0 +1,50 @@
+import { r as a } from "../../index-C4JCoBvj.js";
+import { u as s, C as x } from "../../kernel_funcs_utils-CAd1h9X1.js";
+const t = 0.7978845608028654, r = 0.044715, c = x + `
+    float x3 = x * x * x;
+    float inner = x + ${r} * x3;
+    inner = ${t} * inner;
+    inner = tanh(inner);
+    inner = 0.5 * (1.0 + inner);
+    x = x * inner;
+    return x;
+`, d = s({ opSnippet: c }), i = {
+  kernelName: "Gelu",
+  backendName: "webgl",
+  kernelFunc: d
+};
+a(i);
+class f {
+  // Inputs: dy, x
+  variableNames = ["dy", "x"];
+  outputShape;
+  userCode;
+  constructor(n) {
+    this.outputShape = n, this.userCode = `
+            void main() {
+                float dy = getDyAtOutCoords();
+                float x  = getXAtOutCoords();
+                float x2 = x * x;
+                float x3 = x2 * x;
+                float u  = ${t} * (x + ${r} * x3);
+                float t  = tanh(u);
+                float sech2 = 1.0 - t * t;
+                float du_dx = ${t} * (1.0 + 3.0 * ${r} * x2);
+                float dgelu = 0.5 * (1.0 + t) + 0.5 * x * sech2 * du_dx;
+                setOutput(dy * dgelu);
+            }`;
+  }
+}
+function g(e) {
+  const { dy: n, x: o } = e.inputs, u = e.backend, l = new f(o.shape);
+  return u.runWebGLProgram(l, [n, o], "float32");
+}
+const p = {
+  kernelName: "GeluGrad",
+  backendName: "webgl",
+  kernelFunc: g
+};
+a(p);
+export {
+  d as gelu
+};

package/dist/ops/webgl/mulDropout.js

@@ -1,4 +1,4 @@
-import { r as m } from "../../index-XjBAhiFO.js";
+import { r as m } from "../../index-C4JCoBvj.js";
 class f {
   variableNames = ["a", "b"];
   outputShape;

package/dist/ops/webgl/qkv.js

@@ -1,4 +1,4 @@
-import { r as i } from "../../index-XjBAhiFO.js";
+import { r as i } from "../../index-C4JCoBvj.js";
 class l {
   variableNames = ["x", "kernel"];
   outputShape;

package/dist/ops/webgl/rope.js

@@ -1,4 +1,4 @@
-import { r as u } from "../../index-XjBAhiFO.js";
+import { r as u } from "../../index-C4JCoBvj.js";
 class l {
   variableNames = ["x", "sin", "cos"];
   outputShape;

package/dist/ops/webgl/scatterSub.js

@@ -1,4 +1,4 @@
-import { r as i } from "../../index-XjBAhiFO.js";
+import { r as i } from "../../index-C4JCoBvj.js";
 class u {
   variableNames = ["labels", "softmaxProbs", "dy"];
   outputShape;

package/dist/{range-DQMNzBWs.js → range-9AzeApCc.js}

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

package/dist/{reshape-DFzh97Sc.js → reshape-Boe4DuIO.js}

@@ -1,4 +1,4 @@
-import { o, h as t, E as a, q as p } from "./index-XjBAhiFO.js";
+import { o, h as t, E as a, q as p } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sin-BYM-U4Ut.js → sin-KmhiDuMa.js}

@@ -1,4 +1,4 @@
-import { o, h as t, E as c, Q as a, T as e } from "./index-XjBAhiFO.js";
+import { o, h as t, E as c, Q as a, T as e } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{slice_util-CnVNPQI-.js → slice_util-19zDNNSn.js}

@@ -1,5 +1,5 @@
-import { o as u, h as p, k as g, w as m, E as w, a1 as x, j as i } from "./index-XjBAhiFO.js";
-import { r as y } from "./reshape-DFzh97Sc.js";
+import { o as u, h as p, k as g, w as m, E as w, a1 as x, j as i } from "./index-C4JCoBvj.js";
+import { r as y } from "./reshape-Boe4DuIO.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{softmax-4DOn6cPq.js → softmax-Cujsg4ay.js}

@@ -1,4 +1,4 @@
-import { o as r, h as f, E as e, S as i } from "./index-XjBAhiFO.js";
+import { o as r, h as f, E as e, S as i } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{split-CkbeVdF8.js → split-DbcNm1-i.js}

@@ -1,4 +1,4 @@
-import { o as p, h as i, E as a, u as c } from "./index-XjBAhiFO.js";
+import { o as p, h as i, E as a, u as c } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{stack-DaIMO5iX.js → stack-D1YjmgKN.js}

@@ -1,4 +1,4 @@
-import { o as e, i as c, j as n, E as i, P as k } from "./index-XjBAhiFO.js";
+import { o as e, i as c, j as n, E as i, P as k } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sum-C6u3xMi3.js → sum-R28pucR5.js}

@@ -1,4 +1,4 @@
-import { o as e, h as u, x as c, E as l, y as m } from "./index-XjBAhiFO.js";
+import { o as e, h as u, x as c, E as l, y as m } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor-Cu1fU7H7.js → tensor-BVeHdl7V.js}

@@ -1,4 +1,4 @@
-import { J as t, K as a } from "./index-XjBAhiFO.js";
+import { J as t, K as a } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor2d-D0CKdG6B.js → tensor2d-DqFGNs_K.js}

@@ -1,4 +1,4 @@
-import { I as t, J as s, K as a } from "./index-XjBAhiFO.js";
+import { I as t, J as s, K as a } from "./index-C4JCoBvj.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.