From 8b2772313874c9fa81e9919b3d0c5e664c90386c Mon Sep 17 00:00:00 2001 From: xhcao Date: Mon, 26 Dec 2022 04:13:29 +0800 Subject: [PATCH] webgl: fix a typing nit (#7199) DOC --- tfjs-backend-webgl/src/lrn_grad_gpu.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tfjs-backend-webgl/src/lrn_grad_gpu.ts b/tfjs-backend-webgl/src/lrn_grad_gpu.ts index 3dedc3b903e..998f535c887 100644 --- a/tfjs-backend-webgl/src/lrn_grad_gpu.ts +++ b/tfjs-backend-webgl/src/lrn_grad_gpu.ts @@ -74,7 +74,7 @@ export class LRNGradProgram implements GPGPUProgram { else if (k >= depthBegin && k < depthEnd){ float dyi = -2.0 * float(${alpha}) * float(${beta}) - * getInputImage(b ,r ,c, k) * getOutputImage(b, r, c, d) + * getInputImage(b, r, c, k) * getOutputImage(b, r, c, d) / norm; if (k == d) { dyi += pow(norm, -1.0 * ${beta});