use shared util func

tafsiri · tafsiri · commit b91f1d929d91 · 2019-05-20T17:48:34.000-04:00
diff --git a/src/backends/webgl/shader_compiler.ts b/src/backends/webgl/shader_compiler.ts
@@ -908,7 +908,7 @@ function getPackedSampler3D(inputInfo: InputInfo): string {
   const funcName = 'get' + texName.charAt(0).toUpperCase() + texName.slice(1);
 
   if (shape[0] === 1) {
-    const squeezedShape = shape.slice(1);
+    const squeezedShape = util.packedShapeTransform(shape);
     const keptDims = [1, 2];
     const newInputInfo = squeezeInputInfo(inputInfo, squeezedShape);
     const params = ['b', 'row', 'col'];
