Fixed computing channel-last output shape

satyajandhyala · satyajandhyala · commit da2fb5e68bf3 · 2023-10-19T15:14:54.000-07:00
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/pool.ts b/js/web/lib/wasm/jsep/webgpu/ops/pool.ts
@@ -47,7 +47,7 @@ const getAdjustedPoolAttributesAndOutputShape = <AttributeType extends AveragePo
     Object.assign(newAttributes, {kernelShape, strides, pads, cacheKey: attributes.cacheKey});
   }
   const outputShapeAsChannelLast = outputShapeAsChannelFirst.slice();
-  outputShapeAsChannelLast.splice(1, 0, outputShapeAsChannelLast.pop()!);
+  outputShapeAsChannelLast.push(outputShapeAsChannelLast.splice(1, 1)[0]);
   return [newAttributes, isChannelsLast ? outputShapeAsChannelLast : outputShapeAsChannelFirst];
 };
 

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ const getAdjustedPoolAttributesAndOutputShape = <AttributeType extends AveragePo`
`47`	`47`	`Object.assign(newAttributes, {kernelShape, strides, pads, cacheKey: attributes.cacheKey});`
`48`	`48`	`}`
`49`	`49`	`const outputShapeAsChannelLast = outputShapeAsChannelFirst.slice();`
`50`		`- outputShapeAsChannelLast.splice(1, 0, outputShapeAsChannelLast.pop()!);`
	`50`	`+ outputShapeAsChannelLast.push(outputShapeAsChannelLast.splice(1, 1)[0]);`
`51`	`51`	`return [newAttributes, isChannelsLast ? outputShapeAsChannelLast : outputShapeAsChannelFirst];`
`52`	`52`	`};`
`53`	`53`