XNNPACK: Support 1D input for Conv and ConvTranspose #20349
skottmckay
merged
15 commits into
main
from
skottmckay/XNNPACK_1D_Conv_ConvTranspose
Support 1D Conv and ConvTranspose in XNNPACK by faking a height of 1.
72cdc246
Add 1D test with group.
2246e9d5
Improve consistency/comments.
d0d7d560
Merge remote-tracking branch 'origin/main' into skottmckay/XNNPACK_1D…
dc4c6bee
Fix DML check to handle 1D input.
04317472
edgchen1
approved these changes
on 2024-04-19
edgchen1
dismissed these changes
on 2024-04-19
Fix CUDA NHWC ConvTranspose
0fdd76f4
Merge remote-tracking branch 'origin/main' into skottmckay/XNNPACK_1D…
88f20695
Fix unused pararm error from making `if` constexpr
3a783747
skottmckay
commented
on 2024-04-19
Fix build error when CUDA NHWC ops are disabled
d81ce659
Fix Linux build error due to making `if` for NHWC constexpr
9c368369
Fix unused params.
714ff43e
Update 1D test so the W values are not all 1 to show issues with tran…
d7cc3611
hariharans29
commented
on 2024-04-20
hariharans29
commented
on 2024-04-20
Support either the input channels or output channels of W being trans…
974fa036
Make consistent with Conv usage of GetCudnnConv1dPadToNc1d
8caf3fda
hariharans29
dismissed these changes
on 2024-04-20
hariharans29
commented
on 2024-04-20
Fix issue with Y shape cleanup. Manually verified. Tests don't curren…
083027c7
gedoensmax
approved these changes
on 2024-04-22
hariharans29
approved these changes
on 2024-04-23
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub