[cpu] Cast pixel to number in maxpool utils (#7534)

This is in preparation for TypeScript 5.0. See cl/520770605 for details.
tensorflow · Mar 31, 2023 · 53e17f6 · 53e17f6
1 parent 430bc13
commit 53e17f6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tfjs-backend-cpu/src/utils/pool_utils.ts b/tfjs-backend-cpu/src/utils/pool_utils.ts
@@ -127,7 +127,10 @@ export function maxPoolPositions(
             const wR = xR - xRCorner;
             for (let xC = xCMin; xC < xCMax; xC += dilationWidth) {
               const wC = xC - xCCorner;
-              const pixel = xBuf.get(b, xR, xC, d);
+              // For some reason, disable-next-line is not working
+              // TODO(mattsoulanille): Remove this when switching to TS5.
+              /* tslint:disable: no-unnecessary-type-assertion */
+              const pixel = xBuf.get(b, xR, xC, d) as number;
               if (pixel > maxValue) {
                 maxValue = pixel as number;
                 if (flattenPositions) {
@@ -313,7 +316,8 @@ export function maxPool3dPositions(
                 for (let xCol = xColMin; xCol < xColMax;
                      xCol += dilationWidth) {
                   const wCol = xCol - xColCorner;
-                  const pixel = xBuf.get(batch, xDepth, xRow, xCol, channel);
+                  const pixel = xBuf.get(batch, xDepth, xRow, xCol,
+                                         channel) as number;
                   if (pixel >= maxValue) {
                     maxValue = pixel as number;
                     maxPosition =