From 417aa004069175335e64a2b8f1490b857e3ad304 Mon Sep 17 00:00:00 2001 From: xhcao Date: Mon, 19 Aug 2024 06:45:13 +0800 Subject: [PATCH] [js/webgpu] fix conv1d error (#21585) ### Description ### Motivation and Context --- js/web/lib/wasm/jsep/webgpu/ops/conv.ts | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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]], ), ); };