Skip to content

Commit

Permalink
[js/webgpu] fix conv1d error (microsoft#21585)
Browse files Browse the repository at this point in the history
### 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. -->
  • Loading branch information
xhcao authored Aug 18, 2024
1 parent d1d40fb commit 417aa00
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion js/web/lib/wasm/jsep/webgpu/ops/conv.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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]],
),
);
};
Expand Down

0 comments on commit 417aa00

Please sign in to comment.