diff --git a/js/web/lib/wasm/jsep/webgpu/ops/conv.ts b/js/web/lib/wasm/jsep/webgpu/ops/conv.ts index f1469d4ce67be..2a32b566ba4ba 100644 --- a/js/web/lib/wasm/jsep/webgpu/ops/conv.ts +++ b/js/web/lib/wasm/jsep/webgpu/ops/conv.ts @@ -359,7 +359,9 @@ const conv1d = (context: ComputeContext, attributes: ConvAttributes): void => { ); context.compute( createGroupedConvProgramInfo(inputs, adjustedAttributes, (outputShape) => - isChannelLast ? [outputShape[0], outputShape[2], outputShape[3]] : [], + isChannelLast + ? [outputShape[0], outputShape[2], outputShape[3]] + : [outputShape[0], outputShape[1], outputShape[3]], ), ); };