[js/webgpu] Add BatchNormalization Op #18468
Add BatchNormalization Op
51f61772
Fix rebase errors
b97cfe16
fix shader compilation errors
be6fbf8d
qjia7
changed the title [js/webgpu] Add BatchNormalization Op [WIP][js/webgpu] Add BatchNormalization Op 2 years ago
qjia7
marked this pull request as draft 2 years ago
Use the new algorithm for inference
30697b3d
remove training mode
cbee7a0b
register nhwc for opset 7-8
3ca14eea
Add uniforms support
a5cea1b2
Revert "Add uniforms support"
1c358cc7
nits
1dedfd75
qjia7
force pushed
from
87d4adc4
to
1dedfd75
2 years ago
qjia7
changed the title [WIP][js/webgpu] Add BatchNormalization Op [js/webgpu] Add BatchNormalization Op 2 years ago
qjia7
marked this pull request as ready for review 2 years ago
Merge branch 'main' into batch_norm
9adde7b7
add vectorize support
b1749efb
enable uniforms
dd145134
put validateInputs behind flag
933a8ec0
address comments
23b6cb8d
qjia7
dismissed their stale review
via 23b6cb8d
2 years ago
qjia7
commented
on 2023-11-21
use ?: instead of Object.seal
74783b03
Fix errors when spatial = false with NHWC
bb58c898
add all float data types support
f8f53996
fs-eire
approved these changes
on 2023-11-22
fs-eire
approved these changes
on 2023-11-22
fs-eire
merged
64dacc28
into main 2 years ago
qjia7
deleted the batch_norm branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub