LayerNormalization broadcast (limited support for axis=2) #23297
broadcast 3D scale and bias in layer norm for axis=2
0d89e5d1
tianleiwu
marked this pull request as draft 350 days ago
refactoring for test
35d27130
format
233a60ce
fp16
aac9f028
clean up
590643a1
tianleiwu
marked this pull request as ready for review 349 days ago
update comment
102cbca7
refine error message
46d7c38c
update comment
5b2290fe
Merge branch 'main' into tlwu/layer_norm_broadcast
64381aee
refactoring
ee01f927
tianleiwu
dismissed their stale review
via ee01f927
348 days ago
update comments
e765ce62
tianleiwu
merged
73f5b0c5
into main 348 days ago
tianleiwu
deleted the tlwu/layer_norm_broadcast branch 348 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub