onnxruntime
[JS/Web] Add ConvTranspose implementation using MatMul
#17573
Merged

[JS/Web] Add ConvTranspose implementation using MatMul #17573

satyajandhyala
satyajandhyala satyajandhyala added ep:WebGPU
satyajandhyala Add ConvTranspose implementation using MatMul.
e3677051
satyajandhyala Merged changes from main.
dfeb109e
satyajandhyala Hardcode dispatch.
fdd29bc6
satyajandhyala Added a new test case to exercise vec4 version.
8ed9038b
satyajandhyala Fixed filter setting
6794407e
satyajandhyala Added dilation support for ConvTranspose matmul implementation.
33ecf4c2
satyajandhyala Added dilation fix.
01c644f6
satyajandhyala satyajandhyala force pushed from b972882e to 01c644f6 2 years ago
satyajandhyala Formatted
cbf19076
satyajandhyala Fix format.
2a1d2367
satyajandhyala satyajandhyala marked this pull request as ready for review 2 years ago
satyajandhyala satyajandhyala changed the title [WIP][JS/Web] Add ConvTranspose implementation using MatMul [JS/Web] Add ConvTranspose implementation using MatMul 2 years ago
satyajandhyala Minor change
36689641
satyajandhyala Minor optimization changes.
fd374e9c
satyajandhyala modified logic to increase readability.
4cd2334f
fs-eire
fs-eire commented on 2023-09-21
satyajandhyala Add ConvTranspose MatMul support NHWC
1c32af2f
satyajandhyala satyajandhyala requested a review from fs-eire fs-eire 2 years ago
satyajandhyala Use effectiveFilterDims.
8200c4c3
satyajandhyala Use naive implementation if stride is non-trivial.
1568f5ea
qjia7
qjia7 commented on 2023-09-26
satyajandhyala Enable MatMul version of ConvTranspose even for non-unit strides.
0ec2ed47
satyajandhyala Use naive implementation for non-unit dilations.
f98341f7
satyajandhyala Enable ConvTranspose matmul implementation even if dilations > 1.
cedc66bf
satyajandhyala satyajandhyala requested a review from qjia7 qjia7 2 years ago
fs-eire
fs-eire approved these changes on 2023-09-29
fs-eire fs-eire merged b4fbc25b into main 2 years ago
fs-eire fs-eire deleted the sajandhy/webgpu_add_matmul_convtranspose branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone