onnxruntime
[js/webgpu] Optimize InstanceNormalization
#21995
Merged

[js/webgpu] Optimize InstanceNormalization #21995

fs-eire merged 7 commits into microsoft:main from qjia7:opt_instanceNorm
qjia7
qjia7 InstanceNorm is NCHW friendly
9640c235
qjia7 Revert "InstanceNorm is NCHW friendly"
89985830
qjia7 always use NCHW laytout for InstanceNorm
d9f59293
qjia7
qjia7
qjia7 qjia7 marked this pull request as draft 1 year ago
qjia7 qjia7 changed the title [js/webgpu] Don't do layout conversion for InstanceNormalization [WIP][js/webgpu] Don't do layout conversion for InstanceNormalization 1 year ago
qjia7 Optimize the computeMean algorithm
1aede045
qjia7 use snake naming
212ed06e
qjia7 Revert "always use NCHW laytout for InstanceNorm"
bc28bea3
qjia7 qjia7 changed the title [WIP][js/webgpu] Don't do layout conversion for InstanceNormalization [WIP][js/webgpu] Optimize InstanceNormalization 1 year ago
qjia7 qjia7 changed the title [WIP][js/webgpu] Optimize InstanceNormalization [js/webgpu] Optimize InstanceNormalization 1 year ago
qjia7 qjia7 marked this pull request as ready for review 1 year ago
qjia7
qjia7
fs-eire
azure-pipelines
qjia7 split InstanceNormalization shader to fix the gpu crash issue
9e06e055
qjia7
qjia7
fs-eire
azure-pipelines
fs-eire
fs-eire
fs-eire
azure-pipelines
azure-pipelines
azure-pipelines
snnn
snnn approved these changes on 2024-09-23
fs-eire
fs-eire approved these changes on 2024-09-23
fs-eire fs-eire merged 80e9df82 into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone