fix wasm backend memory leak issue. (#7915)

LiujhDev · web-flow · commit c8915b212c84 · 2023-08-16T19:40:57.000-07:00
BUG
diff --git a/tfjs-backend-wasm/src/kernels/BatchToSpaceND.ts b/tfjs-backend-wasm/src/kernels/BatchToSpaceND.ts
@@ -56,7 +56,7 @@ function batchToSpaceND(args: {
 
   backend.disposeData(xReshaped.dataId);
   backend.disposeData(xTransposed.dataId);
-  backend.disposeData(xReshaped.dataId);
+  backend.disposeData(xTransposedReshaped.dataId);
 
   return result;
 }

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ function batchToSpaceND(args: {`
`56`	`56`
`57`	`57`	`backend.disposeData(xReshaped.dataId);`
`58`	`58`	`backend.disposeData(xTransposed.dataId);`
`59`		`- backend.disposeData(xReshaped.dataId);`
	`59`	`+ backend.disposeData(xTransposedReshaped.dataId);`
`60`	`60`
`61`	`61`	`return result;`
`62`	`62`	`}`