Skip to content

Commit 417aa00

Browse files
authored
[js/webgpu] fix conv1d error (microsoft#21585)
### Description <!-- Describe your changes. --> ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent d1d40fb commit 417aa00

File tree

1 file changed

+3
-1
lines changed
  • js/web/lib/wasm/jsep/webgpu/ops

1 file changed

+3
-1
lines changed

js/web/lib/wasm/jsep/webgpu/ops/conv.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,9 @@ const conv1d = (context: ComputeContext, attributes: ConvAttributes): void => {
359359
);
360360
context.compute(
361361
createGroupedConvProgramInfo(inputs, adjustedAttributes, (outputShape) =>
362-
isChannelLast ? [outputShape[0], outputShape[2], outputShape[3]] : [],
362+
isChannelLast
363+
? [outputShape[0], outputShape[2], outputShape[3]]
364+
: [outputShape[0], outputShape[1], outputShape[3]],
363365
),
364366
);
365367
};

0 commit comments

Comments
 (0)