@genai-fi/nanogpt 1.1.1 → 1.1.2

package/dist/{dist-9wGF4ci9.js → dist-BGVm64AY.js}

@@ -4,13 +4,13 @@ import { A as zn, C as Bn, D as Vn, E as Hn, F as Un, I as Wn, M as Gn, N as Kn,
 import { a as sr, o as cr, r as lr, t as ur } from "./slice_util-Dve4TkDn.js";
 import { n as dr, t as fr } from "./concat_util-CWzvQQlK.js";
 import { t as pr } from "./shared-24EoeHJl.js";
-import { a as mr, d as P, i as F, o as hr, r as gr, s as I, t as _r, u as L } from "./webgpu_program-B0PokuRJ.js";
-import { a as vr, c as R, i as z, l as yr, n as B, o as br, r as xr, s as Sr } from "./webgpu_util-DfODg6XI.js";
-import { t as Cr } from "./webgpu-B19Tw8Nd.js";
+import { a as mr, c as P, i as F, l as hr, n as I, o as gr, r as _r, s as vr } from "./webgpu_util-DfODg6XI.js";
+import { t as yr } from "./webgpu-IAn9VotJ.js";
+import { a as br, d as L, i as R, o as xr, r as Sr, s as z, t as Cr, u as B } from "./webgpu_program-B0PokuRJ.js";
 import { n as wr, t as V } from "./binary_op_util-CrYk9LXL.js";
 import { n as Tr, t as H } from "./Reshape-CwxdaU8n.js";
 //#region node_modules/@tensorflow/tfjs-backend-webgpu/dist/base.js
-yr() && Kt("webgpu", async () => {
+hr() && Kt("webgpu", async () => {
 	let e = { powerPreference: A().get("WEBGPU_USE_LOW_POWER_GPU") ? "low-power" : "high-performance" }, t = await navigator.gpu.requestAdapter(e), n = {}, r = [];
 	t.features.has("timestamp-query") && r.push("timestamp-query"), t.features.has("bgra8unorm-storage") && r.push(["bgra8unorm-storage"]), t.features.has("subgroups") && r.push("subgroups"), n.requiredFeatures = r;
 	let i = t.limits;
@@ -21,7 +21,7 @@ yr() && Kt("webgpu", async () => {
 		maxBufferSize: i.maxBufferSize,
 		maxComputeWorkgroupSizeX: i.maxComputeWorkgroupSizeX,
 		maxComputeInvocationsPerWorkgroup: i.maxComputeInvocationsPerWorkgroup
-	}, new Cr(await t.requestDevice(n), "info" in t ? t.info : "requestAdapterInfo" in t ? await t.requestAdapterInfo() : void 0);
+	}, new yr(await t.requestDevice(n), "info" in t ? t.info : "requestAdapterInfo" in t ? await t.requestAdapterInfo() : void 0);
 }, 3);
 //#endregion
 //#region node_modules/@tensorflow/tfjs-backend-webgpu/dist/unary_op_util.js
@@ -110,7 +110,7 @@ function G(e, t = !1, n = !1, r = 3) {
 	else if (e === "sigmoid") i = W(U.SIGMOID, n);
 	else if (e === "leakyrelu") i = W(U.LEAKYRELU, n);
 	else throw Error(`Activation ${e} has not been implemented for the WebGPU backend.`);
-	let a = L(n ? 4 : 1), o = "";
+	let a = B(n ? 4 : 1), o = "";
 	return o = t ? `
       fn activation(a : ${a}, coords : vec${r}<i32>) -> ${a} {
         let b = getPreluActivationWeightsByOutputCoords(coords);
@@ -135,8 +135,8 @@ function _i(e, t, n = !1, r = !1, i = !1, a = 1) {
 
     `, s = t ? "value = getB(batch, col, row);" : "value = getB(batch, row, col);";
 	return `
-  fn mm_readA(batch: i32, row: i32, col: i32) -> ${L(a)} {
-    var value = ${L(a)}(0.0);
+  fn mm_readA(batch: i32, row: i32, col: i32) -> ${B(a)} {
+    var value = ${B(a)}(0.0);
     ${n && i ? o : `
     ${e ? "if(row < uniforms.dimAOuter && col < uniforms.dimInner)" : "if(row < uniforms.aShape[1] && col < uniforms.aShape[2])"}
     {
@@ -146,8 +146,8 @@ function _i(e, t, n = !1, r = !1, i = !1, a = 1) {
     return value;
   }
 
-  fn mm_readB(batch: i32, row: i32, col: i32) -> ${L(a)} {
-    var value = ${L(a)}(0.0);
+  fn mm_readB(batch: i32, row: i32, col: i32) -> ${B(a)} {
+    var value = ${B(a)}(0.0);
     ${s}
     return value;
   }
@@ -156,7 +156,7 @@ function _i(e, t, n = !1, r = !1, i = !1, a = 1) {
 function vi(e, t, n, r, i = !1, a = !1, o = !1, s = 1) {
 	return `
   ${_i(n, r, i, a, o, s)}
-  fn mm_write(batch: i32, row: i32, col: i32, valueIn: ${L(s)}) {
+  fn mm_write(batch: i32, row: i32, col: i32, valueIn: ${B(s)}) {
     ${i && a ? "" : "if (row < uniforms.dimAOuter && col < uniforms.dimBOuter)"}
     {
       var value = valueIn;
@@ -205,7 +205,7 @@ function xi(e, t, n = !1, r = 32, i = !1, a = 32, o = !1) {
   var<workgroup> mm_Asub : array<array<vec${d}<f32>, ${l / d}>, ${u}>;
   var<workgroup> mm_Bsub : array<array<vec4<f32>, ${c / e[0]}>, ${r}>;
 
-  ${I()} {
+  ${z()} {
     let localRow = i32(localId.y);
     let tileRow = localRow * ${p};
     let tileCol = i32(localId.x);
@@ -368,7 +368,7 @@ function wi(e, t, n = !1, r = 32, i = !1, a = 32, o = !1, s = !1) {
     var<workgroup> mm_Asub : array<array<f32, ${u}>, ${d}>;
     var<workgroup> mm_Bsub : array<array<f32, ${l}>, ${r}>;
 
-    ${I()} {
+    ${z()} {
       let batch = ${i ? "0" : "i32(globalId.z)"};
       let batchA = ${i || !s ? "batch" : "batch % uniforms.aShape[0]"};
       let batchB = ${i || !s ? "batch" : "batch % uniforms.bShape[0]"};
@@ -394,7 +394,7 @@ function Ei(e, t = !1) {
 	return `
     var<workgroup> mm_Asub : array<vec4<f32>, ${e[0]}>;
 
-    ${I()} {
+    ${z()} {
       let tileCol = i32(localId.x);
       let globalCol = i32(globalId.x);
       let globalRow = i32(globalId.y);
@@ -450,10 +450,10 @@ var Di = class {
 			1
 		];
 		else {
-			let e = br(t[1], c, t[2], n);
+			let e = gr(t[1], c, t[2], n);
 			this.workgroupSize = e.workgroupSize, this.elementsPerThread = e.elementsPerThread;
 		}
-		this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread);
+		this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread);
 		let l = i != null, u = o != null;
 		l && this.variableNames.push("bias"), u && this.variableNames.push("preluActivationWeights"), this.sequentialAccessByThreads = s, this.transposeA = n, this.transposeB = r, this.addBias = l, this.activation = a, this.hasPreluActivationWeights = u, [this.fitAOuter, this.fitBOuter, this.fitInner] = this.getShapeFit(t[1], t[2], c), this.shaderKey = `matMulPacked_${this.elementsPerThread}_${n}_${r}_${this.activation}_${this.fitAOuter}_${this.fitBOuter}_${this.fitInner}_${this.isVec4}_${this.isVectorA}_${this.sequentialAccessByThreads}`;
 	}
@@ -478,7 +478,7 @@ var Di = class {
 function Oi(e) {
 	return `
     var<workgroup> sumValues : array<f32, ${e}>;
-    ${I()} {
+    ${z()} {
       let coords = getOutputCoords();
       let batch = coords[0];
       let batchA = batch % uniforms.aShape[0];
@@ -521,7 +521,7 @@ var ki = class {
 			x: [],
 			y: [1, 2],
 			z: [0]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize);
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize);
 		let o = r != null, s = a != null;
 		o && this.variableNames.push("bias"), s && this.variableNames.push("preluActivationWeights"), this.transposeA = t, this.transposeB = n, this.addBias = o, this.activation = i, this.hasPreluActivationWeights = s, this.shaderKey = `matMulReduce_${this.activation}_${t}_${n}`;
 	}
@@ -547,7 +547,7 @@ function Ai(e) {
   // shared memory, so it is instruction-Level parallelism for arithmetic
   // operations and others handle IO operations between barrier api, makes ALU
   // and load/store units work simultaneously, could improves the performance.
-  ${I()} {
+  ${z()} {
     let tileRow = i32(localId.y);
     let tileCol = i32(localId.x);
     let globalRow = i32(globalId.y);
@@ -634,7 +634,7 @@ var ji = class {
 			4,
 			4,
 			this.splitedDimInner
-		], this.outputComponent = i ? 4 : 1, i || (this.outputShape[1] < 16 && (this.elementsPerThread[1] = 1), this.outputShape[2] < 16 && (this.elementsPerThread[0] = 1)), this.dispatch = z(this.dispatchLayout, [
+		], this.outputComponent = i ? 4 : 1, i || (this.outputShape[1] < 16 && (this.elementsPerThread[1] = 1), this.outputShape[2] < 16 && (this.elementsPerThread[0] = 1)), this.dispatch = F(this.dispatchLayout, [
 			this.outputShape[0],
 			this.outputShape[1],
 			this.outputShape[2],
@@ -645,14 +645,14 @@ var ji = class {
 		let e = this.outputComponent;
 		return `
       ${_i(!1, this.transposeB, !1, !1, !1, e)}
-      fn mm_write(batch: i32, row : i32, col : i32, value : ${L(e)}) {
+      fn mm_write(batch: i32, row : i32, col : i32, value : ${B(e)}) {
         if (row < uniforms.dimAOuter && col < uniforms.dimBOuter) {
           let coords = vec3<i32>(batch, row, col);
           let flatIndex = getOutputIndexFromCoords(coords);
           // The problem is that we should initialize output to zero before using.
           // Otherwise, the original value will be added to the result.
           for (var i = 0; i < ${e}; i = i + 1) {
-            ${P("&result[flatIndex + i]", `${e > 1 ? "value[i]" : "value"}`, "float32")}
+            ${L("&result[flatIndex + i]", `${e > 1 ? "value[i]" : "value"}`, "float32")}
           }
         }
       }
@@ -665,12 +665,12 @@ var ji = class {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.addBias = t != null, this.hasPreluActivationWeights = r != null, this.activation = n, this.addBias && this.variableNames.push("bias"), this.hasPreluActivationWeights && this.variableNames.push("preluActivationWeights"), this.shaderKey = `biasActivation_${n}`;
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.addBias = t != null, this.hasPreluActivationWeights = r != null, this.activation = n, this.addBias && this.variableNames.push("bias"), this.hasPreluActivationWeights && this.variableNames.push("preluActivationWeights"), this.shaderKey = `biasActivation_${n}`;
 	}
 	getUserCode() {
 		return `
     ${G(this.activation, this.hasPreluActivationWeights)}
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         var value = getXByOutputIndex(index);
@@ -686,11 +686,11 @@ var ji = class {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "fill";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "fill";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         setOutputAtIndex(index, uniforms.value);
       }
@@ -767,13 +767,13 @@ function Ii({ a: e, b: t, transposeA: n, transposeB: r, backend: i, bias: a = nu
 	], ne = A().get("WEBGPU_MATMUL_PROGRAM_TYPE");
 	if (ne < 0) {
 		let e = A().getNumber("WEBGPU_THRESHOLD_TO_INCREASE_WORKGROUPS_FOR_MATMUL"), t = e > 0 ? e : i.thresholdToIncreaseWorkgroups, n = T * Math.ceil(p / 32) * Math.ceil(m / 32);
-		ne = n <= t || p <= 8 && n <= t * 2 ? T * p * m <= 128 ? B.MatMulReduceProgram : T === 1 && f >= 2e3 ? B.MatMulSplitKProgram : B.MatMulSmallOutputSizeProgram : B.MatMulPackedProgram;
+		ne = n <= t || p <= 8 && n <= t * 2 ? T * p * m <= 128 ? I.MatMulReduceProgram : T === 1 && f >= 2e3 ? I.MatMulSplitKProgram : I.MatMulSmallOutputSizeProgram : I.MatMulPackedProgram;
 	}
 	switch (ne) {
-		case B.MatMulReduceProgram:
+		case I.MatMulReduceProgram:
 			D = new ki(te, n, r, a, c, o);
 			break;
-		case B.MatMulSplitKProgram:
+		case I.MatMulSplitKProgram:
 			if (O = q({
 				backend: i,
 				attrs: {
@@ -800,10 +800,10 @@ function Ii({ a: e, b: t, transposeA: n, transposeB: r, backend: i, bias: a = nu
 				return u;
 			}
 			break;
-		case B.MatMulSmallOutputSizeProgram:
+		case I.MatMulSmallOutputSizeProgram:
 			D = new ji(b, x, te, n, r, a, c, o);
 			break;
-		case B.MatMulPackedProgram:
+		case I.MatMulPackedProgram:
 			D = new Di(b, te, n, r, a, c, o, i.adapterInfo.isIntel());
 			break;
 		default: throw Error(`Unsupported MatMulProgramType ${ne}.`);
@@ -852,7 +852,7 @@ var Ri = {
 			128,
 			1,
 			1
-		], this.size = !0, this.outputShape = k(t, n), this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `binaryOpComplex_${e}`, this.op = e;
+		], this.size = !0, this.outputShape = k(t, n), this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `binaryOpComplex_${e}`, this.op = e;
 	}
 	getUserCode() {
 		return `
@@ -861,7 +861,7 @@ var Ri = {
         ${wr(this.op, !1)}
       }
 
-      ${I("index")} {
+      ${z("index")} {
         if(index < uniforms.size) {
           let areal = getARealByOutputIndex(index);
           let aimag = getAImagByOutputIndex(index);
@@ -874,7 +874,7 @@ var Ri = {
 	}
 }, Bi = class {
 	constructor(e, t, n) {
-		if (this.size = !0, this.variableNames = ["A", "B"], this.outputShape = k(t, n), this.dispatchLayout = R(this.outputShape), this.op = e, this.useSharedMemoryWithA = t.length <= 1 && n.length > 1 && t[0] < 128, this.useSharedMemoryWithB = n.length <= 1 && t.length > 1 && n[0] < 128, this.useSharedMemoryWithA || this.useSharedMemoryWithB) this.outputComponent = 1, this.variableComponents = [1, 1], this.lastDimensionSize = this.useSharedMemoryWithB ? n[0] : t[0], this.shaderKey = `binary_${e}_${this.lastDimensionSize}`, this.type = "shared", this.workgroupSize = [
+		if (this.size = !0, this.variableNames = ["A", "B"], this.outputShape = k(t, n), this.dispatchLayout = P(this.outputShape), this.op = e, this.useSharedMemoryWithA = t.length <= 1 && n.length > 1 && t[0] < 128, this.useSharedMemoryWithB = n.length <= 1 && t.length > 1 && n[0] < 128, this.useSharedMemoryWithA || this.useSharedMemoryWithB) this.outputComponent = 1, this.variableComponents = [1, 1], this.lastDimensionSize = this.useSharedMemoryWithB ? n[0] : t[0], this.shaderKey = `binary_${e}_${this.lastDimensionSize}`, this.type = "shared", this.workgroupSize = [
 			256,
 			1,
 			1
@@ -887,7 +887,7 @@ var Ri = {
 				1
 			];
 		}
-		this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.outputComponent,
 			1,
 			1
@@ -906,7 +906,7 @@ var Ri = {
 			e = `
         ${n}
         var<workgroup> sharedBuf : array<f32, ${this.lastDimensionSize}>;
-        ${I("index")} {
+        ${z("index")} {
           // Fill in the shared memory buffer.
           let localIndex = i32(localId.x);
           if(localIndex < ${this.lastDimensionSize}) {
@@ -923,7 +923,7 @@ var Ri = {
         `;
 		} else e = `
        ${n}
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.size) {
            let coords = getCoordsFromIndex(index * ${this.outputComponent});
            let a = ${t}(getAByOutputCoords(coords));
@@ -975,14 +975,14 @@ var Hi = {
 			128,
 			1,
 			1
-		], this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.op = t, n !== "" && (this.uniforms = n), this.shaderKey = `unary_${t}`;
+		], this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.op = t, n !== "" && (this.uniforms = n), this.shaderKey = `unary_${t}`;
 	}
 	getUserCode() {
 		return `
       fn unaryOperation(a : f32) -> f32 {
         ${W(this.op, !1)}
       }
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let a = getAByOutputIndex(index);
           setOutputAtIndex(index, unaryOperation(a));
@@ -1095,7 +1095,7 @@ var { addImpl: Wi, castImpl: Gi, ceilImpl: Ki, concatImpl: qi, equalImpl: Ji, ex
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e[0], this.variableNames = e.map((e, t) => `T${t}`), this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.size = !0, this.outputShape = e[0], this.variableNames = e.map((e, t) => `T${t}`), this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
@@ -1108,7 +1108,7 @@ var { addImpl: Wi, castImpl: Gi, ceilImpl: Ki, concatImpl: qi, equalImpl: Ji, ex
 		});
 		let t = this.variableNames.map((e) => `v${e}`).join(" + ");
 		return `
-      ${I("index")} {
+      ${z("index")} {
         for (var i = 0; i < ${this.workPerThread}; i = i + 1) {
           let flatIndex = index * ${this.workPerThread} + i;
           if (flatIndex < uniforms.size) {
@@ -1148,7 +1148,7 @@ var ja = {
 		this.outputShape = n, this.dispatchLayout = {
 			x: [0],
 			y: [1]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			1,
 			1,
 			1
@@ -1159,7 +1159,7 @@ var ja = {
 		let e = this.workgroupSize[0];
 		return `
       var<workgroup> tile : array<array<f32, ${this.workgroupSize[0] + 1}>, ${this.workgroupSize[0]}>;
-      ${I()} {
+      ${z()} {
         var x = i32(workgroupId.x) * ${e} + i32(localId.x);
         var y = i32(workgroupId.y) * ${e} + i32(localId.y);
         let width = uniforms.outShape[0];
@@ -1187,16 +1187,16 @@ var ja = {
 		], this.size = !0;
 		let n = Array(e.length);
 		for (let r = 0; r < n.length; r++) n[r] = e[t[r]];
-		this.outputShape = n, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		this.outputShape = n, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
 		]), this.newDim = t, this.shaderKey = `transpose_${t}`;
 	}
 	getUserCode() {
-		let e = F(this.outputShape.length), t = Pa(this.newDim);
+		let e = R(this.outputShape.length), t = Pa(this.newDim);
 		return `
-      ${I("index")} {
+      ${z("index")} {
         for(var i = 0; i < ${this.workPerThread}; i = i + 1) {
           let flatIndex = index * ${this.workPerThread} + i;
           if(flatIndex < uniforms.size) {
@@ -1213,7 +1213,7 @@ function Pa(e) {
 	let t = e.length;
 	if (t > 6) throw Error(`Transpose for rank ${t} is not yet supported`);
 	let n = Array(t);
-	for (let t = 0; t < e.length; t++) n[e[t]] = `coords.${hr(t)}`;
+	for (let t = 0; t < e.length; t++) n[e[t]] = `coords.${xr(t)}`;
 	return n.join();
 }
 //#endregion
@@ -1252,7 +1252,7 @@ var Fa = {
 			64,
 			1,
 			1
-		], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, [
+		], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, [
 			1,
 			1,
 			1
@@ -1279,7 +1279,7 @@ var Fa = {
          let offset = ${this.outputShape.length === 1 ? "outputCoords" : "outputCoords[0]"} * uniforms.reduceSize;
           return offset;
        }
-       ${I("index")} {
+       ${z("index")} {
          let outputIndex = index / ${n};
          let offset = getOffset(outputIndex);
          var bestValue = ${t};
@@ -1389,17 +1389,17 @@ var Va = {
 		let r = [t];
 		this.op = n === "min" ? "<" : ">";
 		let [i, a] = wn(e, r);
-		this.outputShape = i.length === 0 ? [1] : i, this.dispatchLayout = R(this.outputShape), N(a) < 32 ? (this.type = "plain", this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize)) : (this.type = "shared", this.dispatch = z(this.dispatchLayout, this.outputShape, [
+		this.outputShape = i.length === 0 ? [1] : i, this.dispatchLayout = P(this.outputShape), N(a) < 32 ? (this.type = "plain", this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize)) : (this.type = "shared", this.dispatch = F(this.dispatchLayout, this.outputShape, [
 			1,
 			1,
 			1
 		])), this.inputShape = e, this.shaderKey = `argMinMax_${this.op}_${this.type}`;
 	}
 	getUserCode() {
-		let e = this.workgroupSize[0], t = () => this.inputShape.length === 1 ? "uniforms.xShape" : `uniforms.xShape.${hr(this.inputShape.length - 1)}`, n = () => {
+		let e = this.workgroupSize[0], t = () => this.inputShape.length === 1 ? "uniforms.xShape" : `uniforms.xShape.${xr(this.inputShape.length - 1)}`, n = () => {
 			let e = "";
 			if (this.outputShape.length === 1) this.inputShape.length !== 1 && (e += "outputCoords,");
-			else for (let t = 0; t < this.outputShape.length; t++) e += `outputCoords.${hr(t)},`;
+			else for (let t = 0; t < this.outputShape.length; t++) e += `outputCoords.${xr(t)},`;
 			return e;
 		};
 		return this.type === "shared" ? `
@@ -1412,7 +1412,7 @@ var Va = {
       var<workgroup> xBestValues : array<f32, ${e}>;
     `}
 
-      ${I("index")} {
+      ${z("index")} {
         let outputIndex = index / ${e};
         let reduceLength = ${t()};
 
@@ -1452,7 +1452,7 @@ var Va = {
         }
       }
     ` : `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let outputCoords = getCoordsFromIndex(index);
           var bestIndex = 0;
@@ -1536,11 +1536,11 @@ var Ka = {
 			256,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "poolWithFilterSizeEqualsOne";
+		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "poolWithFilterSizeEqualsOne";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let batch = coords[0];
@@ -1563,7 +1563,7 @@ var Ka = {
 			1,
 			1
 		], this.size = !0, t === "avg" && n) throw Error("Cannot compute positions for average pool.");
-		this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.poolType = t, this.computePositions = n, this.flattenPositions = r, this.includeBatchIndex = i, this.shaderKey = `pool2D_${t}_${n}_${r}_${i}`;
+		this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.poolType = t, this.computePositions = n, this.flattenPositions = r, this.includeBatchIndex = i, this.shaderKey = `pool2D_${t}_${n}_${r}_${i}`;
 	}
 	getUserCode() {
 		let e;
@@ -1575,7 +1575,7 @@ var Ka = {
       }` : "resultValue = max(value, resultValue);";
 		let t = "resultValue";
 		return this.poolType === "avg" && (t = "resultValue / max(count, 1.0)"), `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
           let batch = coords[0];
@@ -1617,7 +1617,7 @@ var Ka = {
 			1,
 			1
 		], this.size = !0, t === "avg" && n) throw Error("Cannot compute positions for average pool.");
-		this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.poolType = t, this.computePositions = n, this.flattenPositions = r, this.includeBatchIndex = i, this.shaderKey = `pool3D_${t}_${n}_${r}_${i}`;
+		this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.poolType = t, this.computePositions = n, this.flattenPositions = r, this.includeBatchIndex = i, this.shaderKey = `pool3D_${t}_${n}_${r}_${i}`;
 	}
 	getUserCode() {
 		let e;
@@ -1629,7 +1629,7 @@ var Ka = {
       }` : "resultValue = max(value, resultValue);";
 		let t = "resultValue";
 		return this.poolType === "avg" && (t = "resultValue / max(count, 1.0)"), `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let batch = coords.x;
@@ -1812,11 +1812,11 @@ var lo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "avgPool2DBackprop";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "avgPool2DBackprop";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords[0];
@@ -1861,11 +1861,11 @@ var lo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "avgPool3DBackprop";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "avgPool3DBackprop";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords.x;
@@ -1971,7 +1971,7 @@ var mo = {
 //#region node_modules/@tensorflow/tfjs-backend-webgpu/dist/kernels/AvgPoolGrad.js
 function ho(e) {
 	let { inputs: t, backend: n, attrs: r } = e, { dy: i, input: a } = t, o = a;
-	xr([i, a], "avgPoolGrad");
+	_r([i, a], "avgPoolGrad");
 	let { filterSize: s, strides: c, pad: l } = r, u = S(o.shape, s, c, 1, l), d = new uo(u), f = 1 / (u.filterHeight * u.filterWidth), p = [
 		{
 			type: "int32",
@@ -2031,16 +2031,16 @@ var vo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t, this.rank = t.length, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.size = !0, this.outputShape = t, this.rank = t.length, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
-		]), this.start = e, this.uniforms = `start : ${F(e.length)}, `, this.shaderKey = "slice";
+		]), this.start = e, this.uniforms = `start : ${R(e.length)}, `, this.shaderKey = "slice";
 	}
 	getUserCode() {
-		let e = F(this.rank), t = xo(this.rank), n;
-		return n = this.start.length === 1 ? this.outputShape.map((e, t) => "sourceLoc = uniforms.start + coords;") : this.outputShape.map((e, t) => `sourceLoc.${bo[t]} = uniforms.start.${hr(t)} + coords.${bo[t]};`), `
-      ${I("index")} {
+		let e = R(this.rank), t = xo(this.rank), n;
+		return n = this.start.length === 1 ? this.outputShape.map((e, t) => "sourceLoc = uniforms.start + coords;") : this.outputShape.map((e, t) => `sourceLoc.${bo[t]} = uniforms.start.${xr(t)} + coords.${bo[t]};`), `
+      ${z("index")} {
         if (index < uniforms.size) {
           var sourceLoc : ${e};
           let coords = getCoordsFromIndex(index);
@@ -2112,7 +2112,7 @@ var Co = {
 	}
 }, To = `
   fn bincount_write(index: i32, value: f32) {
-    ${P("&result[index]", "value", "float32")}
+    ${L("&result[index]", "value", "float32")}
   }
 `, Eo = "\n  fn bincount_write(index: i32, value: f32) {\n    atomicStore(&result[index], bitcast<i32>(value));\n  }\n", Do = class {
 	constructor(e, t, n = !1) {
@@ -2120,12 +2120,12 @@ var Co = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.hasWeights = !0, this.binaryOutput = !1, this.outputShape = e, this.rank = e.length, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.binaryOutput = n, n && (this.atomic = !1), this.hasWeights = t, this.hasWeights && this.variableNames.push("w"), this.shaderKey = `bincount_${this.hasWeights}_${this.binaryOutput}_${this.rank}`;
+		], this.atomic = !0, this.hasWeights = !0, this.binaryOutput = !1, this.outputShape = e, this.rank = e.length, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.binaryOutput = n, n && (this.atomic = !1), this.hasWeights = t, this.hasWeights && this.variableNames.push("w"), this.shaderKey = `bincount_${this.hasWeights}_${this.binaryOutput}_${this.rank}`;
 	}
 	getUserCode() {
 		return `
     ${this.binaryOutput ? Eo : To}
-  ${I("index")} {
+  ${z("index")} {
     ${this.rank === 1 ? `if (index < uniforms.xShape) {
       let indexVal = i32(getX(index));
       if (indexVal < uniforms.binCountSize) {
@@ -2170,11 +2170,11 @@ var ko = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = [e], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "broadcastArgs";
+		], this.size = !0, this.outputShape = [e], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "broadcastArgs";
 	}
 	getUserCode() {
 		return `
-  ${I("index")} {
+  ${z("index")} {
     if (index < uniforms.size) {
       var s0 = 1.0;
       var s1 = 1.0;
@@ -2333,7 +2333,7 @@ var zo = {
 			64,
 			1,
 			1
-		], this.outputComponent = 4, this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.outputComponent = 4, this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
@@ -2341,7 +2341,7 @@ var zo = {
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if(index < uniforms.size) {
           let value = getAByOutputIndex(index);
           var clampedValue = clamp(
@@ -2358,11 +2358,11 @@ var zo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "clip";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "clip";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if(index < uniforms.size) {
           let value = getAByOutputIndex(index);
           if (isnan(value)) {
@@ -2397,11 +2397,11 @@ var Wo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "complexAbs";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "complexAbs";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let re = abs(getRealByOutputIndex(index));
         let im = abs(getImagByOutputIndex(index));
@@ -2438,7 +2438,7 @@ var Jo = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = dr(e, 1), this.variableNames = e.map((e, t) => `T${t}`), this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.size = !0, this.outputShape = dr(e, 1), this.variableNames = e.map((e, t) => `T${t}`), this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
@@ -2455,7 +2455,7 @@ var Jo = {
 			e.push(`else { setOutputAtCoords(coords.x, coords.y, getT${t}(yR, yC - uniforms.offset${n})); }`);
 		} else e.push("setOutputAtCoords(coords.x, coords.y, getT0(yR, yC));");
 		return `
-      ${I("index")} {
+      ${z("index")} {
         for(var i = 0; i < ${this.workPerThread}; i = i + 1) {
           let flatIndex = index * ${this.workPerThread} + i;
           if(flatIndex < uniforms.size) {
@@ -2606,7 +2606,7 @@ function ns(e, t, n, r, i = !1, a = null, o = !1, s = 4, c = 4, l = 4) {
       let xRow = outRow * uniforms.strides[0] + uniforms.dilations[0] * WRow - uniforms.pads[0];
       let xCol = outCol * uniforms.strides[1] + uniforms.dilations[1] * WCol - uniforms.pads[1];
       let xCh = ${_} % inChannels;
-      var resData = ${L(s)}(0.0);
+      var resData = ${B(s)}(0.0);
       // The bounds checking is always needed since we use it to pad zero for
       // the 'same' padding type.
       if (xRow >= 0 && xRow < ${m} && xCol >= 0 && xCol < ${h}) {
@@ -2619,12 +2619,12 @@ function ns(e, t, n, r, i = !1, a = null, o = !1, s = 4, c = 4, l = 4) {
       if (row < uniforms.dimAOuter && col < uniforms.dimInner) {
         ${v}
       }
-      return ${L(s)}(0.0);` : r && n ? `
+      return ${B(s)}(0.0);` : r && n ? `
       ${v}` : `
       if (row < uniforms.dimInner && col < uniforms.dimBOuter) {
         ${v}
       }
-      return ${L(s)}(0.0);`, b = `${d(c)}`, x = L(l), S = L(e ? s : c), C = L(e ? c : s);
+      return ${B(s)}(0.0);`, b = `${d(c)}`, x = B(l), S = B(e ? s : c), C = B(e ? c : s);
 	return `
       ${G(a, o, l === 4, 4)}
       fn mm_readA(batch: i32, row : i32, col : i32) -> ${S} {
@@ -2656,7 +2656,7 @@ var rs = class {
 			x: [2, 3],
 			y: [1],
 			z: [0]
-		}, this.workgroupSize = Sr(this.dispatchLayout, this.outputShape, this.isVec4), this.elementsPerThread = vr(this.dispatchLayout, this.outputShape, this.isVec4), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread), this.isVec4 ? (this.outputComponent = 4, this.isChannelsLast && e.inChannels % 4 != 0 ? (this.innerElementSize = 3, this.variableComponents = [1, 4]) : (this.innerElementSize = 4, this.variableComponents = [4, 4]), i && (this.variableNames.push("bias"), this.variableComponents.push(4)), o && (this.variableNames.push("preluActivationWeights"), this.variableComponents.push(4))) : (this.innerElementSize = this.elementsPerThread[0], i && this.variableNames.push("bias"), o && this.variableNames.push("preluActivationWeights")), this.sequentialAccessByThreads = s, this.addBias = i, this.activation = a, this.hasPreluActivationWeights = o, this.tileAOuter = this.workgroupSize[1] * this.elementsPerThread[1], this.tileBOuter = this.workgroupSize[0] * this.elementsPerThread[0], this.tileInner = Math.max(this.workgroupSize[0] * this.innerElementSize, this.workgroupSize[1]), this.fitAOuter = t % this.tileAOuter === 0, this.fitBOuter = n % this.tileBOuter === 0, this.fitInner = r % this.tileInner === 0, this.shaderKey = `conv2DMM_${this.elementsPerThread}_${this.activation}}_${this.fitAOuter}_${this.fitBOuter}_${this.fitInner}_${this.isVec4}_${this.innerElementSize}_${this.isChannelsLast}_${this.sequentialAccessByThreads}`;
+		}, this.workgroupSize = vr(this.dispatchLayout, this.outputShape, this.isVec4), this.elementsPerThread = mr(this.dispatchLayout, this.outputShape, this.isVec4), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread), this.isVec4 ? (this.outputComponent = 4, this.isChannelsLast && e.inChannels % 4 != 0 ? (this.innerElementSize = 3, this.variableComponents = [1, 4]) : (this.innerElementSize = 4, this.variableComponents = [4, 4]), i && (this.variableNames.push("bias"), this.variableComponents.push(4)), o && (this.variableNames.push("preluActivationWeights"), this.variableComponents.push(4))) : (this.innerElementSize = this.elementsPerThread[0], i && this.variableNames.push("bias"), o && this.variableNames.push("preluActivationWeights")), this.sequentialAccessByThreads = s, this.addBias = i, this.activation = a, this.hasPreluActivationWeights = o, this.tileAOuter = this.workgroupSize[1] * this.elementsPerThread[1], this.tileBOuter = this.workgroupSize[0] * this.elementsPerThread[0], this.tileInner = Math.max(this.workgroupSize[0] * this.innerElementSize, this.workgroupSize[1]), this.fitAOuter = t % this.tileAOuter === 0, this.fitBOuter = n % this.tileBOuter === 0, this.fitInner = r % this.tileInner === 0, this.shaderKey = `conv2DMM_${this.elementsPerThread}_${this.activation}}_${this.fitAOuter}_${this.fitBOuter}_${this.fitInner}_${this.isVec4}_${this.innerElementSize}_${this.isChannelsLast}_${this.sequentialAccessByThreads}`;
 	}
 	getUserCode() {
 		let e = this.isVec4 ? xi(this.elementsPerThread, this.workgroupSize, !this.isChannelsLast, this.tileInner) : wi(this.elementsPerThread, this.workgroupSize, !this.isChannelsLast, this.tileInner, !1, null, this.sequentialAccessByThreads), t = this.isVec4 ? [
@@ -2687,7 +2687,7 @@ var rs = class {
 			x: [3],
 			y: [2],
 			z: [0, 1]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.addBias = t, this.activation = n, this.hasPreluActivationWeights = r, t && this.variableNames.push("bias"), r && this.variableNames.push("preluActivationWeights"), this.shaderKey = `conv2dnaive_${this.activation}_${this.isChannelsLast}`;
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.addBias = t, this.activation = n, this.hasPreluActivationWeights = r, t && this.variableNames.push("bias"), r && this.variableNames.push("preluActivationWeights"), this.shaderKey = `conv2dnaive_${this.activation}_${this.isChannelsLast}`;
 	}
 	getUserCode() {
 		return `
@@ -2716,7 +2716,7 @@ var rs = class {
            setOutputAtCoords(coords.x, coords.y, coords.z, coords.w, value);
          }
        }
-       ${I("index")} {
+       ${z("index")} {
          let coords = getOutputCoords();
          let batch = coords[0];
          let outChannel = ${this.isChannelsLast ? "coords[3];" : "coords[1];"}
@@ -2744,12 +2744,12 @@ var rs = class {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = t, this.shaderKey = `im2col_${this.isChannelsLast}`;
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = t, this.shaderKey = `im2col_${this.isChannelsLast}`;
 	}
 	getUserCode() {
 		let e = this.isChannelsLast ? 1 : 2, t = this.isChannelsLast ? 2 : 3, n = this.isChannelsLast ? "coords[1]" : "coords[2]", r = this.isChannelsLast ? "coords[2]" : "coords[1]", i = this.isChannelsLast ? "getX(batch, xRow, xCol, ch)" : "getX(batch, ch, xRow, xCol)";
 		return `
-    ${I("index")} {
+    ${z("index")} {
       let coords = getCoordsFromIndex(index);
       if(index < uniforms.size) {
         let batch = coords[0];
@@ -3054,7 +3054,7 @@ var ds = {
 			x: [3],
 			y: [2],
 			z: [0, 1]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			4,
 			this.workPerThread,
 			1
@@ -3062,11 +3062,11 @@ var ds = {
 			64,
 			1,
 			1
-		], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize)), this.shaderKey = `conv2DDerInput_${this.isChannelsLast}_${this.isVec4}_${this.workPerThread}`;
+		], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize)), this.shaderKey = `conv2DDerInput_${this.isChannelsLast}_${this.isVec4}_${this.workPerThread}`;
 	}
 	getUserCode() {
 		let e = this.isChannelsLast ? 1 : 2, t = this.isChannelsLast ? 2 : 3, n = this.isChannelsLast ? 3 : 1, r = `
-    ${I()} {
+    ${z()} {
       let batch = i32(globalId.z) / uniforms.outShape[1];
       let r = i32(globalId.z) % uniforms.outShape[1];
       let c = i32(globalId.y) * ${this.workPerThread};
@@ -3168,7 +3168,7 @@ var ds = {
 		return this.isVec4 ? `
     ${r}
     ` : `
-    ${I("index")} {
+    ${z("index")} {
       if(index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords[0];
@@ -3217,11 +3217,11 @@ var ds = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = e.dataFormat === "channelsLast", this.shaderKey = `conv2DDerFilter_${this.isChannelsLast}`;
+		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = e.dataFormat === "channelsLast", this.shaderKey = `conv2DDerFilter_${this.isChannelsLast}`;
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if(index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let wR = coords[0];
@@ -3269,11 +3269,11 @@ var ds = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3DDerFilter";
+		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3DDerFilter";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if(index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let wF = coords.x;
@@ -3320,11 +3320,11 @@ var ds = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3DDerInput";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3DDerInput";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if(index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords.x;
@@ -3438,10 +3438,10 @@ function vs(e = 4) {
       let xR = f32(outRow - uniforms.pads[0] + WRow) / f32(uniforms.strides[0]);
       let xC = f32(outCol - uniforms.pads[1] + WCol) / f32(uniforms.strides[1]);
       if (xR < 0.0 || xR >= f32(uniforms.outBackprop[1]) || fract(xR) > 0.0) {
-        return ${L(e)}(0.0);
+        return ${B(e)}(0.0);
       }
       if (xC < 0.0 || xC >= f32(uniforms.outBackprop[2]) || fract(xC) > 0.0) {
-        return ${L(e)}(0.0);
+        return ${B(e)}(0.0);
       }
       let coord = vec4<i32>(
           batch,
@@ -3450,13 +3450,13 @@ function vs(e = 4) {
           col % uniforms.outBackprop[3]);
       return x[getIndexFromCoords4D(coord, uniforms.xShape)/${e}];`}
       }
-      return ${L(e)}(0.0);`;
+      return ${B(e)}(0.0);`;
 	return `
-  fn mm_readA(batch: i32, row : i32, col : i32) -> ${L(e)} {
+  fn mm_readA(batch: i32, row : i32, col : i32) -> ${B(e)} {
     ${n}
   }
 
-  fn mm_readB(batch: i32, row : i32, col : i32) -> ${L(e)} {
+  fn mm_readB(batch: i32, row : i32, col : i32) -> ${B(e)} {
     let coordX = uniforms.filterDims.x - 1 -
         row / (uniforms.filterDims[1] * uniforms.outBackprop[3]);
     let coordY = uniforms.filterDims.y - 1 -
@@ -3467,10 +3467,10 @@ function vs(e = 4) {
       let coord = vec4<i32>(coordX, coordY, col, rowInner);
       ${t(e)}
     }
-    return ${L(e)}(0.0);
+    return ${B(e)}(0.0);
   }
 
-  fn mm_write(batch: i32, row : i32, col : i32, valueInput : ${L(e)}) {
+  fn mm_write(batch: i32, row : i32, col : i32, valueInput : ${B(e)}) {
     if (row < uniforms.dimAOuter && col < uniforms.dimBOuter) {
       var value = valueInput;
       let outCoord = vec4<i32>(
@@ -3488,7 +3488,7 @@ var ys = class {
 			x: [3],
 			y: [1, 2],
 			z: [0]
-		}, this.workgroupSize = Sr(this.dispatchLayout, this.outputShape, this.isVec4), this.elementsPerThread = vr(this.dispatchLayout, this.outputShape, this.isVec4), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread), this.isVec4 && (this.outputComponent = 4, this.variableComponents = [4, 1]), this.shaderKey = `conv2DDerInputMM_${this.isVec4}_${this.elementsPerThread}`;
+		}, this.workgroupSize = vr(this.dispatchLayout, this.outputShape, this.isVec4), this.elementsPerThread = mr(this.dispatchLayout, this.outputShape, this.isVec4), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, this.elementsPerThread), this.isVec4 && (this.outputComponent = 4, this.variableComponents = [4, 1]), this.shaderKey = `conv2DDerInputMM_${this.isVec4}_${this.elementsPerThread}`;
 	}
 	getUserCode() {
 		let e = this.isVec4 ? xi(this.elementsPerThread, this.workgroupSize) : wi(this.elementsPerThread, this.workgroupSize);
@@ -3551,11 +3551,11 @@ var xs = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3dnaive";
+		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "conv3dnaive";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getOutputCoords();
         let batch = coords.x;
@@ -3815,7 +3815,7 @@ var Os = {
 			n[0],
 			n[1],
 			e
-		], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.methodId = +(r === "bilinear"), this.cropHeightBiggerThan1 = this.outputShape[1] > 1, this.cropWidthBiggerThan1 = this.outputShape[2] > 1, this.shaderKey = `cropAndResize_${this.methodId}_${this.cropHeightBiggerThan1}_${this.cropWidthBiggerThan1}`;
+		], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.methodId = +(r === "bilinear"), this.cropHeightBiggerThan1 = this.outputShape[1] > 1, this.cropWidthBiggerThan1 = this.outputShape[2] > 1, this.shaderKey = `cropAndResize_${this.methodId}_${this.cropHeightBiggerThan1}_${this.cropWidthBiggerThan1}`;
 	}
 	getUserCode() {
 		let [e, t] = ["f32(uniforms.imageShape[1] - 1)", "f32(uniforms.imageShape[2] - 1)"], [n, r, i] = this.cropHeightBiggerThan1 ? [
@@ -3836,7 +3836,7 @@ var Os = {
 			`0.5 * (x1+x2) * ${t}`
 		];
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let height_ratio = f32(${n});
@@ -3917,12 +3917,12 @@ var Ps = class {
 			128,
 			1,
 			1
-		], this.outputShape = t, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.exclusive = n, this.reverse = r, this.op = e, this.shaderKey = `cum_${this.op}_${this.exclusive}_${this.reverse}`;
+		], this.outputShape = t, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.exclusive = n, this.reverse = r, this.op = e, this.shaderKey = `cum_${this.op}_${this.exclusive}_${this.reverse}`;
 	}
 	getUserCode() {
 		let e = this.outputShape.length, t = this.op === Ns.Prod ? "1.0" : "0.0", n = this.exclusive ? t : `getX(${Fs(e, "coords", this.op)})`, r = this.outputShape[this.outputShape.length - 1], i = "", a = "";
 		return this.exclusive ? (i = this.reverse ? `end != ${r - 1}` : "end != 0", a = this.reverse ? "end + 1" : "end - 1") : (i = this.reverse ? `end + pow2 < ${r}` : "end >= pow2", a = this.reverse ? "end + pow2" : "end - pow2"), `
-      ${I("index")} {
+      ${z("index")} {
        if (index < uniforms.size) {
          var coords = getCoordsFromIndex(index);
 
@@ -4041,11 +4041,11 @@ var Us = {
 			64,
 			1,
 			1
-		], this.size = !0, this.uniforms = "blockSize : i32,", this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `depthToSpace_${t}`, this.dataFormat = t;
+		], this.size = !0, this.uniforms = "blockSize : i32,", this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `depthToSpace_${t}`, this.dataFormat = t;
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let b = coords[0];
@@ -4115,7 +4115,7 @@ var Ks = {
 			x: [3],
 			y: [2],
 			z: [0, 1]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), r && this.variableNames.push("bias"), a && this.variableNames.push("preluActivationWeights"), this.addBias = r, this.activation = i, this.hasPreluActivation = a, this.filterHeight = t, this.filterWidth = n, this.shaderKey = `depthwiseNCHW_${this.activation}_${this.filterHeight}_${this.filterWidth}`;
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), r && this.variableNames.push("bias"), a && this.variableNames.push("preluActivationWeights"), this.addBias = r, this.activation = i, this.hasPreluActivation = a, this.filterHeight = t, this.filterWidth = n, this.shaderKey = `depthwiseNCHW_${this.activation}_${this.filterHeight}_${this.filterWidth}`;
 	}
 	getUserCode() {
 		let e = this.filterWidth * this.filterHeight, t = this.workgroupSize[0] * this.workgroupSize[1] * this.workgroupSize[2], n = this.workgroupSize[1] + this.filterHeight - 1, r = this.workgroupSize[0] + this.filterWidth - 1;
@@ -4133,7 +4133,7 @@ var Ks = {
         return value;
       }
 
-      ${I()} {
+      ${z()} {
         let coords = getOutputCoords();
         let batch = coords[0];
         let xRCCorner = vec2<i32>(coords.zw) - uniforms.pads;
@@ -4196,7 +4196,7 @@ var Ks = {
 			this.virtualWidth,
 			this.outputShape[3]
 		];
-		this.dispatchLayout = R(i), this.dispatch = z(this.dispatchLayout, i, this.workgroupSize, [
+		this.dispatchLayout = P(i), this.dispatch = F(this.dispatchLayout, i, this.workgroupSize, [
 			this.outputComponent * this.workPerThread,
 			1,
 			1
@@ -4214,7 +4214,7 @@ var Ks = {
         return value;
       }
 
-      ${I("index")} {
+      ${z("index")} {
         let width0 = uniforms.outShape[3] / ${this.outputComponent};
         let d1 = (index % width0) * ${this.outputComponent};
         var index1 = index / width0;
@@ -4267,14 +4267,14 @@ var Ks = {
 			256,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = e.dataFormat === "channelsLast", t && this.variableNames.push("bias"), r && this.variableNames.push("preluActivationWeights"), this.convInfo = e, this.addBias = t, this.activation = n, this.hasPreluActivation = r, this.shaderKey = `depthwise_${this.activation}_${this.isChannelsLast}`;
+		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.isChannelsLast = e.dataFormat === "channelsLast", t && this.variableNames.push("bias"), r && this.variableNames.push("preluActivationWeights"), this.convInfo = e, this.addBias = t, this.activation = n, this.hasPreluActivation = r, this.shaderKey = `depthwise_${this.activation}_${this.isChannelsLast}`;
 	}
 	getUserCode() {
 		let e = this.isChannelsLast ? "getX(batch, xR, xC, d1);" : "getX(batch, d1, xR, xC);";
 		return `
       ${G(this.activation, this.hasPreluActivation, !1, 4)}
 
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getOutputCoords();
           let batch = coords[0];
@@ -4379,11 +4379,11 @@ var Zs = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "depthwise_conv2d_backprop_filter";
+		], this.size = !0, this.outputShape = e.filterShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "depthwise_conv2d_backprop_filter";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let wR = coords[0];
@@ -4425,11 +4425,11 @@ var Zs = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "depthwise_conv2d_backprop_input";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "depthwise_conv2d_backprop_input";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords[0];
@@ -4562,11 +4562,11 @@ var rc = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = [e, e], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "diag";
+		], this.size = !0, this.outputShape = [e, e], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "diag";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getOutputCoords();
           let value = select(0.0, getX(coords[0]), coords[0] == coords[1]);
@@ -4600,11 +4600,11 @@ var oc = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "dilation2d";
+		], this.size = !0, this.outputShape = e.outShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "dilation2d";
 	}
 	getUserCode() {
 		return `
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.size) {
            let neg_infinity = -3.4e38;
            let coords = getOutputCoords();
@@ -4675,13 +4675,13 @@ var lc = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = e.inShape, this.dispatchLayout = R(e.outShape), this.dispatch = z(this.dispatchLayout, e.outShape, this.workgroupSize), t !== "float32" && t !== "int32") throw Error(`Dilation2DBackpropInput only supports float32 and int32
+		], this.atomic = !0, this.outputShape = e.inShape, this.dispatchLayout = P(e.outShape), this.dispatch = F(this.dispatchLayout, e.outShape, this.workgroupSize), t !== "float32" && t !== "int32") throw Error(`Dilation2DBackpropInput only supports float32 and int32
           types, does not support ${t} type.`);
 		this.type = t, this.shaderKey = "dilation2DBackpropInput";
 	}
 	getUserCode() {
 		return `
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.dySize) {
            let coords = getDyCoordsFromIndex(index);
            let b = coords[0];
@@ -4720,7 +4720,7 @@ var lc = {
            let flatIndexIn = d + uniforms.xShape[3] *
                (xCMax + uniforms.xShape[2] * (xRMax + uniforms.xShape[1] * b));
            let value = getDy(b, r, c, d);
-           ${P("&result[flatIndexIn]", "value", this.type)}
+           ${L("&result[flatIndexIn]", "value", this.type)}
          }
        }
      `;
@@ -4735,13 +4735,13 @@ var lc = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = e.filterShape, this.dispatchLayout = R(e.outShape), this.dispatch = z(this.dispatchLayout, e.outShape, this.workgroupSize), n !== "float32" && n !== "int32") throw Error(`Dilation2DBackpropFilter only supports float32 and int32
+		], this.atomic = !0, this.outputShape = e.filterShape, this.dispatchLayout = P(e.outShape), this.dispatch = F(this.dispatchLayout, e.outShape, this.workgroupSize), n !== "float32" && n !== "int32") throw Error(`Dilation2DBackpropFilter only supports float32 and int32
           types, does not support ${n} type.`);
 		this.type = n, this.shaderKey = "dilation2DBackpropFilter";
 	}
 	getUserCode() {
 		return `
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.dySize) {
            let coords = getDyCoordsFromIndex(index);
            let b = coords[0];
@@ -4779,7 +4779,7 @@ var lc = {
 
            let flatIndexIn = d + uniforms.wShape[2] * (wCMax + wRMax * uniforms.wShape[1]);
            let value = getDy(b, r, c, d);
-           ${P("&result[flatIndexIn]", "value", this.type)}
+           ${L("&result[flatIndexIn]", "value", this.type)}
          }
        }
      `;
@@ -4876,7 +4876,7 @@ var hc = {
 			64,
 			1,
 			1
-		], this.pixelsOpType = _r.DRAW, this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.type = t, this.textureFormat = n, this.shaderKey = `draw_${t}_${n}`;
+		], this.pixelsOpType = Cr.DRAW, this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.type = t, this.textureFormat = n, this.shaderKey = `draw_${t}_${n}`;
 	}
 	getUserCode() {
 		let e, t = this.type === "float32" ? "value" : "value / 255.0";
@@ -4889,7 +4889,7 @@ var hc = {
         rgba[d] = ${t};
       }`, `
        @group(0) @binding(0) var outImage : texture_storage_2d<${this.textureFormat}, write>;
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.size) {
            var rgba = vec4<f32>(0.0, 0.0, 0.0, uniforms.alpha);
            for (var d = 0; d < uniforms.numChannels; d = d + 1) {
@@ -5063,7 +5063,7 @@ var jc = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.component = e, this.shaderKey = `fft_${e}`;
+		], this.size = !0, this.outputShape = t, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.component = e, this.shaderKey = `fft_${e}`;
 	}
 	getUserCode() {
 		return `
@@ -5093,7 +5093,7 @@ var jc = {
       return result;
     }
 
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getOutputCoords();
         setOutputAtIndex(index, mulMatDFT(coords[0], coords[1]));
@@ -5160,11 +5160,11 @@ var Ic = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "flipLeftRight";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "flipLeftRight";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let coordX = uniforms.xShape[2] - coords[2] - 1;
@@ -5198,11 +5198,11 @@ var Ic = {
 	})
 }, Vc = class {
 	constructor(e, t, n = !1) {
-		this.pixelsOpType = _r.FROM_PIXELS, this.outputShape = [0], this.variableNames = [], this.workgroupSize = [
+		this.pixelsOpType = Cr.FROM_PIXELS, this.outputShape = [0], this.variableNames = [], this.workgroupSize = [
 			256,
 			1,
 			1
-		], this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			t,
 			1,
 			1
@@ -5212,7 +5212,7 @@ var Ic = {
 		let e = this.importVideo ? "textureLoad(src, vec2<i32>(coords.yx));" : "textureLoad(src, vec2<i32>(coords.yx), 0)";
 		return `
       @binding(1) @group(0) var src: ${this.importVideo ? "texture_external" : "texture_2d<f32>"};
-      ${I("index")} {
+      ${z("index")} {
         let flatIndex = index * uniforms.numChannels;
         if (flatIndex < uniforms.size) {
           let coords = getCoordsFromIndex(flatIndex);
@@ -5287,14 +5287,14 @@ var Kc = class {
 			"x",
 			"mean",
 			"variance"
-		], k(e, t), k(e, n), this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), r != null && (k(e, r), this.variableNames.push("offset")), i != null && (k(e, i), this.variableNames.push("scale")), this.offsetShape = r, this.scaleShape = i, this.shaderKey = "batchNorm";
+		], k(e, t), k(e, n), this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), r != null && (k(e, r), this.variableNames.push("offset")), i != null && (k(e, i), this.variableNames.push("scale")), this.offsetShape = r, this.scaleShape = i, this.shaderKey = "batchNorm";
 	}
 	getUserCode() {
 		let e = "0.0";
 		this.offsetShape != null && (e = "getOffsetByOutputIndex(index)");
 		let t = "1.0";
 		return this.scaleShape != null && (t = "getScaleByOutputIndex(index)"), `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size)
         {
           let xValue = getXByOutputIndex(index);
@@ -5391,12 +5391,12 @@ var Zc = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `gathernd_${e}`, this.sliceDim = e, this.uniforms = `sliceDim : i32, strides : ${F(e)},`;
+		], this.size = !0, this.outputShape = t, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = `gathernd_${e}`, this.sliceDim = e, this.uniforms = `sliceDim : i32, strides : ${R(e)},`;
 	}
 	getUserCode() {
 		let e;
 		return e = this.sliceDim > 1 ? "uniforms.strides[j]" : "uniforms.strides", `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           var flattenIndex = 0;
@@ -5451,12 +5451,12 @@ var el = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.slice(), this.aShape = e, this.outputShape = t, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "gather";
+		], this.size = !0, this.outputShape = e.slice(), this.aShape = e, this.outputShape = t, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "gather";
 	}
 	getUserCode() {
 		let e = nl(this.aShape);
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let resRC = getCoordsFromIndex(index);
           let indexZ = i32(getIndices(resRC.x, resRC.z));
@@ -5602,11 +5602,11 @@ var pl = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = [e], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "linSpace";
+		], this.size = !0, this.outputShape = [e], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "linSpace";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           setOutputAtIndex(index, uniforms.start + f32(index) * uniforms.step);
         }
@@ -5662,11 +5662,11 @@ var vl = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "lrn";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "lrn";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getOutputCoords();
         let b = coords[0];
@@ -5700,7 +5700,7 @@ var vl = {
 			x: [3],
 			y: [2],
 			z: [0, 1]
-		}, this.dispatch = z(this.dispatchLayout, this.outputShape, [
+		}, this.dispatch = F(this.dispatchLayout, this.outputShape, [
 			this.elementsPerWorkgroup,
 			this.workgroupSize[1],
 			this.workgroupSize[2]
@@ -5712,7 +5712,7 @@ var vl = {
     const elementsPerWorkgroup = ${this.elementsPerWorkgroup};
     const maxAllowRadius = ${this.maxAllowRadius};
 
-    ${I()} {
+    ${z()} {
       let localDepth = i32(localId.x);
       let workgroupDepth = i32(workgroupId.x) * elementsPerWorkgroup;
       let xDepth = workgroupDepth + localDepth - maxAllowRadius;
@@ -5781,11 +5781,11 @@ var Ol = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "lrn_grad";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "lrn_grad";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getOutputCoords();
         let b = coords[0];
@@ -5940,11 +5940,11 @@ var Il = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "maxPool2DBackprop";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "maxPool2DBackprop";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords[0];
@@ -5995,11 +5995,11 @@ var Il = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "maxPool3DBackprop";
+		], this.size = !0, this.outputShape = e.inShape, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "maxPool3DBackprop";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
       if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
         let batch = coords.x;
@@ -6149,7 +6149,7 @@ var Bl = {
 //#region node_modules/@tensorflow/tfjs-backend-webgpu/dist/kernels/MaxPoolGrad.js
 function Vl(e) {
 	let { inputs: t, backend: n, attrs: r } = e, { dy: i, input: a, output: o } = t, s = a;
-	xr([a, o], "maxPoolGrad");
+	_r([a, o], "maxPoolGrad");
 	let { filterSize: c, strides: l, pad: u, dimRoundingMode: d } = r, f = S(s.shape, c, l, 1, u, d), p = new $a(f, "max", !0), m = [
 		{
 			type: "int32",
@@ -6265,19 +6265,19 @@ var Kl = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t.map((t, n) => t[0] + e[n] + t[1]), this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.xShape = e, t.map((e, t) => {
+		], this.size = !0, this.outputShape = t.map((t, n) => t[0] + e[n] + t[1]), this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.xShape = e, t.map((e, t) => {
 			this.uniforms += ` pad${t} : vec2<i32>,`;
 		}), this.offset = n === "reflect" ? 0 : 1, this.shaderKey = `mirrorPad_${n}`;
 	}
 	getUserCode() {
-		let e = this.xShape.length, t = this.xShape.map((e, t) => `uniforms.pad${t}[0]`).join(","), n = this.xShape.map((t, n) => `uniforms.pad${n}[0] + uniforms.xShape${e > 1 ? `[${n}]` : ""}`).join(","), r = e === 1 ? "start" : "start[i]", i = e === 1 ? "end" : "end[i]", a = e === 1 ? "outC" : "outC[i]", o = F(e), s = e > 1 ? [
+		let e = this.xShape.length, t = this.xShape.map((e, t) => `uniforms.pad${t}[0]`).join(","), n = this.xShape.map((t, n) => `uniforms.pad${n}[0] + uniforms.xShape${e > 1 ? `[${n}]` : ""}`).join(","), r = e === 1 ? "start" : "start[i]", i = e === 1 ? "end" : "end[i]", a = e === 1 ? "outC" : "outC[i]", o = R(e), s = e > 1 ? [
 			"coords[0]",
 			"coords[1]",
 			"coords[2]",
 			"coords[3]"
 		].slice(0, e) : "coords";
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let start = ${o}(${t});
           let end = ${o}(${n});
@@ -6315,7 +6315,7 @@ var Kl = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = [e, t], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "multinomial";
+		], this.size = !0, this.outputShape = [e, t], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "multinomial";
 	}
 	getUserCode() {
 		return `
@@ -6329,7 +6329,7 @@ var Kl = {
       return fract((p3.x + p3.y) * p3.z);
     }
 
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let coords = getOutputCoords();
         let batch = coords[0];
@@ -6355,7 +6355,7 @@ var Kl = {
 	}
 }, Ql = class {
 	constructor(e) {
-		this.variableNames = ["logits"], this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = [
+		this.variableNames = ["logits"], this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = [
 			this.outputShape[0],
 			1,
 			1
@@ -6375,7 +6375,7 @@ var Kl = {
     var<workgroup> rowMaxShared : f32;
     var<workgroup> rowSumShared : f32;
     const blockSize = ${this.workgroupSize[0]};
-    ${I("index")} {
+    ${z("index")} {
       let row = index / blockSize;
       let tid = i32(localId.x);
       let cols = uniforms.outShape[1];
@@ -6517,11 +6517,11 @@ var cu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = [e, t], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "onehot";
+		], this.size = !0, this.outputShape = [e, t], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "onehot";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if(index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           setOutputAtIndex(index, mix(uniforms.offValue, uniforms.onValue,
@@ -6672,7 +6672,7 @@ var _u = {
 //#endregion
 //#region node_modules/@tensorflow/tfjs-backend-webgpu/dist/pad_webgpu.js
 function vu(e, t = !1) {
-	let n = e.length, r = F(n), i = e.map((e, t) => `uniforms.pad${t}[0]`).join(","), a = e.map((e, t) => `uniforms.pad${t}[0] + uniforms.xShape${n > 1 ? `[${t}]` : ""}`).join(","), o = n > 1 ? `${r}(${i})` : `${i}`, s = n > 1 ? `${r}(${a})` : `${a}`, c = n > 1 ? "any(paddedCoords < start)" : "paddedCoords < start", l = n > 1 ? "any(paddedCoords >= end)" : "paddedCoords >= end", u = n > 1 ? [
+	let n = e.length, r = R(n), i = e.map((e, t) => `uniforms.pad${t}[0]`).join(","), a = e.map((e, t) => `uniforms.pad${t}[0] + uniforms.xShape${n > 1 ? `[${t}]` : ""}`).join(","), o = n > 1 ? `${r}(${i})` : `${i}`, s = n > 1 ? `${r}(${a})` : `${a}`, c = n > 1 ? "any(paddedCoords < start)" : "paddedCoords < start", l = n > 1 ? "any(paddedCoords >= end)" : "paddedCoords >= end", u = n > 1 ? [
 		"coords[0]",
 		"coords[1]",
 		"coords[2]",
@@ -6695,13 +6695,13 @@ var yu = class {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t.map((t, n) => t[0] + e[n] + t[1]), this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), t.map((e, t) => {
+		], this.size = !0, this.outputShape = t.map((t, n) => t[0] + e[n] + t[1]), this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), t.map((e, t) => {
 			this.uniforms += ` pad${t} : vec2<i32>,`;
 		}), this.xShape = e, this.shaderKey = "pad";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let paddedCoords = getCoordsFromIndex(index);
           ${vu(this.xShape)}
@@ -6797,11 +6797,11 @@ var Tu = {
 			t,
 			n,
 			e[3]
-		], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "resizeBilinear";
+		], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "resizeBilinear";
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
         let coords = getCoordsFromIndex(index);
           let b = coords[0];
@@ -6868,11 +6868,11 @@ var Nu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.alignCorners = t, this.shaderKey = `resizeBilinearBackprop_${t}`;
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.alignCorners = t, this.shaderKey = `resizeBilinearBackprop_${t}`;
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getOutputCoords();
           let b = coords[0];
@@ -7002,12 +7002,12 @@ var Iu = {
 			t,
 			n,
 			e[3]
-		], this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.halfPixelCenters = r, this.shaderKey = `resizeNearest_${r}`;
+		], this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.halfPixelCenters = r, this.shaderKey = `resizeNearest_${r}`;
 	}
 	getUserCode() {
 		let e;
 		return e = this.halfPixelCenters ? "max((vec2<f32>(rc) + vec2<f32>(0.5)) * effectiveInputOverOutputRatioRC, vec2<f32>(0.0))" : "vec2<f32>(rc) * effectiveInputOverOutputRatioRC", `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let b = coords[0];
@@ -7062,11 +7062,11 @@ var zu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.alignCorners = t, this.shaderKey = `resizeNearestNeigborBackprop_${t}`;
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.alignCorners = t, this.shaderKey = `resizeNearestNeigborBackprop_${t}`;
 	}
 	getUserCode() {
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getOutputCoords();
           let b = coords[0];
@@ -7168,7 +7168,7 @@ var Hu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = " axis : vec4<i32>,", this.shaderKey = "reverse";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = " axis : vec4<i32>,", this.shaderKey = "reverse";
 	}
 	getUserCode() {
 		return `
@@ -7193,7 +7193,7 @@ var Hu = {
         return reverseCoords;
       }
     
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let reverseCoords = getReverseCoords(coords);
@@ -7258,11 +7258,11 @@ var Gu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "centerX : f32, centerY : f32, sinRadians : f32,\n          cosRadians : f32,", this.shaderKey = "rotate", this.outputShape = e, typeof t == "number" ? (this.uniforms += " fillValue : f32,", this.fillSnippet = "var outputValue = uniforms.fillValue;", this.shaderKey += "_float") : (this.uniforms += " fillValue : vec3<f32>,", this.fillSnippet = "var outputValue = uniforms.fillValue[coords[3]];", this.shaderKey += "_vec3");
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "centerX : f32, centerY : f32, sinRadians : f32,\n          cosRadians : f32,", this.shaderKey = "rotate", this.outputShape = e, typeof t == "number" ? (this.uniforms += " fillValue : f32,", this.fillSnippet = "var outputValue = uniforms.fillValue;", this.shaderKey += "_float") : (this.uniforms += " fillValue : vec3<f32>,", this.fillSnippet = "var outputValue = uniforms.fillValue[coords[3]];", this.shaderKey += "_vec3");
 	}
 	getUserCode() {
 		return `
-        ${I("index")} {
+        ${z("index")} {
           if (index < uniforms.size) {
             let coords = getCoordsFromIndex(index);
             let coordXFloat = (f32(coords[2]) - uniforms.centerX) *
@@ -7330,8 +7330,8 @@ var Gu = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = a, this.type = o, this.sumDupeIndices = s, this.dispatchLayout = R(e), this.dispatch = z(this.dispatchLayout, e, this.workgroupSize), this.sliceDimGreaterThanOne = t > 1, this.shaderKey = `scatter_${n}_${r}_${this.sliceDimGreaterThanOne}_${o}_${s}_${i.length}`;
-		let c = F(i.length);
+		], this.atomic = !0, this.outputShape = a, this.type = o, this.sumDupeIndices = s, this.dispatchLayout = P(e), this.dispatch = F(this.dispatchLayout, e, this.workgroupSize), this.sliceDimGreaterThanOne = t > 1, this.shaderKey = `scatter_${n}_${r}_${this.sliceDimGreaterThanOne}_${o}_${s}_${i.length}`;
+		let c = R(i.length);
 		this.uniforms = `sliceDim : i32, strides: ${c}, updatesSize: i32,`, this.updatesRank = r, this.indicesRank = n;
 	}
 	getUserCode() {
@@ -7342,7 +7342,7 @@ var Gu = {
 		let a = `getUpdates(${Array.from({ length: this.updatesRank }, (e, t) => `coords[${t}]`).join(", ")})`;
 		return `
     ${i}
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.updatesSize) {
           let coords = getUpdatesCoordsFromFlatIndex(index);
           var flattenedIndex = 0;
@@ -7351,10 +7351,10 @@ var Gu = {
             flattenedIndex = flattenedIndex + indexInside * ${n};
           }
           let updateValue =
-              ${gr(this.type)}(${a});
+              ${Sr(this.type)}(${a});
           let flatIndex = getOutputIndexFromCoords(${r});
 
-          ${this.sumDupeIndices ? P("&result[flatIndex]", "updateValue", this.type) : "atomicStore(&result[flatIndex], bitcast<i32>(updateValue));"}
+          ${this.sumDupeIndices ? L("&result[flatIndex]", "updateValue", this.type) : "atomicStore(&result[flatIndex], bitcast<i32>(updateValue));"}
         }
       }`;
 	}
@@ -7409,7 +7409,7 @@ var Qu = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.side = t, this.shaderKey = `search_sorted_${t}`;
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.side = t, this.shaderKey = `search_sorted_${t}`;
 	}
 	getUserCode() {
 		return `
@@ -7427,7 +7427,7 @@ var Qu = {
         return right;
       }
 
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let value = getValuesByOutputIndex(index);
@@ -7460,7 +7460,7 @@ var td = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = t, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.cRank = e, this.rank = n, this.shaderKey = "select";
+		], this.size = !0, this.outputShape = t, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.cRank = e, this.rank = n, this.shaderKey = "select";
 	}
 	getUserCode() {
 		let e, t;
@@ -7477,7 +7477,7 @@ var td = {
 			e = r.join(), t = i.join();
 		}
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let resRC = getCoordsFromIndex(index);
           let cVal = getC(${e});
@@ -7538,15 +7538,15 @@ var id = {
 		], this.size = !0;
 		let o = Array(r.length);
 		for (let e = 0; e < o.length; e++) o[e] = r[i[e]];
-		this.outputShape = o, this.newDim = i, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.xShape = e, this.paddedXShape = t, this.uniforms += `reshapedPaddedXShape : ${F(r.length)}, paddedXShapeStrides : ${F(a)}, `, n.map((e, t) => {
+		this.outputShape = o, this.newDim = i, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.xShape = e, this.paddedXShape = t, this.uniforms += `reshapedPaddedXShape : ${R(r.length)}, paddedXShapeStrides : ${R(a)}, `, n.map((e, t) => {
 			this.uniforms += ` pad${t} : vec2<i32>,`;
 		}), this.shaderKey = `spaceToBatchND_${i}`;
 	}
 	getUserCode() {
-		let e = F(this.outputShape.length), t = Pa(this.newDim);
+		let e = R(this.outputShape.length), t = Pa(this.newDim);
 		return `
-      ${mr(this.paddedXShape, "PaddedX")}
-      ${I("index")} {
+      ${br(this.paddedXShape, "PaddedX")}
+      ${z("index")} {
         if(index < uniforms.size) {
           let coords = getCoordsFromIndex(index);
           let switchedIndex = getIndexFromCoords${this.outputShape.length}D(${e}(${t}), uniforms.reshapedPaddedXShape);
@@ -7593,11 +7593,11 @@ var id = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = e, this.type = n, this.dispatchLayout = R([t]), this.dispatch = z(this.dispatchLayout, [t], this.workgroupSize), this.shaderKey = "sparseSegmentSum";
+		], this.atomic = !0, this.outputShape = e, this.type = n, this.dispatchLayout = P([t]), this.dispatch = F(this.dispatchLayout, [t], this.workgroupSize), this.shaderKey = "sparseSegmentSum";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.sparseSize) {
         let indexInSegmentIds = index / uniforms.segmentSize;
         let indexInSegment = index % uniforms.segmentSize;
@@ -7606,7 +7606,7 @@ var id = {
 
         let value = input[indexInInput * uniforms.segmentSize + indexInSegment];
         let outIndex = segmentId * uniforms.segmentSize + indexInSegment;
-        ${P("&result[outIndex]", "value", this.type)}
+        ${L("&result[outIndex]", "value", this.type)}
       }
     }
   `;
@@ -7617,14 +7617,14 @@ var id = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = [e], this.dispatchLayout = R(t), this.dispatch = z(this.dispatchLayout, t, this.workgroupSize), this.shaderKey = "sparseSegmentIdCountProgram";
+		], this.atomic = !0, this.outputShape = [e], this.dispatchLayout = P(t), this.dispatch = F(this.dispatchLayout, t, this.workgroupSize), this.shaderKey = "sparseSegmentIdCountProgram";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.segmentIdsShape) {
         let segmentId = segmentIds[index];
-        ${P("&result[segmentId]", "1", "int32")}
+        ${L("&result[segmentId]", "1", "int32")}
       }
     }
   `;
@@ -7635,11 +7635,11 @@ var id = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.type = t, this.dispatchLayout = R(e), this.dispatch = z(this.dispatchLayout, e, this.workgroupSize), this.shaderKey = "sparseSegmentMean";
+		], this.size = !0, this.outputShape = e, this.type = t, this.dispatchLayout = P(e), this.dispatch = F(this.dispatchLayout, e, this.workgroupSize), this.shaderKey = "sparseSegmentMean";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.size) {
         let segmentId = index / uniforms.segmentSize;
         let count = sameSegmentIdCount[segmentId];
@@ -7731,12 +7731,12 @@ var bd = {
 		], this.size = !0;
 		let n = Array(e.length);
 		for (let r = 0; r < n.length; r++) n[r] = e[r] * t[r];
-		this.outputShape = n, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.rank = this.outputShape.length, this.shaderKey = "tile";
+		this.outputShape = n, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.rank = this.outputShape.length, this.shaderKey = "tile";
 	}
 	getUserCode() {
 		let e = Sd(this.rank, "uniforms.");
 		return `
-      ${I("index")} {
+      ${z("index")} {
         if (index < uniforms.size) {
           let resRC = getCoordsFromIndex(index);
           setOutputAtIndex(index, getA(${e}));
@@ -7901,12 +7901,12 @@ var Nd = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize, [
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize, [
 			this.workPerThread,
 			1,
 			1
 		]);
-		let t = F(this.outputShape.length);
+		let t = R(this.outputShape.length);
 		this.uniforms = `begin : ${t},  strides : ${t}, `, this.shaderKey = "stridedSlice";
 	}
 	getUserCode() {
@@ -7917,7 +7917,7 @@ var Nd = {
 			t = this.outputShape.map((t, n) => (e++, this.outputShape.length === 1 ? `coords * uniforms.strides[${n}] + uniforms.begin[${n}]` : `coords[${e - 1}] * uniforms.strides[${n}] + uniforms.begin[${n}]`)).join(",");
 		}
 		return `
-       ${I("index")} {
+       ${z("index")} {
          if (index < uniforms.size) {
            let coords = getCoordsFromIndex(index);
            setOutputAtIndex(index, getX(${t}));
@@ -8060,11 +8060,11 @@ var Ud = {
 			256,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "inputSize : i32, firstPass : i32, negativeInf : f32,\n        dir : i32, inc : i32,", this.shaderKey = "swap";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "inputSize : i32, firstPass : i32, negativeInf : f32,\n        dir : i32, inc : i32,", this.shaderKey = "swap";
 	}
 	getUserCode() {
 		return `
-        ${I("index")} {
+        ${z("index")} {
           if (index < uniforms.size) {
             let outC = getCoordsFromIndex(index);
             let batch = outC[0];
@@ -8138,11 +8138,11 @@ var Ud = {
 			256,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "inputSize : i32, firstPass : i32, k : i32,", this.shaderKey = "merge";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.uniforms = "inputSize : i32, firstPass : i32, k : i32,", this.shaderKey = "merge";
 	}
 	getUserCode() {
 		return `
-        ${I("index")} {
+        ${z("index")} {
           if (index < uniforms.size) {
             let outC = getCoordsFromIndex(index);
             let batch = outC[0];
@@ -8324,7 +8324,7 @@ var Yd = {
 			64,
 			1,
 			1
-		], this.size = !0, this.outputShape = e, this.dispatchLayout = R(this.outputShape), this.dispatch = z(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "transform";
+		], this.size = !0, this.outputShape = e, this.dispatchLayout = P(this.outputShape), this.dispatch = F(this.dispatchLayout, this.outputShape, this.workgroupSize), this.shaderKey = "transform";
 	}
 	getUserCode() {
 		return `
@@ -8391,7 +8391,7 @@ var Yd = {
             return outputValue;
           }
 
-          ${I("index")} {
+          ${z("index")} {
             if (index < uniforms.size) {
               let coords = getCoordsFromIndex(index);
               var outputValue : f32;
@@ -8531,13 +8531,13 @@ var ef = {
 			64,
 			1,
 			1
-		], this.atomic = !0, this.outputShape = t, this.dispatchLayout = R(e), this.dispatch = z(this.dispatchLayout, e, this.workgroupSize), n !== "float32" && n !== "int32") throw Error(`UnsortedSegmentSum only supports float32 and int32
+		], this.atomic = !0, this.outputShape = t, this.dispatchLayout = P(e), this.dispatch = F(this.dispatchLayout, e, this.workgroupSize), n !== "float32" && n !== "int32") throw Error(`UnsortedSegmentSum only supports float32 and int32
               types, does not support ${n} type.`);
 		this.type = n, this.shaderKey = "unsortedSegmentSum";
 	}
 	getUserCode() {
 		return `
-    ${I("index")} {
+    ${z("index")} {
       if (index < uniforms.xSize) {
         let coords = getXCoordsFromIndex(index);
         let b = coords[0];
@@ -8548,7 +8548,7 @@ var ef = {
           let flatIndex = b * uniforms.numSegments + segmentId % uniforms.numSegments;
           let value = getX(b, inCol);
 
-          ${P("&result[flatIndex]", "value", this.type)}
+          ${L("&result[flatIndex]", "value", this.type)}
         }
       }
     }